Reduce diff

bh2smith · Dec 26, 2024 · a52863d · a52863d
1 parent 373ba4e
commit a52863d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/unit/job_test.py b/tests/unit/job_test.py
@@ -1,5 +1,5 @@
 import unittest
-from unittest.mock import AsyncMock, patch
+from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
 from dune_client.query import QueryBase
@@ -36,8 +36,9 @@ async def test_metrics_collection(self, mock_metrics_push, *_):
             db_url="postgresql://postgres:postgres@localhost:5432/postgres",
             table_name="some_table",
         )
+
         src.fetch = AsyncMock()
-        dest.save = AsyncMock()
+        dest.save = Mock()
         test_job = Job(name="job name", source=src, destination=dest)
 
         with patch("src.metrics.env", return_value=None):