Save was not async

tests/unit/job_test.py

@@ -1,10 +1,12 @@
 import unittest
-from unittest.mock import AsyncMock, patch
+from unittest.mock import AsyncMock, Mock, patch
 
+from pandas import DataFrame
 import pytest
 from dune_client.query import QueryBase
 
 from src.destinations.postgres import PostgresDestination
+from src.interfaces import TypedDataFrame
 from src.job import Database, Job
 from src.sources.dune import DuneSource
 
@@ -36,8 +38,11 @@ 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()
+
+        # Create mock DataFrame with explicit is_empty return value
+        test_df = TypedDataFrame(DataFrame({"col1": [1, 2, 3], "col2": ["a", "b", "c"]}), {})
+        src.fetch = AsyncMock(return_value=test_df)
+        dest.save = Mock(return_value=3)
         test_job = Job(name="job name", source=src, destination=dest)
 
         with patch("src.metrics.env", return_value=None):