def testKill(self, getStubMock): stubMock = mock.Mock() stubMock.Delete.return_value = job_pb2.JobKillResponse() getStubMock.return_value = stubMock job = opencue.wrappers.job.Job(job_pb2.Job(name=TEST_JOB_NAME)) job.kill() stubMock.Kill.assert_called_with(job_pb2.JobKillRequest(job=job.data), timeout=mock.ANY)
def kill(self): """Kills the job.""" self.stub.Kill(job_pb2.JobKillRequest(job=self.data), timeout=Cuebot.Timeout)