From b635dff88a2a9201ec8f3ef5c83d1d3a5c3cf83a Mon Sep 17 00:00:00 2001
From: Amir Mofakhar <amir.mofakhar@wise.com>
Date: Wed, 17 Jul 2024 16:42:07 +0100
Subject: [PATCH] fix unit test

---
 tests/units/cli/cli_args.py | 2 ++
 tests/units/test_utils.py   | 7 ++++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/tests/units/cli/cli_args.py b/tests/units/cli/cli_args.py
index 613339867..415a0f2cf 100644
--- 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
index c505f027e..cbd1417fd 100644
--- 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"""