refactor:update conn_clickhouse unit test

eosphoros-ai · Dec 21, 2023 · 8d8af98 · 8d8af98
1 parent bba2941
commit 8d8af98
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/intetration_tests/datasource/test_conn_clickhouse.py b/tests/intetration_tests/datasource/test_conn_clickhouse.py
@@ -16,6 +16,7 @@
     (102, 'Sort your data based on your commonly-used queries', today(),     2.718   ),
     (101, 'Granules are the smallest chunks of data read',      now() + 5,   3.14159 )
     """
+from typing import Dict, List
 
 import pytest
 
@@ -50,7 +51,9 @@ def test_get_table_names(db):
 
 
 def test_get_indexes(db):
-    assert db.get_indexes("") == ""
+    assert [index.get("name") for index in db.get_indexes("my_first_table")][
+        0
+    ] == "primary_key"
 
 
 def test_get_fields(db):