def testParseStateAssessment(self):
# Arrange
ingest_assessment = ingest_info_pb2.StateAssessment(
assessment_class='RISK',
assessment_type='LSIR',
state_assessment_id='ASSESSMENT_ID',
assessment_date='1/2/2111',
state_code='US_ND',
assessment_score='17',
assessment_level='MEDIUM',
assessment_metadata='{"high_score_domains": ["a", "c", "q"]}',
)
# Act
assessment_builder = entities.StateAssessment.builder()
state_assessment.copy_fields_to_builder(
assessment_builder, ingest_assessment, _EMPTY_METADATA)
result = assessment_builder.build()
# Assert
expected_result = entities.StateAssessment.new_with_defaults(
assessment_class=StateAssessmentClass.RISK,
assessment_class_raw_text='RISK',
assessment_type=StateAssessmentType.LSIR,
assessment_type_raw_text='LSIR',
assessment_level=StateAssessmentLevel.MEDIUM,
assessment_level_raw_text='MEDIUM',
external_id='ASSESSMENT_ID',
assessment_date=date(year=2111, month=1, day=2),
state_code='US_ND',
assessment_score=17,
assessment_metadata='{"HIGH_SCORE_DOMAINS": ["A", "C", "Q"]}',
)
self.assertEqual(result, expected_result)
def _convert_assessment(self, ingest_assessment: StateAssessment) \
-> entities.StateAssessment:
"""Converts an ingest_info proto StateAssessment to a
persistence entity."""
assessment_builder = entities.StateAssessment.builder()
state_assessment.copy_fields_to_builder(assessment_builder,
ingest_assessment,
self.metadata)
assessment_builder.conducting_agent = \
fn(lambda i: state_agent.convert(self.agents[i], self.metadata),
'conducting_agent_id',
ingest_assessment)
return assessment_builder.build()
