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)
