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)
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])
