def setPriority(self, priority):
"""Sets the priority number
@type priority: int
@param priority: New priority number"""
self.stub.SetPriority(job_pb2.JobSetPriorityRequest(job=self.data,
val=priority),
timeout=Cuebot.Timeout)
def testSetPriority(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetPriority.return_value = job_pb2.JobSetPriorityResponse()
getStubMock.return_value = stubMock
priority = 50
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
job.setPriority(priority)
stubMock.SetPriority.assert_called_with(
job_pb2.JobSetPriorityRequest(job=job.data, val=priority), timeout=mock.ANY)
