github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 20:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(flow): Add Chat Data template (#2716)

Browse Source
This commit is contained in:
Fangyin Cheng
2025-05-22 15:52:24 +08:00
committed by GitHub
parent 7e7581e891
commit f79f81ccc3
7 changed files with 3353 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
tests/intetration_tests/datasource/test_conn_oracle.py
View File

@@ -1,22 +1,23 @@
"""
Run unit test with command: pytest dbgpt/datasource/rdbms/tests/test_conn_mysql.py
docker run -itd --name mysql-test -p 3307:3306 -e MYSQL_ROOT_PASSWORD=12345678 mysql:5.7
mysql -h 127.0.0.1 -uroot -p -P3307
Enter password:
Welcome to the MySQL monitor. Commands end with ; or \g.
Your MySQL connection id is 2
Server version: 5.7.41 MySQL Community Server (GPL)
Run unit test with command: pytest dbgpt/datasource/rdbms/tests/test_conn_mysql.py
docker run -itd --name mysql-test -p 3307:3306 -e MYSQL_ROOT_PASSWORD=12345678 mysql:5.7
mysql -h 127.0.0.1 -uroot -p -P3307
Enter password:
Welcome to the MySQL monitor. Commands end with ; or \g.
Your MySQL connection id is 2
Server version: 5.7.41 MySQL Community Server (GPL)
Copyright (c) 2000, 2023, Oracle and/or its affiliates.
Copyright (c) 2000, 2023, Oracle and/or its affiliates.
Oracle is a registered trademark of Oracle Corporation and/or its
affiliates. Other names may be trademarks of their respective
owners.
Oracle is a registered trademark of Oracle Corporation and/or its
affiliates. Other names may be trademarks of their respective
owners.
Type 'help;' or '\h' for help. Type '\c' to clear the current input statement.
> create database test;
Type 'help;' or '\h' for help. Type '\c' to clear the current input statement.
> create database test;
"""
import pytest
from dbgpt_ext.datasource.rdbms.conn_oracle import OracleConnector
@@ -27,6 +28,7 @@ CREATE TABLE test (
)
"""
@pytest.fixture
def db():
# 注意：Oracle 默认端口是 1521，连接方式建议用 service_name
@@ -45,32 +47,37 @@ def db():
except Exception:
pass # 如果表不存在也忽略错误
def test_get_usable_table_names(db):
db.run(_create_table_sql)
db.run("COMMIT")
table_names = db.get_usable_table_names()
assert "TEST" in map(str.upper, table_names)
def test_get_table_info(db):
db.run(_create_table_sql)
db.run("COMMIT")
table_info = db.get_table_info()
assert "CREATE TABLE TEST" in table_info.upper()
def test_run_no_throw(db):
result = db.run_no_throw("this is a error sql")
# run_no_throw 返回的是 list，错误时为空
assert result == [] or isinstance(result, list)
def test_get_index_empty(db):
db.run(_create_table_sql)
db.run("COMMIT")
indexes = db.get_indexes("TEST")
assert indexes == []
def test_get_fields(db):
#db.run(_create_table_sql)
#db.run("COMMIT")
# db.run(_create_table_sql)
# db.run("COMMIT")
print("进入方法...")
fields = db.get_fields("PY_TEST")
print("正在打印字段信息...")
@@ -81,16 +88,21 @@ def test_get_fields(db):
print(f"Is Nullable: {field[3]}")
print(f"Column Comment: {field[4]}")
print("-" * 30) # 可选的分隔符
#assert fields[0][0].upper() == "ID"
# assert fields[0][0].upper() == "ID"
def test_get_charset(db):
result = db.run("SELECT VALUE FROM NLS_DATABASE_PARAMETERS WHERE PARAMETER = 'NLS_CHARACTERSET'")
result = db.run(
"SELECT VALUE FROM NLS_DATABASE_PARAMETERS WHERE PARAMETER = 'NLS_CHARACTERSET'"
)
assert result[1][0] in ("AL32UTF8", "UTF8") # result[0] 是字段名元组
def test_get_users(db):
users = db.get_users()
assert any(user[0].upper() in ("SYS", "SYSTEM") for user in users)
def test_get_database_lists(db):
cdb_result = db.run("SELECT CDB FROM V$DATABASE")
if cdb_result[1][0] == "YES":
@@ -98,4 +110,4 @@ def test_get_database_lists(db):
pdb_names = [name[0] for name in databases[1:]]
else:
pdb_names = ["ORCL"]
assert any(name in ("ORCLPDB1", "ORCL") for name in pdb_names)
assert any(name in ("ORCLPDB1", "ORCL") for name in pdb_names)