def setGroup(self, parentGroup):
"""Sets this group's parent to parentGroup.
@type parentGroup: Group
@param parentGroup: Group to parent under"""
self.stub.SetGroup(job_pb2.GroupSetGroupRequest(
group=self.data, parent_group=parentGroup),
timeout=Cuebot.Timeout)
def setGroup(self, parentGroup):
"""Sets this group's parent to parentGroup.
:type parentGroup: opencue.wrappers.group.Group
:param parentGroup: Group to parent under"""
self.stub.SetGroup(job_pb2.GroupSetGroupRequest(
group=self.data, parent_group=parentGroup.data),
timeout=Cuebot.Timeout)
def testSetGroup(self, getStubMock):
stubMock = mock.Mock()
stubMock.SetGroup.return_value = job_pb2.GroupSetGroupResponse()
getStubMock.return_value = stubMock
parentGroup = opencue.wrappers.group.Group(job_pb2.Group(name='parentGroup'))
group = opencue.wrappers.group.Group(
job_pb2.Group(name=TEST_GROUP_NAME))
group.setGroup(parentGroup)
stubMock.SetGroup.assert_called_with(
job_pb2.GroupSetGroupRequest(group=group.data, parent_group=parentGroup.data),
timeout=mock.ANY)
