|
7 | 7 |
|
8 | 8 | import sqlmesh.core.dialect as d |
9 | 9 | from sqlmesh.core.dialect import normalize_model_name |
| 10 | +from sqlmesh.core.engine_adapter import SnowflakeEngineAdapter |
10 | 11 | from sqlmesh.core.engine_adapter.base import EngineAdapter |
| 12 | +from sqlmesh.core.engine_adapter.shared import DataObjectType |
11 | 13 | from sqlmesh.core.model import load_sql_based_model |
12 | | -from sqlmesh.core.engine_adapter import SnowflakeEngineAdapter |
13 | 14 | from sqlmesh.core.model.definition import SqlModel |
14 | 15 | from sqlmesh.core.node import IntervalUnit |
15 | 16 | from sqlmesh.utils.errors import SQLMeshError |
@@ -39,6 +40,38 @@ def test_get_temp_table(mocker: MockerFixture, make_mocked_engine_adapter: t.Cal |
39 | 40 | assert value.sql(dialect=adapter.dialect) == '"CATALOG"."DB"."__temp_TEST_TABLE_abcdefgh"' |
40 | 41 |
|
41 | 42 |
|
| 43 | +def test_get_data_objects_lowercases_columns( |
| 44 | + make_mocked_engine_adapter: t.Callable, mocker: MockerFixture |
| 45 | +) -> None: |
| 46 | + adapter = make_mocked_engine_adapter(SnowflakeEngineAdapter, patch_get_data_objects=False) |
| 47 | + |
| 48 | + adapter.get_current_catalog = mocker.Mock(return_value="TEST_CATALOG") |
| 49 | + |
| 50 | + adapter.fetchdf = mocker.Mock( |
| 51 | + return_value=pd.DataFrame( # type: ignore[assignment] |
| 52 | + [ |
| 53 | + { |
| 54 | + "CATALOG": "TEST_CATALOG", |
| 55 | + "NAME": "MY_TABLE", |
| 56 | + "SCHEMA_NAME": "PUBLIC", |
| 57 | + "TYPE": "TABLE", |
| 58 | + "CLUSTERING_KEY": "ID", |
| 59 | + } |
| 60 | + ] |
| 61 | + ) |
| 62 | + ) |
| 63 | + |
| 64 | + data_objects = adapter._get_data_objects("TEST_CATALOG.PUBLIC") |
| 65 | + |
| 66 | + assert len(data_objects) == 1 |
| 67 | + data_object = data_objects[0] |
| 68 | + assert data_object.catalog == "TEST_CATALOG" |
| 69 | + assert data_object.schema_name == "PUBLIC" |
| 70 | + assert data_object.name == "MY_TABLE" |
| 71 | + assert data_object.type == DataObjectType.TABLE |
| 72 | + assert data_object.clustering_key == "ID" |
| 73 | + |
| 74 | + |
42 | 75 | @pytest.mark.parametrize( |
43 | 76 | "current_warehouse, current_warehouse_exp, configured_warehouse, configured_warehouse_exp, should_change", |
44 | 77 | [ |
|
0 commit comments