def test_reports_nothing_for_valid_id(self): model_with_valid_id = auth_models.UserIdByFirebaseAuthIdModel( id='123', user_id='1', created_on=self.NOW, last_updated=self.NOW) output = (self.pipeline | beam.Create([model_with_valid_id]) | beam.ParDo( auth_validation.ValidateUserIdByFirebaseAuthIdModelId())) self.assert_pcoll_equal(output, [])
def test_reports_error_for_invalid_id(self): model_with_invalid_id = auth_models.UserIdByFirebaseAuthIdModel( id='-!\'"', user_id='1', created_on=self.NOW, last_updated=self.NOW) output = ( self.pipeline | beam.Create([model_with_invalid_id]) | beam.ParDo( auth_validation.ValidateUserIdByFirebaseAuthIdModelId()) ) self.assert_pcoll_equal(output, [ base_validation_errors.ModelIdRegexError( model_with_invalid_id, feconf.FIREBASE_AUTH_ID_REGEX), ])