github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-18 07:30:40 +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

4
tests/intetration_tests/datasource/test_conn_clickhouse.py
View File

@@ -20,12 +20,12 @@ from typing import Dict, List
import pytest
from dbgpt.datasource.rdbms.conn_clickhouse import ClickhouseConnect
from dbgpt.datasource.rdbms.conn_clickhouse import ClickhouseConnector
@pytest.fixture
def db():
conn = ClickhouseConnect.from_uri_db("localhost", 8123, "default", "", "default")
conn = ClickhouseConnector.from_uri_db("localhost", 8123, "default", "", "default")
yield conn

4
tests/intetration_tests/datasource/test_conn_doris.py
View File

@@ -4,10 +4,10 @@
import pytest
from dbgpt.datasource.rdbms.conn_doris import DorisConnect
from dbgpt.datasource.rdbms.conn_doris import DorisConnector
@pytest.fixture
def db():
conn = DorisConnect.from_uri_db("localhost", 9030, "root", "", "test")
conn = DorisConnector.from_uri_db("localhost", 9030, "root", "", "test")
yield conn

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"]

4
tests/intetration_tests/datasource/test_conn_starrocks.py
View File

@@ -21,12 +21,12 @@
import pytest
from dbgpt.datasource.rdbms.conn_starrocks import StarRocksConnect
from dbgpt.datasource.rdbms.conn_starrocks import StarRocksConnector
@pytest.fixture
def db():
conn = StarRocksConnect.from_uri_db("localhost", 9030, "root", "", "test")
conn = StarRocksConnector.from_uri_db("localhost", 9030, "root", "", "test")
yield conn