Skip to content

Commit a18e782

Browse files
committed
fix errors
1 parent a60d085 commit a18e782

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

src/qpydao/database_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
import sqlalchemy
88
from qpyconf import settings
9-
from sqlalchemy import Row, RowMapping, text
9+
from sqlalchemy import Row, RowMapping, text,MetaData
1010
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
1111
from sqlmodel import Session, SQLModel, select
1212

@@ -251,7 +251,7 @@ def get_db(self, db_name: str = "default") -> DatabaseClient:
251251
databases: Databases = Databases()
252252
db = databases.default_client()
253253

254-
def init_database(database: DatabaseClient, schema_name: str = "")->MetaData:
254+
def init_database(database: DatabaseClient, schema_name: str = ""):
255255
"""Init postgresql database
256256
:param database:
257257
:param schema_name:

tests/test_database_client.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,10 @@
33
Databases,
44
database_config,
55
databases,
6-
db
6+
db,
7+
init_database
78
)
9+
from qpydao.database_client import init_database
810

911
config = database_config()
1012
client = DatabaseClient(config=config)
@@ -29,3 +31,6 @@ def test_default_client():
2931
def test_database_singleton():
3032
db = Databases()
3133
assert databases == db
34+
35+
def test_init_database():
36+
init_database(client)

0 commit comments

Comments
 (0)