def generate_court_case(person, **kwargs) -> schema.StateCourtCase:
args = {
"status": StateCourtCaseStatus.PRESENT_WITHOUT_INFO.value,
"state_code": _STATE_CODE,
}
args.update(kwargs)
return schema.StateCourtCase(person=person, **args)
def generate_test_court_case(person_id) -> state_schema.StateCourtCase:
instance = state_schema.StateCourtCase(
court_case_id=8888,
state_code='us_ca',
court_type=StateCourtType.PRESENT_WITHOUT_INFO.value,
court_type_raw_text=None,
person_id=person_id,
)
return instance
def test_addChildToEntity_singular(self) -> None:
charge = schema.StateCharge(
state_code=_STATE_CODE,
status=ChargeStatus.PRESENT_WITHOUT_INFO,
charge_id=_ID,
)
court_case = schema.StateCourtCase(state_code=_STATE_CODE,
court_case_id=_ID)
add_child_to_entity(entity=charge,
child_field_name="court_case",
child_to_add=court_case)
self.assertEqual(charge.court_case, court_case)