def test_validate_post_commit_is_public_when_status_is_public(self):
invalid_commit_status = base_models.BaseCommitLogEntryModel(
id='123',
created_on=self.YEAR_AGO,
last_updated=self.NOW,
commit_type='create',
user_id='',
post_commit_status='public',
post_commit_community_owned=True,
commit_cmds=[])
output = (self.pipeline
| beam.Create([invalid_commit_status])
| beam.ParDo(base_validation.ValidatePostCommitIsPublic()))
self.assert_pcoll_empty(output)
def test_validate_post_commit_is_public_raise_exception(self):
invalid_commit_status = base_models.BaseCommitLogEntryModel(
id='123',
created_on=self.YEAR_AGO,
last_updated=self.NOW,
commit_type='create',
user_id='',
post_commit_status='public',
post_commit_community_owned=False,
commit_cmds=[])
output = (self.pipeline
| beam.Create([invalid_commit_status])
| beam.ParDo(base_validation.ValidatePostCommitIsPublic()))
self.assert_pcoll_equal(output, [
base_validation_errors.InvalidPublicCommitStatusError(
invalid_commit_status),
])