aboutsummaryrefslogtreecommitdiff
path: root/tests/managers/thl/test_buyer.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/managers/thl/test_buyer.py')
-rw-r--r--tests/managers/thl/test_buyer.py11
1 files changed, 9 insertions, 2 deletions
diff --git a/tests/managers/thl/test_buyer.py b/tests/managers/thl/test_buyer.py
index 69ea105..6776ab3 100644
--- a/tests/managers/thl/test_buyer.py
+++ b/tests/managers/thl/test_buyer.py
@@ -1,3 +1,8 @@
+from __future__ import annotations
+
+from collections.abc import Callable
+
+from generalresearch.managers.thl.buyer import BuyerManager
from generalresearch.models import Source
@@ -5,10 +10,12 @@ class TestBuyer:
def test(
self,
- delete_buyers_surveys,
- buyer_manager,
+ delete_buyers_surveys: Callable[..., None],
+ buyer_manager: BuyerManager,
):
+ delete_buyers_surveys()
+
bs = buyer_manager.bulk_get_or_create(source=Source.TESTING, codes=["a", "b"])
assert len(bs) == 2
buyer_a = bs[0]