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]
])
