def generate_supervision_case_type_entry(
person, **kwargs) -> schema.StateSupervisionCaseTypeEntry:
args = {
"state_code": _STATE_CODE,
}
args.update(kwargs)
return schema.StateSupervisionCaseTypeEntry(person=person, **args)
def generate_test_supervision_case_type(person_id) \
-> state_schema.StateSupervisionCaseTypeEntry:
instance = state_schema.StateSupervisionCaseTypeEntry(
person_id=person_id,
supervision_case_type_entry_id=12345,
state_code='us_ca',
case_type=StateSupervisionCaseType.DOMESTIC_VIOLENCE.value,
case_type_raw_text='DV')
return instance