from jb.managers.amt import AMTManager
from jb.models.hit import HitType, HitQuestion

from jb.managers.hit import HitQuestionManager, HitTypeManager, HitManager


class TestAMTManager:

    def test_create_hit_type(self, hit_type: HitType):
        assert hit_type.amt_hit_type_id is None
        AMTManager.create_hit_type(hit_type=hit_type)
        assert hit_type.amt_hit_type_id is not None

    def test_create_hit_with_hit_type(
        self,
        hqm: HitQuestionManager,
        htm: HitTypeManager,
        hm: HitManager,
        hit_type_record_with_amt_id: HitType,
        question_record: HitQuestion,
    ):

        hit_type = hit_type_record_with_amt_id
        hit_type = [
            x
            for x in htm.filter_active()
            if x.amt_hit_type_id == hit_type.amt_hit_type_id
        ][0]

        hit = AMTManager.create_hit_with_hit_type(
            hit_type=hit_type, question=question_record
        )
        assert hit.amt_hit_id is not None
        assert hit.id is None
        hm.create(hit)
        assert hit.id is not None