Fixed as per review comments

Canner · Nov 13, 2024 · 2416ed6 · 2416ed6
1 parent 4c51be0
commit 2416ed6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/ibis-server/app/model/metadata/snowflake.py b/ibis-server/app/model/metadata/snowflake.py
@@ -142,7 +142,7 @@ def _transform_column_type(self, data_type):
             "smallint": WrenEngineColumnType.SMALLINT,
             "tinyint": WrenEngineColumnType.TINYINT,
             "byteint": WrenEngineColumnType.TINYINT,
-            # Decimal
+            # Float
             "float4": WrenEngineColumnType.FLOAT4,
             "float": WrenEngineColumnType.FLOAT8,
             "float8": WrenEngineColumnType.FLOAT8,

diff --git a/ibis-server/tests/routers/v2/connector/test_snowflake.py b/ibis-server/tests/routers/v2/connector/test_snowflake.py
@@ -267,6 +267,27 @@ def test_metadata_list_tables():
             json={"connectionInfo": connection_info},
         )
         assert response.status_code == 200
+        tables = response.json()
+        assert len(tables) == 8
+        table = next(filter(lambda t: t["name"] == "TPCH_SF1.ORDERS", tables))
+        assert table["name"] == "TPCH_SF1.ORDERS"
+        assert table["primaryKey"] is not None
+        assert table["description"] == "Orders data as defined by TPC-H"
+        assert table["properties"] == {
+            "catalog": "SNOWFLAKE_SAMPLE_DATA",
+            "schema": "TPCH_SF1",
+            "table": "ORDERS",
+        }
+        assert len(table["columns"]) == 9
+        column = next(filter(lambda c: c["name"] == "O_COMMENT", table["columns"]))
+        assert column == {
+            "name": "O_COMMENT",
+            "nestedColumns": None,
+            "type": "TEXT",
+            "notNull": True,
+            "description": None,
+            "properties": None,
+        }
 
     def test_metadata_list_constraints():
         response = client.post(