def test_reports_nothing_for_valid_id(self):
model_with_valid_id = auth_models.FirebaseSeedModel(
id=auth_models.ONLY_FIREBASE_SEED_MODEL_ID,
created_on=self.NOW,
last_updated=self.NOW)
output = (self.pipeline
| beam.Create([model_with_valid_id])
| beam.ParDo(auth_validation.ValidateFirebaseSeedModelId()))
self.assert_pcoll_equal(output, [])
def test_reports_error_for_invalid_id(self):
model_with_invalid_id = auth_models.FirebaseSeedModel(
id='2', created_on=self.NOW, last_updated=self.NOW)
output = (self.pipeline
| beam.Create([model_with_invalid_id])
| beam.ParDo(auth_validation.ValidateFirebaseSeedModelId()))
self.assert_pcoll_equal(output, [
base_validation_errors.ModelIdRegexError(
model_with_invalid_id,
auth_models.ONLY_FIREBASE_SEED_MODEL_ID),
])