Exemple #1
0
    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),
        ])