def testKill(self, getStubMock): stubMock = mock.Mock() stubMock.Kill.return_value = host_pb2.ProcKillResponse() getStubMock.return_value = stubMock proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME)) proc.kill() stubMock.Kill.assert_called_with( host_pb2.ProcKillRequest(proc=proc.data), timeout=mock.ANY)
def testUnbookKill(self, getStubMock): stubMock = mock.Mock() stubMock.Unbook.return_value = host_pb2.ProcUnbookResponse() getStubMock.return_value = stubMock proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME)) proc.unbook(kill=True) stubMock.Unbook.assert_called_with(host_pb2.ProcUnbookRequest( proc=proc.data, kill=True), timeout=mock.ANY)
def testGetLayer(self, getStubMock): stubMock = mock.Mock() stubMock.GetLayer.return_value = host_pb2.ProcGetLayerResponse( layer=job_pb2.Layer(name=TEST_LAYER_NAME)) getStubMock.return_value = stubMock proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME)) layer = proc.getLayer() stubMock.GetLayer.assert_called_with( host_pb2.ProcGetLayerRequest(proc=proc.data), timeout=mock.ANY) self.assertEqual(layer.name(), TEST_LAYER_NAME)
def testGetHost(self, getStubMock): stubMock = mock.Mock() stubMock.GetHost.return_value = host_pb2.ProcGetHostResponse( host=host_pb2.Host(name=TEST_HOST_NAME)) getStubMock.return_value = stubMock proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME)) host = proc.getHost() stubMock.GetHost.assert_called_with( host_pb2.ProcGetHostRequest(proc=proc.data), timeout=mock.ANY) self.assertEqual(host.name(), TEST_HOST_NAME)
def testGetJob(self, getStubMock): stubMock = mock.Mock() stubMock.GetJob.return_value = host_pb2.ProcGetJobResponse( job=job_pb2.Job(name=TEST_JOB_NAME)) getStubMock.return_value = stubMock proc = opencue.wrappers.proc.Proc(host_pb2.Proc(name=TEST_PROC_NAME)) job = proc.getJob() stubMock.GetJob.assert_called_with( host_pb2.ProcGetJobRequest(proc=proc.data), timeout=mock.ANY) self.assertEqual(job.name(), TEST_JOB_NAME)
def testGetProcs(self, getStubMock): stubMock = mock.Mock() stubMock.GetProcs.return_value = host_pb2.ProcGetProcsResponse( procs=host_pb2.ProcSeq(procs=[host_pb2.Proc(name=TEST_PROC_NAME)])) getStubMock.return_value = stubMock procs = opencue.api.getProcs(show=[TEST_SHOW_NAME], alloc=[TEST_ALLOC_NAME]) stubMock.GetProcs.assert_called_with( host_pb2.ProcGetProcsRequest( r=host_pb2.ProcSearchCriteria(shows=[TEST_SHOW_NAME], allocs=[TEST_ALLOC_NAME])), timeout=mock.ANY) self.assertEqual([TEST_PROC_NAME], [proc.name() for proc in procs])
def testGetProcs(self, getStubMock): procName = 'testProc' stubMock = mock.Mock() stubMock.GetProcs.return_value = host_pb2.HostGetProcsResponse( procs=host_pb2.ProcSeq(procs=[host_pb2.Proc(name=procName)])) getStubMock.return_value = stubMock host = opencue.wrappers.host.Host(host_pb2.Host(name=TEST_HOST_NAME)) procs = host.getProcs() stubMock.GetProcs.assert_called_with( host_pb2.HostGetProcsRequest(host=host.data), timeout=mock.ANY) self.assertEqual(len(procs), 1) self.assertEqual(procs[0].name(), procName)