def addComment(self, subject, message):
"""Appends a comment to the hosts'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(host_pb2.HostAddCommentRequest(
host=self.data, new_comment=comment),
timeout=Cuebot.Timeout)
def testAddComment(self, getStubMock):
stubMock = mock.Mock()
stubMock.AddComment.return_value = host_pb2.HostAddCommentResponse()
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)
host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME))
host.addComment(subject, message)
stubMock.AddComment.assert_called_with(host_pb2.HostAddCommentRequest(
host=host.data, new_comment=comment),
timeout=mock.ANY)