ensure SeaResultSet resp

varun-edachali-dbx · varun-edachali-dbx · commit 4be68082711e · 2025-08-04T03:43:33.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/tests/unit/test_sea_backend.py b/tests/unit/test_sea_backend.py
@@ -974,7 +974,7 @@ def test_get_tables_with_cloud_fetch(
     ):
         """Test the get_tables method with cloud fetch enabled."""
         # Mock the execute_command method and ResultSetFilter
-        mock_result_set = Mock()
+        mock_result_set = Mock(spec=SeaResultSet)
 
         with patch.object(
             sea_client_cloud_fetch, "execute_command", return_value=mock_result_set
@@ -1013,7 +1013,7 @@ def test_get_schemas_with_cloud_fetch(
     ):
         """Test the get_schemas method with cloud fetch enabled."""
         # Mock the execute_command method
-        mock_result_set = Mock()
+        mock_result_set = Mock(spec=SeaResultSet)
         with patch.object(
             sea_client_cloud_fetch, "execute_command", return_value=mock_result_set
         ) as mock_execute:
