make sure exported files get deleted after tests

hashtopolis · Feb 17, 2024 · 422e940 · 422e940
1 parent c1912c7
commit 422e940
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ci/apiv2/test_hashlist.py b/ci/apiv2/test_hashlist.py
@@ -48,6 +48,8 @@ def test_export_cracked_hashes(self):
         self.assertEqual(int(file.id), obj.id)
         self.assertIn('Pre-cracked_', obj.filename)
 
+        self.delete_after_test(obj)
+
     def test_export_left_hashes(self):
         model_obj = self.create_test_object(file_id='001')
 
@@ -58,6 +60,8 @@ def test_export_left_hashes(self):
         self.assertEqual(int(file.id), obj.id)
         self.assertIn('Leftlist_', obj.filename)
 
+        self.delete_after_test(obj)
+
     def test_export_wordlist(self):
         model_obj = self.create_test_object(file_id='001')
 
@@ -72,6 +76,8 @@ def test_export_wordlist(self):
         self.assertEqual(int(file.id), obj.id)
         self.assertIn('Wordlist_', obj.filename)
 
+        self.delete_after_test(obj)
+
     def test_import_cracked_hashes(self):
         model_obj = self.create_test_object(file_id='001')