From a52863d758af34cddaadeae6e7f7c1cffcaacc9d Mon Sep 17 00:00:00 2001 From: Benjamin Smith Date: Thu, 26 Dec 2024 09:35:51 -0500 Subject: [PATCH] Reduce diff --- tests/unit/job_test.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/unit/job_test.py b/tests/unit/job_test.py index 1321ec4..2d35546 100644 --- 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):