def addComment(self, subject, message):
"""Appends a comment to the job's comment list
@type subject: str
@param subject: Subject data
@type message: str
@param message: Message data"""
comment = comment_pb2.Comment(user=os.getenv("USER", "unknown"),
subject=subject,
message=message or " ",
timestamp=0)
self.stub.AddComment(job_pb2.JobAddCommentRequest(job=self.data,
new_comment=comment),
timeout=Cuebot.Timeout)
def testAddComment(self, getStubMock):
stubMock = mock.Mock()
stubMock.AddComment.return_value = job_pb2.JobAddCommentResponse()
getStubMock.return_value = stubMock
subject = 'test'
message = 'this is a test.'
comment = comment_pb2.Comment(user=os.getenv('USER', 'unknown'), subject=subject,
message=message, timestamp=0)
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
job.addComment(subject, message)
stubMock.AddComment.assert_called_with(
job_pb2.JobAddCommentRequest(job=job.data, new_comment=comment),
timeout=mock.ANY)
