change db in inheritance tests to one in model def file

This commit is contained in:
collerek
2020-12-14 21:07:16 +01:00
parent fed54a7a71
commit 5f488fe2ca
2 changed files with 2 additions and 6 deletions

View File

@ -1,16 +1,14 @@
import datetime import datetime
import databases
import pytest import pytest
import sqlalchemy import sqlalchemy
from fastapi import FastAPI from fastapi import FastAPI
from starlette.testclient import TestClient from starlette.testclient import TestClient
from tests.settings import DATABASE_URL from tests.settings import DATABASE_URL
from tests.test_inheritance_concrete import Category, Subject, metadata # type: ignore from tests.test_inheritance_concrete import Category, Subject, metadata, db as database # type: ignore
app = FastAPI() app = FastAPI()
database = databases.Database(DATABASE_URL, force_rollback=True)
app.state.database = database app.state.database = database

View File

@ -1,16 +1,14 @@
import datetime import datetime
import databases
import pytest import pytest
import sqlalchemy import sqlalchemy
from fastapi import FastAPI from fastapi import FastAPI
from starlette.testclient import TestClient from starlette.testclient import TestClient
from tests.settings import DATABASE_URL from tests.settings import DATABASE_URL
from tests.test_inheritance_mixins import Category, Subject, metadata # type: ignore from tests.test_inheritance_mixins import Category, Subject, metadata, db as database # type: ignore
app = FastAPI() app = FastAPI()
database = databases.Database(DATABASE_URL, force_rollback=True)
app.state.database = database app.state.database = database