fixed some tests

cowprotocol · Sep 20, 2024 · cb7d7d0 · cb7d7d0
1 parent 111a5d9
commit cb7d7d0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/tests/e2e/test_query_runner.py b/tests/e2e/test_query_runner.py
@@ -18,7 +18,8 @@ def test_query_runner(self, mocked_post):
         query = load_config(filepath("v2-test-data.yaml")).query
         dune = DuneClient(os.environ["DUNE_API_KEY"])
         slack_client = BasicSlackClient(token="Fake Token", channel="Fake Channel")
-        query_runner = QueryRunner(query, dune, slack_client)
+        ping_frequency = 10
+        query_runner = QueryRunner(query, dune, slack_client, ping_frequency)
         query_runner.run_loop()
         mocked_post.assert_called_with(
             f"{query.name} - detected 1 cases. Results available at {query.result_url()}"
@@ -30,7 +31,8 @@ def test_v3_query(self, mocked_post):
         query = load_config(filepath("v3-left-bounded.yaml")).query
         dune = DuneClient(os.environ["DUNE_API_KEY"])
         slack_client = BasicSlackClient(token="Fake Token", channel="Fake Channel")
-        query_runner = QueryRunner(query, dune, slack_client)
+        ping_frequency = 10
+        query_runner = QueryRunner(query, dune, slack_client, ping_frequency)
         query_runner.run_loop()
         mocked_post.assert_called_with(
             f"{query.name} - detected 1 cases. Results available at {query.result_url()}"
@@ -42,7 +44,8 @@ def test_v3_last_hour(self, mocked_post):
         query = load_config(filepath("v3-last-hour.yaml")).query
         dune = DuneClient(os.environ["DUNE_API_KEY"])
         slack_client = BasicSlackClient(token="Fake Token", channel="Fake Channel")
-        query_runner = QueryRunner(query, dune, slack_client)
+        ping_frequency = 10
+        query_runner = QueryRunner(query, dune, slack_client, ping_frequency)
         query_runner.run_loop()
         mocked_post.assert_called_with(
             f"{query.name} - detected 1 cases. Results available at {query.result_url()}"

diff --git a/tests/unit/test_models.py b/tests/unit/test_models.py
@@ -67,9 +67,9 @@ def test_as_query_params(self):
         self.assertEqual(
             window.as_query_parameters(),
             [
-                QueryParameter.date_type(name="StartTime", value=self.start),
+                QueryParameter.date_type(name="start_time", value=self.start),
                 QueryParameter.date_type(
-                    name="EndTime", value=self.start + timedelta(hours=window.length)
+                    name="end_time", value=self.start + timedelta(hours=window.length)
                 ),
             ],
         )