from jb.managers.amt import AMTManager def test_create_hit_type(hit_type): 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(hqm, htm, hm, hit_type_with_amt_id, question): question = hqm.get_or_create(question) hit_type = hit_type_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) assert hit.amt_hit_id is not None assert hit.id is None hm.create(hit) assert hit.id is not None