fix unit test

transferwise · Jul 17, 2024 · b635dff · b635dff
1 parent 2d51fee
commit b635dff
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tests/units/cli/cli_args.py b/tests/units/cli/cli_args.py
@@ -21,6 +21,7 @@ def __init__(
         extra_log=False,
         debug=False,
         profiler=False,
+        force=False
     ):
         self.target = target
         self.tap = tap
@@ -34,6 +35,7 @@ def __init__(
         self.extra_log = extra_log
         self.debug = debug
         self.profiler = profiler
+        self.force = False
 
     # "log" Getters and setters
     @property

diff --git a/tests/units/test_utils.py b/tests/units/test_utils.py
@@ -171,9 +171,10 @@ def test_filter_out_selected_tables(self):
 
         actual_output = utils.filter_out_selected_tables(all_schema_tables, selected_tables)
 
-        self.assertListEqual(
-            actual_output, expected_output
-        )
+        self.assertEqual(len(actual_output), 2)
+
+        for item in expected_output:
+            self.assertIn(item, actual_output)
 
     def test_get_schema_of_tables_set(self):
         """Test get_schema_of_mysql_tables method works correctly"""