Passing Managers into flow tasks for better pytest usage. Conftests broken out into seperate fixture files. Extensive type hinting.

4 files changed, 64 insertions, 0 deletions

diff --git a/tests/models/__init__.py b/tests/models/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/models/__init__.py
diff --git a/tests/models/test_assignment.py b/tests/models/test_assignment.py
new file mode 100644
index 0000000..2a87364
--- /dev/null
+++ b/tests/models/test_assignment.py
@@ -0,0 +1,41 @@
+from jb.models.assignment import Assignment, AssignmentStub
+from mypy_boto3_mturk.type_defs import (
+    GetAssignmentResponseTypeDef,
+)
+
+
+class TestAssignmentStub:
+
+    def test_init(self, assignment_stub: AssignmentStub):
+        assert isinstance(assignment_stub, AssignmentStub)
+        assert assignment_stub.id is None
+
+    def test_record(self, assignment_stub_record: AssignmentStub):
+        assert isinstance(assignment_stub_record, AssignmentStub)
+        assert assignment_stub_record.id is not None
+
+
+class TestAssignment:
+
+    def test_get_assignment(self, assignment_record: Assignment):
+        assert isinstance(assignment_record, Assignment)
+        assert assignment_record.id is not None
+
+
+class TestAssignmentResponse:
+
+    def test_get_assignment_no_tsid(
+        self, assignment_response_no_tsid: GetAssignmentResponseTypeDef
+    ):
+        assignment = Assignment.from_amt_get_assignment(
+            assignment_response_no_tsid["Assignment"]
+        )
+        assert assignment.tsid is None
+
+    def test_get_assignment_bad_tsid(
+        self, assignment_response_bad_tsid: GetAssignmentResponseTypeDef
+    ):
+        assignment = Assignment.from_amt_get_assignment(
+            assignment_response_bad_tsid["Assignment"]
+        )
+        assert assignment.tsid is None
diff --git a/tests/models/test_event.py b/tests/models/test_event.py
new file mode 100644
index 0000000..0496574
--- /dev/null
+++ b/tests/models/test_event.py
@@ -0,0 +1,13 @@
+import pytest
+
+
+from jb.models.event import MTurkEvent
+
+
+class TestMTurkEvent:
+
+    @pytest.mark.anyio
+    def test_init(self, mturk_event: MTurkEvent):
+        assert isinstance(mturk_event, MTurkEvent)
+        assert isinstance(mturk_event.amt_assignment_id, str)
+        assert isinstance(mturk_event.amt_hit_type_id, str)
diff --git a/tests/models/test_hit.py b/tests/models/test_hit.py
new file mode 100644
index 0000000..3952068
--- /dev/null
+++ b/tests/models/test_hit.py
@@ -0,0 +1,10 @@
+import pytest
+from jb.models.hit import Hit
+
+
+class TestAssignment:
+
+    @pytest.mark.anyio
+    def test_get_assignment(self, hit_record: Hit):
+        assert isinstance(hit_record, Hit)
+        assert hit_record.id is not None