def getComments(self): """returns the jobs comments""" response = self.stub.GetComments( job_pb2.JobGetCommentsRequest(job=self.data), timeout=Cuebot.Timeout) commentSeq = response.comments return [comment.Comment(cmt) for cmt in commentSeq.comments]
def getComments(self): """Returns the job's comment list. :rtype: list<opencue.wrappers.comment.Comment> :return: the job's comment list """ response = self.stub.GetComments( job_pb2.JobGetCommentsRequest(job=self.data), timeout=Cuebot.Timeout) commentSeq = response.comments return [ opencue.wrappers.comment.Comment(cmt) for cmt in commentSeq.comments ]
def testGetComments(self, getStubMock): message = 'this is a test.' stubMock = mock.Mock() stubMock.GetComments.return_value = job_pb2.JobGetCommentsResponse( comments=comment_pb2.CommentSeq( comments=[comment_pb2.Comment(message=message)])) getStubMock.return_value = stubMock job = opencue.wrappers.job.Job(job_pb2.Job(name=TEST_JOB_NAME)) comments = job.getComments() stubMock.GetComments.assert_called_with( job_pb2.JobGetCommentsRequest(job=job.data), timeout=mock.ANY) self.assertEqual(len(comments), 1) self.assertEqual(comments[0].message(), message)