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)
