def test_process_reports_model_mutated_during_job_error(self): invalid_timestamp = base_models.BaseModel(id='124', created_on=self.NOW, last_updated=self.YEAR_LATER) output = (self.pipeline | beam.Create([invalid_timestamp]) | beam.ParDo(base_validation.ValidateModelTimestamps())) self.assert_pcoll_equal(output, [ base_validation_errors.ModelMutatedDuringJobError( invalid_timestamp), ])
def test_process_reports_model_timestamp_relationship_error(self): invalid_timestamp = base_models.BaseModel(id='123', created_on=self.NOW, last_updated=self.YEAR_AGO) output = (self.pipeline | beam.Create([invalid_timestamp]) | beam.ParDo(base_validation.ValidateModelTimestamps())) self.assert_pcoll_equal(output, [ base_validation_errors.InconsistentTimestampsError( invalid_timestamp), ])