attempt to fix flaky test

Author: snopoke

tests/test_cli_run.py

@@ -52,7 +52,8 @@ def _should_update_task(last_update, update_frequency_seconds):
         return True
 
     with mock.patch("taskbadger.process._should_update", new=_should_update_task):
-        _test_cli_run(["echo test; sleep 0.11"], 0, args=["task_name"], update_call_count=3)
+        update_mock = _test_cli_run(["echo test; sleep 0.2"], 0, args=["task_name"], update_call_count=0)
+        assert update_mock.call_count > 2
 
 
 def test_cli_capture_output():
@@ -160,7 +161,8 @@ def _update(*args, **kwargs):
                 data={"return_code": return_code},
             )
 
-        assert update_mock.call_count == update_call_count
+        if update_call_count:
+            assert update_mock.call_count == update_call_count
         update_mock.assert_called_with(
             client=mock.ANY,
             organization_slug="org",