1 files changed, 41 insertions, 0 deletions
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