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