def setMaxRetries(self, maxRetries): """Sets the number of retries before a frame goes dead @type maxRetries: int @param maxRetries: New max retries""" self.stub.SetMaxRetries(job_pb2.JobSetMaxRetriesRequest( job=self.data, max_retries=maxRetries), timeout=Cuebot.Timeout)
def testSetMaxRetries(self, getStubMock): stubMock = mock.Mock() stubMock.SetMaxRetries.return_value = job_pb2.JobSetMaxRetriesResponse() getStubMock.return_value = stubMock retries = 10 job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) job.setMaxRetries(retries) stubMock.SetMaxRetries.assert_called_with( job_pb2.JobSetMaxRetriesRequest(job=job.data, max_retries=retries), timeout=mock.ANY)