fix tests

Author: Light2Dark

tests/_runtime/test_runtime_datasets.py

@@ -7,6 +7,7 @@
 from marimo._dependencies.dependencies import DependencyManager
 from marimo._messaging.ops import (
     DataSourceConnections,
+    SQLMetadata,
     SQLTableListPreview,
     SQLTablePreview,
     ValidateSQLResult,
@@ -99,7 +100,12 @@ async def test_non_existent_engine(
         ]
         assert preview_sql_table_results == [
             SQLTablePreview(
-                request_id=RequestId("0"), table=None, error="Engine not found"
+                request_id=RequestId("0"),
+                table=None,
+                error="Engine not found",
+                metadata=SQLMetadata(
+                    connection=DUCKDB_CONN, database="test", schema="test"
+                ),
             )
         ]
 
@@ -126,7 +132,14 @@ async def test_catalog_engine(
             op for op in stream.operations if isinstance(op, SQLTablePreview)
         ]
         assert preview_sql_table_results == [
-            SQLTablePreview(request_id=RequestId("0"), table=None, error=None)
+            SQLTablePreview(
+                request_id=RequestId("0"),
+                table=None,
+                error=None,
+                metadata=SQLMetadata(
+                    connection=DUCKDB_CONN, database="test", schema="test"
+                ),
+            )
         ]
 
     async def test_query_engine(
@@ -156,6 +169,9 @@ async def test_query_engine(
                 request_id=RequestId("0"),
                 table=None,
                 error="Connection does not support catalog operations",
+                metadata=SQLMetadata(
+                    connection=SQLITE_CONN, database="test", schema="test"
+                ),
             )
         ]
 
@@ -182,7 +198,12 @@ async def test_non_existent_engine(
         ]
         assert preview_sql_table_list_results == [
             SQLTableListPreview(
-                request_id=RequestId("0"), tables=[], error="Engine not found"
+                request_id=RequestId("0"),
+                tables=[],
+                error="Engine not found",
+                metadata=SQLMetadata(
+                    connection=DUCKDB_CONN, database="test", schema="test"
+                ),
             )
         ]
 
@@ -211,7 +232,12 @@ async def test_catalog_engine(
         ]
         assert preview_sql_table_list_results == [
             SQLTableListPreview(
-                request_id=RequestId("0"), tables=[], error=None
+                request_id=RequestId("0"),
+                tables=[],
+                error=None,
+                metadata=SQLMetadata(
+                    connection=DUCKDB_CONN, database="test", schema="test"
+                ),
             )
         ]
 
@@ -243,6 +269,9 @@ async def test_query_engine(
                 request_id=RequestId("0"),
                 tables=[],
                 error="Connection does not support catalog operations",
+                metadata=SQLMetadata(
+                    connection=SQLITE_CONN, database="test", schema="test"
+                ),
             )
         ]