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

Rename DB classes

Browse Source
This commit is contained in:
François Voron
2019-10-07 17:34:12 +02:00
parent 51b67b0739
commit 20aa806375
6 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/conftest.py
View File

@ -1,6 +1,6 @@
import pytest
from fastapi_users.db import UserDBInterface
from fastapi_users.db import BaseUserDatabase
from fastapi_users.models import UserDB
from fastapi_users.password import get_password_hash
@ -21,7 +21,7 @@ def user() -> UserDB:
return active_user
class MockUserDBInterface(UserDBInterface):
class MockUserDBInterface(BaseUserDatabase):
async def get_by_email(self, email: str) -> UserDB:
if email == active_user.email:

0
tests/test_db_interface.py → tests/test_db_base.py
View File

6
tests/test_db_sqlalchemy.py
View File

@ -4,11 +4,11 @@ import pytest
import sqlalchemy
from databases import Database
from fastapi_users.db.sqlalchemy import Base, SQLAlchemyUserDB
from fastapi_users.db.sqlalchemy import Base, SQLAlchemyUserDatabase
@pytest.fixture
async def sqlalchemy_user_db() -> SQLAlchemyUserDB:
async def sqlalchemy_user_db() -> SQLAlchemyUserDatabase:
DATABASE_URL = 'sqlite:///./test.db'
database = Database(DATABASE_URL)
@ -19,7 +19,7 @@ async def sqlalchemy_user_db() -> SQLAlchemyUserDB:
await database.connect()
yield SQLAlchemyUserDB(database)
yield SQLAlchemyUserDatabase(database)
Base.metadata.drop_all(engine)