Update tests

Author: tmichaeldb

tests/test_sdk.py

@@ -237,11 +237,25 @@ def test_flow(self, mock_post, mock_put, mock_get):
         assert call_args[0][0] == 'https://cloud.mindsdb.com/api/status'
 
         # --------- databases -------------
-        response_mock(mock_post, pd.DataFrame([{'NAME': 'db1','ENGINE': 'postgres'}]))
+        response_mock(
+            mock_post,
+            pd.DataFrame(
+                [
+                    {
+                        "NAME": "db1",
+                        "ENGINE": "postgres",
+                        "CONNECTION_DATA": {"host": "zoop"},
+                    }
+                ]
+            ),
+        )
 
         databases = server.list_databases()
 
-        check_sql_call(mock_post, "select NAME, ENGINE from information_schema.databases where TYPE='data'")
+        check_sql_call(
+            mock_post,
+            "select NAME, ENGINE, CONNECTION_DATA from information_schema.databases where TYPE='data'",
+        )
 
         database = databases[0]
         str(database)
@@ -465,7 +479,6 @@ def check_project_models(self, project, database, mock_post):
                 }
             )
 
-
     @patch('requests.Session.post')
     def check_project_models_versions(self, project, database, mock_post):
         # -----------  model version --------------
@@ -495,7 +508,6 @@ def check_project_models_versions(self, project, database, mock_post):
         project.drop_model_version('m1', 1)
         check_sql_call(mock_post, f"DROP PREDICTOR m1.`1`")
 
-
     @patch('requests.Session.post')
     def check_database(self, database, mock_post):
 
@@ -545,7 +557,6 @@ def check_database(self, database, mock_post):
         database.drop_table('t3')
         check_sql_call(mock_post, f'drop table {database.name}.t3')
 
-
     @patch('requests.Session.post')
     def check_project_jobs(self, project, mock_post):
 
@@ -1415,7 +1426,6 @@ def test_create(self, mock_get, mock_post):
 
         assert new_agent == expected_agent
 
-
     @patch('requests.Session.get')
     @patch('requests.Session.put')
     # Mock creating new skills.
@@ -1480,7 +1490,6 @@ def test_update(self, mock_get, mock_put, _):
 
         assert updated_agent == expected_agent
 
-
     @patch('requests.Session.post')
     def test_completion(self, mock_post):
         response_mock(mock_post, {
@@ -1693,18 +1702,29 @@ def test_add_database(self, mock_post, mock_put, mock_get):
                 'provider': 'mindsdb'
             },
         ])
-        responses_mock(mock_post, [
-            # DB get (POST /sql).
-            pd.DataFrame([
-                {'NAME': 'existing_db', 'ENGINE': 'postgres'}
-            ]),
-            # DB tables get (POST /sql).
-            pd.DataFrame([
-                {'name': 'existing_table'}
-            ]),
-            # Skill creation.
-            {'name': 'new_skill', 'type': 'sql', 'params': {'database': 'existing_db', 'tables': ['existing_table']}}
-        ])
+        responses_mock(
+            mock_post,
+            [
+                # DB get (POST /sql).
+                pd.DataFrame(
+                    [
+                        {
+                            "NAME": "existing_db",
+                            "ENGINE": "postgres",
+                            "CONNECTION_DATA": {"host": "boop"},
+                        }
+                    ]
+                ),
+                # DB tables get (POST /sql).
+                pd.DataFrame([{"name": "existing_table"}]),
+                # Skill creation.
+                {
+                    "name": "new_skill",
+                    "type": "sql",
+                    "params": {"database": "existing_db", "tables": ["existing_table"]},
+                },
+            ],
+        )
         responses_mock(mock_put, [
             # Agent update with new skill.
             {