diff options
| -rw-r--r-- | test_utils/models/conftest.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/test_utils/models/conftest.py b/test_utils/models/conftest.py index 468bea2..64bdec6 100644 --- a/test_utils/models/conftest.py +++ b/test_utils/models/conftest.py @@ -590,7 +590,7 @@ def ip_record_factory( @pytest.fixture(scope="session") -def buyer(buyer_manager: "BuyerManager") -> Buyer: +def buyer(buyer_manager: "BuyerManager") -> "Buyer": buyer_code = uuid4().hex buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=[buyer_code]) b = Buyer( @@ -601,7 +601,7 @@ def buyer(buyer_manager: "BuyerManager") -> Buyer: @pytest.fixture(scope="session") -def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., Buyer]: +def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., "Buyer"]: def _inner() -> Buyer: return buyer_manager.bulk_get_or_create( @@ -612,7 +612,7 @@ def buyer_factory(buyer_manager: "BuyerManager") -> Callable[..., Buyer]: @pytest.fixture(scope="session") -def survey(survey_manager: "SurveyManager", buyer: "Buyer") -> Survey: +def survey(survey_manager: "SurveyManager", buyer: "Buyer") -> "Survey": s = Survey(source=Source.TESTING, survey_id=uuid4().hex, buyer_code=buyer.code) survey_manager.create_bulk([s]) return s |