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