Fixed tests

This commit is contained in:
Manuel Cortez 2020-07-07 21:14:22 -05:00
parent ff8c8ac22b
commit b1f239fe82
2 changed files with 6 additions and 6 deletions

View File

@ -10,9 +10,9 @@ app_type = ""
def setup():
global data_directory, app_type
if len(glob.glob("Uninstall.exe")) > 0: # installed copy
if os.path.exists(paths.app_data_path("musicDL")) == False:
paths.prepare_app_data_path("musicDL")
data_directory = paths.app_data_path("musicDL")
if os.path.exists(paths.data_path("musicDL")) == False:
os.mkdir(paths.data_path("musicDL"))
data_directory = paths.data_path("musicDL")
app_type = "installed"
else:
app_type = "portable"

View File

@ -27,13 +27,13 @@ class storageTestCase(unittest.TestCase):
storage.setup()
self.assertEquals(storage.app_type, "installed")
self.assertTrue(os.path.exists(storage.data_directory))
self.assertEquals(storage.data_directory, paths.app_data_path("musicDL"))
self.assertEquals(storage.data_directory, paths.data_path("musicDL"))
def tearDown(self):
""" Removes uninstall.exe created for tests and data path."""
fix_winpaths.fix()
if os.path.exists(paths.app_data_path("musicDL")):
shutil.rmtree(paths.app_data_path("musicDL"))
if os.path.exists(paths.data_path("musicDL")):
shutil.rmtree(paths.data_path("musicDL"))
if os.path.exists(os.path.join(paths.app_path(), "uninstall.exe")):
os.remove(os.path.join(paths.app_path(), "uninstall.exe"))