From 89b8be075585f330a5806e5239574cff399a400f Mon Sep 17 00:00:00 2001 From: mike cullerton Date: Tue, 28 Sep 2021 10:25:13 -0400 Subject: [PATCH] Take saved files into account during assertion --- tests/files/test_file_data_cleanup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/files/test_file_data_cleanup.py b/tests/files/test_file_data_cleanup.py index 76df91eb..5222339c 100644 --- 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)