Merge pull request #583 from jennydaman/cleanup-FileBrowserViewTests

Clean up storage after FileBrowserViewTests
FNNDSC · Oct 2, 2024 · 646df44 · 646df44
2 parents a1b1d0a + 4424e21
commit 646df44
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/chris_backend/filebrowser/tests/test_views.py b/chris_backend/filebrowser/tests/test_views.py
@@ -55,8 +55,13 @@ def setUpClass(cls):
 
     @classmethod
     def tearDownClass(cls):
-        User.objects.get(username=cls.username).delete()
-        User.objects.get(username=cls.other_username).delete()
+        storage_manager = connect_storage(settings)
+
+        for username in [cls.username, cls.other_username]:
+            User.objects.get(username=username).delete()
+            home = f'home/{username}'
+            if storage_manager.path_exists(home):
+                storage_manager.delete_path(home)
 
         # re-enable logging
         logging.disable(logging.NOTSET)