add test cases and format code using ruff

ibis-server/app/model/metadata/bigquery.py

@@ -111,4 +111,3 @@ def get_constraints(self) -> list[Constraint]:
                 )
             )
         return constraints
-

ibis-server/app/model/metadata/dto.py

@@ -11,7 +11,7 @@ class MetadataDTO(BaseModel):
 class WrenEngineColumnType(Enum):
     # Boolean Types
     BOOLEAN = "BOOLEAN"
-    
+
     # Numeric Types
     TINYINT = "TINYINT"
     INT2 = "INT2"

ibis-server/app/model/metadata/metadata.py

@@ -2,12 +2,12 @@
 from app.model.metadata.dto import Table, Constraint
 
 
-class Metadata():
+class Metadata:
     def __init__(self, connection_info: ConnectionInfo):
         self.connection_info = connection_info
 
     def get_table_list(self) -> list[Table]:
         raise NotImplementedError
-    
+
     def get_constraints(self) -> list[Constraint]:
-        raise NotImplementedError
+        raise NotImplementedError

ibis-server/app/routers/ibis/postgres.py

@@ -40,11 +40,10 @@ def validate(rule_name: str, dto: ValidateDTO) -> Response:
 def get_postgres_table_list(dto: MetadataDTO) -> list[Table]:
     metadata = MetadataFactory(DataSource.postgres, dto.connection_info)
     return metadata.get_table_list()
-    
+
 
 @router.post("/metadata/constraints", response_model=list[Constraint])
 @log_dto
 def get_postgres_constraints(dto: MetadataDTO) -> list[Constraint]:
     metadata = MetadataFactory(DataSource.postgres, dto.connection_info)
     return metadata.get_constraints()
-

ibis-server/tests/routers/ibis/test_bigquery.py

@@ -307,3 +307,19 @@ def test_validate_rule_column_is_valid_without_one_parameter(self):
         )
         assert response.status_code == 422
         assert response.text == "Missing required parameter: `modelName`"
+
+    def test_metadata_list_tables(self):
+        connection_info = self.get_connection_info()
+        response = client.post(
+            url="/v2/ibis/bigquery/metadata/tables",
+            json={"connectionInfo": connection_info},
+        )
+        assert response.status_code == 200
+
+    def test_metadata_list_constraints(self):
+        connection_info = self.get_connection_info()
+        response = client.post(
+            url="/v2/ibis/bigquery/metadata/constraints",
+            json={"connectionInfo": connection_info},
+        )
+        assert response.status_code == 200

ibis-server/tests/routers/ibis/test_postgres.py

@@ -365,3 +365,19 @@ def test_validate_rule_column_is_valid_without_one_parameter(
         )
         assert response.status_code == 422
         assert response.text == "Missing required parameter: `modelName`"
+
+    def test_metadata_list_tables(self, postgres: PostgresContainer):
+        connection_info = self.to_connection_info(postgres)
+        response = client.post(
+            url="/v2/ibis/postgres/metadata/tables",
+            json={"connectionInfo": connection_info},
+        )
+        assert response.status_code == 200
+
+    def test_metadata_list_constraints(self, postgres: PostgresContainer):
+        connection_info = self.to_connection_info(postgres)
+        response = client.post(
+            url="/v2/ibis/postgres/metadata/constraints",
+            json={"connectionInfo": connection_info},
+        )
+        assert response.status_code == 200