python/fastapi-users
1
0
Fork 0
mirror of https://github.com/fastapi-users/fastapi-users.git synced 2025-11-01 01:48:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use a Base table class for SQLAlchemy adapter

Browse Source
This commit is contained in:
François Voron
2019-10-09 18:03:10 +02:00
parent f2bd2c6485
commit 9f41a8b9a7
2 changed files with 20 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_db_sqlalchemy.py
View File

@ -3,12 +3,18 @@ import sqlite3
import pytest
import sqlalchemy
from databases import Database
from sqlalchemy.ext.declarative import declarative_base
from fastapi_users.db.sqlalchemy import Base, SQLAlchemyUserDatabase
from fastapi_users.db.sqlalchemy import BaseUser, SQLAlchemyUserDatabase
@pytest.fixture
async def sqlalchemy_user_db() -> SQLAlchemyUserDatabase:
Base = declarative_base()
class User(BaseUser, Base):
pass
DATABASE_URL = "sqlite:///./test.db"
database = Database(DATABASE_URL)
@ -19,7 +25,7 @@ async def sqlalchemy_user_db() -> SQLAlchemyUserDatabase:
await database.connect()
yield SQLAlchemyUserDatabase(database)
yield SQLAlchemyUserDatabase(database, User.__table__)
Base.metadata.drop_all(engine)