def testCompletionDateInFuture(self):
# Arrange
ingest_sentence = ingest_info_pb2.Sentence(
sentence_id="SENTENCE_ID",
completion_date="2020-1-1",
)
# Act
sentence_builder = entities.Sentence.builder()
sentence.copy_fields_to_builder(
sentence_builder,
ingest_sentence,
IngestMetadata.new_with_defaults(
ingest_time=datetime.datetime(year=2018, month=6, day=8)
),
)
result = sentence_builder.build()
# Assert
expected_result = entities.Sentence.new_with_defaults(
external_id="SENTENCE_ID",
completion_date=None,
projected_completion_date=datetime.date(year=2020, month=1, day=1),
status=SentenceStatus.PRESENT_WITHOUT_INFO,
)
self.assertEqual(result, expected_result)
def testParseSentence_completionDateWithoutStatus_marksAsCompleted(self):
# Arrange
ingest_sentence = ingest_info_pb2.Sentence(
sentence_id="SENTENCE_ID",
completion_date="2018-1-1",
)
# Act
sentence_builder = entities.Sentence.builder()
sentence.copy_fields_to_builder(
sentence_builder,
ingest_sentence,
IngestMetadata.new_with_defaults(
ingest_time=datetime.datetime(year=2018, month=6, day=8)
),
)
result = sentence_builder.build()
# Assert
expected_result = entities.Sentence.new_with_defaults(
external_id="SENTENCE_ID",
completion_date=datetime.date(year=2018, month=1, day=1),
status=SentenceStatus.COMPLETED,
)
self.assertEqual(result, expected_result)
def testParseSentence(self):
# Arrange
ingest_sentence = ingest_info_pb2.Sentence(
sentence_id="SENTENCE_ID",
min_length="1",
date_imposed="2000-1-1",
post_release_supervision_length="",
)
# Act
sentence_builder = entities.Sentence.builder()
sentence.copy_fields_to_builder(
sentence_builder,
ingest_sentence,
IngestMetadata.new_with_defaults(
ingest_time=datetime.datetime(year=2018, month=6, day=8)
),
)
result = sentence_builder.build()
# Assert
expected_result = entities.Sentence.new_with_defaults(
external_id="SENTENCE_ID",
min_length_days=1,
post_release_supervision_length_days=0,
date_imposed=datetime.date(year=2000, month=1, day=1),
status=SentenceStatus.PRESENT_WITHOUT_INFO,
)
self.assertEqual(result, expected_result)
def _convert_sentence(self, ingest_sentence):
"""Converts an ingest_info proto Sentence to a persistence entity."""
sentence_builder = entities.Sentence.builder()
sentence.copy_fields_to_builder(sentence_builder, ingest_sentence,
self.metadata)
return sentence_builder.build()
