Take saved files into account during assertion

2021-09-28 10:25:13 -04:00 · 2021-09-28 10:25:13 -04:00 · 89b8be0755
parent 767a90faba
commit 89b8be0755
1 changed files with 1 additions and 1 deletions
--- a/tests/files/test_file_data_cleanup.py
+++ b/tests/files/test_file_data_cleanup.py
@ -138,5 +138,5 @@ class TestFileDataCleanup(BaseTest):

        # assert correct versions are removed
        new_count = session.query(FileDataModel).count()
-        self.assertEqual(set(deleted_models), set(replaced_models))
+        self.assertEqual(set(deleted_models).union(set(saved_models)), set(replaced_models))
        self.assertEqual(file_data_model_count, new_count)