github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-04 18:40:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Refactor datasource module (#1309)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-18 18:06:40 +08:00
committed by GitHub
parent 84bedee306
commit 4970c9f813
108 changed files with 1194 additions and 1066 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/intetration_tests/datasource/test_conn_mysql.py
View File

@@ -20,7 +20,7 @@
import pytest
from dbgpt.datasource.rdbms.conn_mysql import MySQLConnect
from dbgpt.datasource.rdbms.conn_mysql import MySQLConnector
_create_table_sql = """
CREATE TABLE IF NOT EXISTS `test` (
@@ -31,7 +31,7 @@ _create_table_sql = """
@pytest.fixture
def db():
conn = MySQLConnect.from_uri_db(
conn = MySQLConnector.from_uri_db(
"localhost",
3307,
"root",
@@ -89,4 +89,4 @@ def test_get_users(db):
def test_get_database_lists(db):
assert db.get_database_list() == ["test"]
assert db.get_database_names() == ["test"]