Fix Windows env vars for config test

This commit is contained in:
Adrian Sampson 2016-06-09 09:58:10 -07:00
parent e508daea63
commit a45dcdc674

View file

@ -657,6 +657,11 @@ class ConfigTest(unittest.TestCase, TestHelper, _common.Assertions):
self._old_home = os.environ.get('HOME') self._old_home = os.environ.get('HOME')
os.environ['HOME'] = self.temp_dir os.environ['HOME'] = self.temp_dir
# Also set APPDATA, the Windows equivalent of setting $HOME.
self._old_appdata = os.environ.get('APPDATA')
os.environ['APPDATA'] = \
os.path.join(self.temp_dir, 'AppData', 'Roaming')
self._orig_cwd = os.getcwd() self._orig_cwd = os.getcwd()
self.test_cmd = self._make_test_cmd() self.test_cmd = self._make_test_cmd()
commands.default_commands.append(self.test_cmd) commands.default_commands.append(self.test_cmd)
@ -685,6 +690,10 @@ class ConfigTest(unittest.TestCase, TestHelper, _common.Assertions):
os.chdir(self._orig_cwd) os.chdir(self._orig_cwd)
if self._old_home is not None: if self._old_home is not None:
os.environ['HOME'] = self._old_home os.environ['HOME'] = self._old_home
if self._old_appdata is None:
del os.environ['APPDATA']
else:
os.environ['APPDATA'] = self._old_appdata
self.teardown_beets() self.teardown_beets()
def _make_test_cmd(self): def _make_test_cmd(self):