def setMaxCores(self, maxCores):
"""Sets the maximum procs value
@type maxCores: int
@param maxCores: New maximum cores value"""
self.stub.SetMaxCores(job_pb2.JobSetMaxCoresRequest(job=self.data,
val=maxCores),
timeout=Cuebot.Timeout)
def setMaxCores(self, maxCores):
"""Sets the maximum number of cores the job will use.
:type maxCores: int
:param maxCores: new maximum cores value
"""
self.stub.SetMaxCores(job_pb2.JobSetMaxCoresRequest(job=self.data,
val=maxCores),
timeout=Cuebot.Timeout)
def testSetMaxCores(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetMaxCores.return_value = job_pb2.JobSetMaxCoresResponse()
getStubMock.return_value = stubMock
cores = 2000
job = opencue.wrappers.job.Job(
job_pb2.Job(name=TEST_JOB_NAME))
job.setMaxCores(cores)
stubMock.SetMaxCores.assert_called_with(
job_pb2.JobSetMaxCoresRequest(job=job.data, val=cores), timeout=mock.ANY)
