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)