def getRootGroup(self):
"""Get the root group for the show
@rtype: opencue.wrappers.group.Group
@return: Group wrapper of the root group
"""
response = self.stub.GetRootGroup(
show_pb2.ShowGetRootGroupRequest(show=self.data),
timeout=Cuebot.Timeout)
return group.Group(response.group)
def testGetRootGroup(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetRootGroup.return_value = show_pb2.ShowGetRootGroupResponse(
group=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))
rootGroup = show.getRootGroup()
stubMock.GetRootGroup.assert_called_with(
show_pb2.ShowGetRootGroupRequest(show=show.data), timeout=mock.ANY)
self.assertEqual(rootGroup.name(), TEST_GROUP_NAME)
self.assertEqual(rootGroup.department(), TEST_GROUP_DEPT)
