def test_raises_commit_cmd_none_error(self): # type: () -> None invalid_commit_cmd_model = (story_models.StoryCommitLogEntryModel( id='model_id123', created_on=self.YEAR_AGO, last_updated=self.NOW, story_id='story-id', user_id='user-id', commit_type='test-type', post_commit_status='private', commit_cmds=[{ 'cmd': base_models.VersionedModel.CMD_DELETE_COMMIT }])) output = (self.pipeline | beam.Create([invalid_commit_cmd_model]) | beam.ParDo( story_validation.ValidateStoryCommitLogEntryModel())) self.assert_pcoll_equal( # type: ignore[no-untyped-call] output, [ base_validation_errors. CommitCmdsNoneError( # type: ignore[no-untyped-call] invalid_commit_cmd_model) ])
def test_validate_story_model(self) -> None: valid_commit_cmd_model = (story_models.StoryCommitLogEntryModel( id='story_id123', created_on=self.YEAR_AGO, last_updated=self.NOW, story_id='story-id', user_id='user-id', commit_type='test-type', post_commit_status='private', commit_cmds=[{ 'cmd': base_models.VersionedModel.CMD_DELETE_COMMIT }])) output = (self.pipeline | beam.Create([valid_commit_cmd_model]) | beam.ParDo( story_validation.ValidateStoryCommitLogEntryModel())) self.assert_pcoll_equal(output, [])