aboutsummaryrefslogtreecommitdiff
path: root/tests/models/test_assignment.py
blob: 2a87364cd41e402ae10ddbadcf167b57b61df6ac (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
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