def testProxyUniqueIdArray(self, getStubMock):
"""convert a list of strings and a class name to a proxy"""
ids = ['A0000000-0000-0000-0000-000000000000', 'B0000000-0000-0000-0000-000000000000']
stubMock = mock.Mock()
stubMock.GetGroup.side_effect = lambda request : job_pb2.GroupGetGroupResponse(
group=job_pb2.Group(id=request.id))
getStubMock.return_value = stubMock
proxyList = opencue.proxy(ids, 'Group')
stubMock.GetGroup.assert_has_calls([
mock.call(job_pb2.GroupGetGroupRequest(id=ids[0])),
mock.call(job_pb2.GroupGetGroupRequest(id=ids[1])),
])
self.assertEqual(ids, [proxy.group.id for proxy in proxyList])
def getGroup(uniq):
"""Returns a Group object from its uniq id.
@rtype: Group
@return: The matching group object"""
return Group(
Cuebot.getStub('group').GetGroup(job_pb2.GroupGetGroupRequest(id=uniq),
timeout=Cuebot.Timeout).group)
def getGroup(uniq):
"""Returns a Group object from its unique ID.
@type uniq: str
@param uniq: a unique group identifier
@rtype: Group
@return: the matching group object"""
return Group(
Cuebot.getStub('group').GetGroup(job_pb2.GroupGetGroupRequest(id=uniq),
timeout=Cuebot.Timeout).group)
def testGetGroup(self, getStubMock):
stubMock = mock.Mock()
stubMock.GetGroup.return_value = job_pb2.GroupGetGroupResponse(
group=job_pb2.Group(id=TEST_GROUP_ID))
getStubMock.return_value = stubMock
group = opencue.api.getGroup(TEST_GROUP_ID)
stubMock.GetGroup.assert_called_with(
job_pb2.GroupGetGroupRequest(id=TEST_GROUP_ID), timeout=mock.ANY)
self.assertEqual(TEST_GROUP_ID, group.id())
def testProxyUniqueId(self, getStubMock):
"""convert a string and class name to proxy"""
id = 'A0000000-0000-0000-0000-000000000000'
stubMock = mock.Mock()
stubMock.GetGroup.return_value = job_pb2.GroupGetGroupResponse(group=job_pb2.Group(id=id))
getStubMock.return_value = stubMock
proxy = opencue.proxy(id, 'Group')
stubMock.GetGroup.assert_called_with(job_pb2.GroupGetGroupRequest(id=id))
self.assertEqual(id, proxy.group.id)
