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()