def test_validate_skill_model(self) -> None: valid_commit_cmd_model = skill_models.SkillCommitLogEntryModel( id='skill_id123', created_on=self.YEAR_AGO, last_updated=self.NOW, skill_id='skill-id', user_id='user-id', commit_type='test-type', post_commit_status='private', commit_cmds=[{ 'cmd': 'create_new' }]) output = (self.pipeline | beam.Create([valid_commit_cmd_model]) | beam.ParDo( skill_validation.ValidateSkillCommitLogEntryModel())) self.assert_pcoll_equal(output, [])
def test_raises_commit_cmd_none_error(self): # type: () -> None invalid_commit_cmd_model = skill_models.SkillCommitLogEntryModel( id='model_id123', created_on=self.YEAR_AGO, last_updated=self.NOW, skill_id='skill-id', user_id='user-id', commit_type='test-type', post_commit_status='private', commit_cmds=[{'cmd': 'create_new'}]) output = ( self.pipeline | beam.Create([invalid_commit_cmd_model]) | beam.ParDo( skill_validation.ValidateSkillCommitLogEntryModel()) ) self.assert_pcoll_equal( # type: ignore[no-untyped-call] output, [ base_validation_errors.CommitCmdsNoneError(invalid_commit_cmd_model) # type: ignore[no-untyped-call] ])