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