def getGroups(self):
"""Get the groups for this show
@rtype: list<Group>
@return: list of group wrappers for this show
"""
response = self.stub.GetGroups(
show_pb2.ShowGetGroupsRequest(show=self.data),
timeout=Cuebot.Timeout)
groupSeq = response.groups
return [group.Group(grp) for grp in groupSeq.groups]
def testGetGroups(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetGroups.return_value = show_pb2.ShowGetGroupsResponse(
groups=job_pb2.GroupSeq(groups=[
job_pb2.Group(name=TEST_GROUP_NAME, department=TEST_GROUP_DEPT)
]))
getStubMock.return_value = stubMock
show = opencue.wrappers.show.Show(show_pb2.Show(name=TEST_SHOW_NAME))
groups = show.getGroups()
stubMock.GetGroups.assert_called_with(
show_pb2.ShowGetGroupsRequest(show=show.data), timeout=mock.ANY)
self.assertEqual(len(groups), 1)
self.assertEqual(groups[0].name(), TEST_GROUP_NAME)
self.assertEqual(groups[0].department(), TEST_GROUP_DEPT)
