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 kill(self): """Kill the frame running on this proc""" response = self.stub.Kill(host_pb2.ProcKillRequest(proc=self.data), timeout=Cuebot.Timeout) return response