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)