From 361b4a2f591a3df68d8f97c8011c7cc790d5c8d2 Mon Sep 17 00:00:00 2001 From: dzaslavskiy Date: Tue, 31 May 2022 09:59:12 -0400 Subject: [PATCH] fix import in conftest (#9) --- tests/conftest.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/tests/conftest.py b/tests/conftest.py index a6599e2..bf3e014 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -6,21 +6,19 @@ from fastapi import testclient import db -from db import SessionLocal # pylint: disable=wrong-import-position sys.modules["gpo.database"] = db -from gpo.main import app -from gpo.models import Letter +from gpo import main, models @pytest.fixture def generate_data(): """test fixture""" - def get_letter() -> Letter: + def get_letter() -> models.Letter: """get a letter object""" - return Letter( + return models.Letter( id=5, name="Name", address="Address", @@ -41,11 +39,11 @@ def get_letter() -> Letter: @pytest.fixture(scope="session") def session() -> typing.Generator: """session""" - yield SessionLocal() + yield db.SessionLocal() @pytest.fixture(scope="module") def client() -> typing.Generator: """api test client""" - with testclient.TestClient(app) as test_client: + with testclient.TestClient(main.app) as test_client: yield test_client