update progress bar unit tests

Author: AlyaGomaa

tests/run_all_tests.sh

@@ -6,10 +6,7 @@ printf "0" | ./slips.py -k
 
 # run all unit tests, -n *5 means distribute tests on 5 different process
 # -s to see print statements as they are executed
-python3  -m pytest tests/ --ignore="tests/test_database.py" --ignore="tests/integration_tests" -n 7 -p no:warnings -vvvv -s
-
-## run db tests serially/using 1 worker
-python3  -m pytest tests/test_database.py -p no:warnings -vvvv -s
+python3  -m pytest tests/ --ignore="tests/integration_tests" -n 7 -p no:warnings -vvvv -s
 
 # Close all redis-servers opened by the unit tests
 python3 tests/destrctor.py

tests/test_progress_bar.py

@@ -167,21 +167,19 @@ def test_remove_stats():
 
 
 @pytest.mark.parametrize(
-    "total_flows, current_n",
+    "total_flows, current_n, keyboard_int",
     [
         # testcase1: Normal case
-        (100, 100),
+        (100, 100, 0),
         # testcase2: Edge case - zero flows
-        (0, 0),
+        (0, 0, 1),
         # testcase3: Large number of flows
-        (1000000, 1000000),
+        (1000000, 1000000, 0),
     ],
 )
-def test_shutdown_gracefully(
-    total_flows,
-    current_n,
-):
+def test_shutdown_gracefully(total_flows, current_n, keyboard_int):
     pbar = ModuleFactory().create_progress_bar_obj()
+    pbar.keyboard_int_ctr = keyboard_int
     pbar.total_flows = total_flows
     pbar.pbar_finished = Event()
 
@@ -193,10 +191,13 @@ def test_shutdown_gracefully(
         "tqdm.auto.tqdm.write"
     ) as mock_write:
         pbar.shutdown_gracefully()
-
         mock_remove_stats.assert_called_once()
-        mock_write.assert_called_once_with(
-            "Profiler is done reading all flows. "
-            "Slips is now processing them."
-        )
+
+        if not keyboard_int:
+            mock_write.assert_not_called()
+        else:
+            mock_write.assert_called_once_with(
+                "Profiler is done reading all flows. "
+                "Slips is now processing them."
+            )
         assert pbar.pbar_finished.is_set()