def setGroup(self, group): """Sets the job to a new group. :type group: opencue.wrappers.group.Group :param group: the group you want the job to be in.""" self.stub.SetGroup(job_pb2.JobSetGroupRequest(job=self.data, group_id=group.id()), timeout=Cuebot.Timeout)
def setGroup(self, group): """Sets the job to a new group @type group: Group @param group: the group you want the job to be in.""" self.stub.SetGroup(job_pb2.JobSetGroupRequest(job=self.data, group_id=group.id), timeout=Cuebot.Timeout)
def testSetGroup(self, getStubMock): stubMock = mock.Mock() stubMock.SetGroup.return_value = job_pb2.JobSetGroupResponse() getStubMock.return_value = stubMock groupId = 'ggg-gggg-ggg' group = job_pb2.Group(id=groupId) job = opencue.wrappers.job.Job(job_pb2.Job(name=TEST_JOB_NAME)) job.setGroup(group) stubMock.SetGroup.assert_called_with(job_pb2.JobSetGroupRequest( job=job.data, group_id=groupId), timeout=mock.ANY)