def test_get_entity_name_county_entity_sample(self): self.assertEqual('person', Person.new_with_defaults().get_entity_name()) self.assertEqual('booking', Booking.new_with_defaults().get_entity_name()) self.assertEqual('hold', Hold.new_with_defaults().get_entity_name())
def testConvert_FullIngestInfo_GeneratedIds(self): # Arrange metadata = IngestMetadata('REGION', _JURISDICTION_ID, _INGEST_TIME) ingest_info = IngestInfo() ingest_info.people.add(person_id='PERSON_ID_GENERATE', booking_ids=['BOOKING_ID_GENERATE']) ingest_info.bookings.add( booking_id='BOOKING_ID_GENERATE', arrest_id='ARREST_ID_GENERATE', hold_ids=['HOLD_ID_1_GENERATE', 'HOLD_ID_2_GENERATE'], charge_ids=['CHARGE_ID_GENERATE']) ingest_info.holds.add(hold_id='HOLD_ID_1_GENERATE', jurisdiction_name='jurisdiction') ingest_info.holds.add(hold_id='HOLD_ID_2_GENERATE', jurisdiction_name='jurisdiction') ingest_info.arrests.add(arrest_id='ARREST_ID_GENERATE', agency='PD') ingest_info.charges.add(charge_id='CHARGE_ID_GENERATE', name='DUI', bond_id='BOND_ID_GENERATE', sentence_id='SENTENCE_ID_GENERATE') ingest_info.bonds.add(bond_id='BOND_ID_GENERATE') ingest_info.sentences.add(sentence_id='SENTENCE_ID_GENERATE', is_life='True') result = self._convert_and_throw_on_errors(ingest_info, metadata) # Assert expected_result = [ Person.new_with_defaults( region='REGION', jurisdiction_id='JURISDICTION_ID', bookings=[ Booking.new_with_defaults( admission_date=_INGEST_TIME.date(), admission_date_inferred=True, first_seen_time=_INGEST_TIME, last_seen_time=_INGEST_TIME, custody_status=CustodyStatus.PRESENT_WITHOUT_INFO, arrest=Arrest.new_with_defaults(agency='PD'), holds=[ Hold.new_with_defaults( jurisdiction_name='JURISDICTION', status=HoldStatus.PRESENT_WITHOUT_INFO) ], charges=[ Charge.new_with_defaults( status=ChargeStatus.PRESENT_WITHOUT_INFO, name='DUI', bond=Bond.new_with_defaults( status=BondStatus.PRESENT_WITHOUT_INFO), sentence=Sentence.new_with_defaults( status=SentenceStatus.PRESENT_WITHOUT_INFO, is_life=True)) ]) ]) ] self.assertEqual(result, expected_result)
def test_get_id_county_entity_sample(self): self.assertEqual(123, Person.new_with_defaults(person_id=123).get_id()) self.assertEqual(456, Booking.new_with_defaults(booking_id=456).get_id()) self.assertEqual(789, Hold.new_with_defaults(hold_id=789).get_id()) self.assertIsNone(Charge.new_with_defaults().get_id())
def testConvert_FullIngestInfo_GeneratedIds(self): # Arrange metadata = IngestMetadata("REGION", _JURISDICTION_ID, _INGEST_TIME) ingest_info = IngestInfo() ingest_info.people.add( person_id="PERSON_ID_GENERATE", booking_ids=["BOOKING_ID_GENERATE"] ) ingest_info.bookings.add( booking_id="BOOKING_ID_GENERATE", arrest_id="ARREST_ID_GENERATE", hold_ids=["HOLD_ID_1_GENERATE", "HOLD_ID_2_GENERATE"], charge_ids=["CHARGE_ID_GENERATE"], ) ingest_info.holds.add( hold_id="HOLD_ID_1_GENERATE", jurisdiction_name="jurisdiction" ) ingest_info.holds.add( hold_id="HOLD_ID_2_GENERATE", jurisdiction_name="jurisdiction" ) ingest_info.arrests.add(arrest_id="ARREST_ID_GENERATE", agency="PD") ingest_info.charges.add( charge_id="CHARGE_ID_GENERATE", name="DUI", bond_id="BOND_ID_GENERATE", sentence_id="SENTENCE_ID_GENERATE", ) ingest_info.bonds.add(bond_id="BOND_ID_GENERATE") ingest_info.sentences.add(sentence_id="SENTENCE_ID_GENERATE", is_life="True") result = self._convert_and_throw_on_errors(ingest_info, metadata) # Assert expected_result = [ Person.new_with_defaults( region="REGION", jurisdiction_id="JURISDICTION_ID", bookings=[ Booking.new_with_defaults( admission_date=_INGEST_TIME.date(), admission_date_inferred=True, first_seen_time=_INGEST_TIME, last_seen_time=_INGEST_TIME, custody_status=CustodyStatus.PRESENT_WITHOUT_INFO, arrest=Arrest.new_with_defaults(agency="PD"), holds=[ Hold.new_with_defaults( jurisdiction_name="JURISDICTION", status=HoldStatus.PRESENT_WITHOUT_INFO, ) ], charges=[ Charge.new_with_defaults( status=ChargeStatus.PRESENT_WITHOUT_INFO, name="DUI", bond=Bond.new_with_defaults( status=BondStatus.PRESENT_WITHOUT_INFO ), sentence=Sentence.new_with_defaults( status=SentenceStatus.PRESENT_WITHOUT_INFO, is_life=True, ), ) ], ) ], ) ] self.assertEqual(result, expected_result)
def test_get_entity_name_county_entity_sample(self) -> None: self.assertEqual("person", Person.new_with_defaults().get_entity_name()) self.assertEqual("booking", Booking.new_with_defaults().get_entity_name()) self.assertEqual("hold", Hold.new_with_defaults().get_entity_name())