tests: clean up tests (#243)

opendatadiscovery · Oct 25, 2023 · ccad4cc · ccad4cc
1 parent b0a9b19
commit ccad4cc
Show file tree

Hide file tree

Showing 10 changed files with 5 additions and 285 deletions.
diff --git a/odd_collector/adapters/databricks/mappers/models.py b/odd_collector/adapters/databricks/mappers/models.py
@@ -6,7 +6,7 @@
 
 @dataclass(frozen=True)
 class DatabricksColumn(Column):
-    odd_metadata: dict[str, Any] = field(default_factory=dict)
+    ...
 
 
 @dataclass()

diff --git a/odd_collector/adapters/databricks/mappers/table.py b/odd_collector/adapters/databricks/mappers/table.py
@@ -34,7 +34,7 @@ def get_column(table_info: dict, column_raw: dict) -> DatabricksColumn:
         name=column_raw.pop("name"),
         type=column_raw.get("type_text"),
         is_nullable=column_raw.get("nullable"),
-        odd_metadata=column_raw,
+        metadata=column_raw,
     )
     return column
 

diff --git a/odd_collector/adapters/superset/tests/__init__.py b/odd_collector/adapters/superset/tests/__init__.py
diff --git a/odd_collector/adapters/superset/tests/raw_data.py b/odd_collector/adapters/superset/tests/raw_data.py
diff --git a/odd_collector/adapters/superset/tests/test_map_column.py b/odd_collector/adapters/superset/tests/test_map_column.py
diff --git a/odd_collector/adapters/superset/tests/test_map_dashboard.py b/odd_collector/adapters/superset/tests/test_map_dashboard.py
diff --git a/odd_collector/adapters/superset/tests/test_map_database.py b/odd_collector/adapters/superset/tests/test_map_database.py
diff --git a/odd_collector/adapters/superset/tests/test_map_table.py b/odd_collector/adapters/superset/tests/test_map_table.py
diff --git a/tests/adapters/mode/mock_rest_client.py b/tests/adapters/mode/mock_rest_client.py
@@ -487,7 +487,7 @@ def __init__(self):
             },
         }
 
-    async def fetch_async_response(self, session, request_args: RequestArgs):
+    async def fetch(self, session, request_args: RequestArgs):
         response = None
         if "/api/account/data_sources" in request_args.url:
             response = self.data_sources

diff --git a/tests/integration/test_postgres.py b/tests/integration/test_postgres.py
@@ -107,8 +107,8 @@ def test_data_base_service(data_entity_list: odd_models.DataEntityList):
     other_schema = find_by_name(data_entity_list, "other_schema")
 
     assert database is not None
-    public_schema is not None
-    other_schema is not None
+    assert public_schema is not None
+    assert other_schema is not None
 
     assert database.data_entity_group is not None
     assert len(database.data_entity_group.entities_list) == 2