def test_compute_summary(self):
story_summary = story_services.compute_summary_of_story(self.story)
self.assertEqual(story_summary.id, self.STORY_ID)
self.assertEqual(story_summary.title, 'Title')
self.assertEqual(story_summary.description, 'Description')
self.assertEqual(story_summary.node_count, 1)
def _update_story_summary(
migrated_story: story_domain.Story,
story_summary_model: story_models.StorySummaryModel
) -> story_models.StorySummaryModel:
"""Generates newly updated story summary model.
Args:
migrated_story: Story. The migrated story domain object.
story_summary_model: StorySummaryModel. The story summary model
to update.
Returns:
StorySummaryModel. The updated story summary model to put into
the datastore.
"""
story_summary = story_services.compute_summary_of_story(migrated_story)
story_summary.version += 1
updated_story_summary_model = (
story_services.populate_story_summary_model_fields(
story_summary_model, story_summary))
return updated_story_summary_model
