def getGroups(self): """Returns child groups of this group. :rtype: list<opencue.wrappers.group.Group> :return: list of child groups """ response = self.stub.GetGroups(job_pb2.GroupGetGroupsRequest(group=self.data), timeout=Cuebot.Timeout) return [Group(g) for g in response.groups.groups]
def testGetGroups(self, getStubMock): stubMock = mock.Mock() stubMock.GetGroups.return_value = job_pb2.GroupGetGroupsResponse( groups=job_pb2.GroupSeq(groups=[job_pb2.Group(name=TEST_GROUP_NAME)])) getStubMock.return_value = stubMock group = opencue.wrappers.group.Group( job_pb2.Group(name=TEST_GROUP_NAME)) groups = group.getGroups() stubMock.GetGroups.assert_called_with( job_pb2.GroupGetGroupsRequest(group=group.data), timeout=mock.ANY) self.assertEqual(len(groups), 1) self.assertEqual(groups[0].name(), TEST_GROUP_NAME)
def getGroups(self): response = self.stub.GetGroups( job_pb2.GroupGetGroupsRequest(group=self.data), timeout=Cuebot.Timeout) return [Group(g) for g in response.groups.groups]