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)