def setMinCores(self, minCores): """Sets the minimum procs value @type minCores: int @param minCores: New minimum cores value""" self.stub.SetMinCores(job_pb2.JobSetMinCoresRequest(job=self.data, val=minCores), timeout=Cuebot.Timeout)
def setMinCores(self, minCores): """Sets the minimum number of cores the job needs. :type minCores: int :param minCores: new minimum cores value """ self.stub.SetMinCores(job_pb2.JobSetMinCoresRequest(job=self.data, val=minCores), timeout=Cuebot.Timeout)
def testSetMinCores(self, getStubMock): stubMock = mock.Mock() stubMock.SetMinCores.return_value = job_pb2.JobSetMinCoresResponse() getStubMock.return_value = stubMock cores = 2000 job = opencue.wrappers.job.Job( job_pb2.Job(name=TEST_JOB_NAME)) job.setMinCores(cores) stubMock.SetMinCores.assert_called_with( job_pb2.JobSetMinCoresRequest(job=job.data, val=cores), timeout=mock.ANY)