AMTManager moved to fixture, and dectorator with parameters on tasks and init / non-static class appraoch. More assertion checks and typing. TestMTurkClient seperated from TestAMTManger

1 files changed, 10 insertions, 0 deletions

diff --git a/tests/fixtures/managers.py b/tests/fixtures/managers.py
index 6bc2e51..a3187d7 100644
--- a/tests/fixtures/managers.py
+++ b/tests/fixtures/managers.py
@@ -2,14 +2,24 @@ from typing import TYPE_CHECKING
 import pytest
 from jb.managers import Permission
 from generalresearchutils.pg_helper import PostgresConfig
+from mypy_boto3_mturk import MTurkClient
 
 if TYPE_CHECKING:
     from jb.managers.hit import HitQuestionManager, HitTypeManager, HitManager
     from jb.managers.assignment import AssignmentManager
     from jb.managers.bonus import BonusManager
+    from jb.managers.amt import AMTManager
 
 
 # --- Managers ---
+@pytest.fixture(scope="session")
+def amtm(amt_client: MTurkClient) -> "AMTManager":
+    from jb.managers.amt import AMTManager
+
+    amtm = AMTManager(amt_client=amt_client)
+    assert "mturk-requester-sandbox" in amtm.amt_client.meta.endpoint_url
+
+    return amtm
 
 
 @pytest.fixture(scope="session")