def assert_audit(actual): job = klio_pb2.KlioJob() job.job_name = "a-job" job.gcp_project = "not-a-real-project" audit_log_item = klio_pb2.KlioJobAuditLogItem() audit_log_item.klio_job.CopyFrom(job) exp_msg = klio_pb2.KlioMessage() exp_msg.version = klio_pb2.Version.V2 exp_msg.metadata.job_audit_log.extend([audit_log_item]) expected = exp_msg.SerializeToString() assert expected == actual return actual
def _create_audit_item(self): audit_log_item = klio_pb2.KlioJobAuditLogItem() audit_log_item.timestamp.GetCurrentTime() current_job = self._generate_current_job_object() audit_log_item.klio_job.CopyFrom(current_job) return audit_log_item