def test(self): u = IntIds() obj = P() obj._p_jar = ConnectionStub() self.assertRaises(KeyError, u.getId, obj) self.assertRaises(KeyError, u.getId, P()) self.assert_(u.queryId(obj) is None) self.assert_(u.queryId(obj, 42) is 42) self.assert_(u.queryId(P(), 42) is 42) self.assert_(u.queryObject(42) is None) self.assert_(u.queryObject(42, obj) is obj) uid = u.register(obj) self.assert_(u.getObject(uid) is obj) self.assert_(u.queryObject(uid) is obj) self.assertEquals(u.getId(obj), uid) self.assertEquals(u.queryId(obj), uid) uid2 = u.register(obj) self.assertEquals(uid, uid2) u.unregister(obj) self.assertRaises(KeyError, u.getObject, uid) self.assertRaises(KeyError, u.getId, obj)
class SearchContainer(ReadonlyContainer): __name__ = 'search' def __init__(self): self.clear() def clear(self): self.tag_container = SearchByTagContainer(self) self.catalog = Catalog() self.catalog['tags'] = KeywordIndex('tags', ITagged) self.catalog['name'] = TextIndex('display_name', IDisplayName, True) self.catalog['__all'] = TextIndex('tokens', ITokenized, True) self.ids = IntIds() def index_object(self, obj): real_obj = follow_symlinks(obj) try: self.catalog.index_doc(self.ids.register(real_obj), real_obj) except NotYet: log.msg("cannot index object %s because it's not yet committed" % obj, system='search') def _index_object(self, obj): real_obj = follow_symlinks(obj) self.catalog.index_doc(self.ids.register(real_obj), real_obj) def unindex_object(self, obj): try: self.catalog.unindex_doc(self.ids.register(obj)) except NotYet: log.msg("cannot unindex object %s because it's not yet committed" % obj, system='search') def search(self, **kwargs): # HACK, we should be able to setup a persistent utility provideUtility(self.ids, IIntIds) return list(self.catalog.searchResults(**kwargs)) def search_goog(self, query): # hack, zope catalog treats ':' specially return self.search(__all=query.replace(':', '_')) @property def _items(self): return {'by-tag': self.tag_container}
def setupCatalog(test): intids = IntIds() component.provideUtility(intids, IIntIds) component.provideAdapter(SimpleKeyReference) cat = Catalog() cat['org_name'] = FieldIndex('name', IOrganization) cat['proj_name'] = FieldIndex('name', IProject) cat['proj_descr'] = FieldIndex('description', IProject) cat['student_name'] = FieldIndex('name', IStudent) cat['student_country'] = FieldIndex('country', IStudent) cat['mentor_name'] = FieldIndex('name', IMentor) cat['all_students'] = AllIndex(IStudent) cat['all_mentors'] = AllIndex(IMentor) cat['all_projects'] = AllIndex(IProject) cat['all_orgs'] = AllIndex(IOrganization) m1 = Mentor(u"John Doe") id = intids.register(m1) cat.index_doc(id, m1) p1 = Project(u"Save the world", u'test') id = intids.register(p1) cat.index_doc(id, p1) s1 = Student(u"Charith", u"Sri Lanka") id = intids.register(s1) cat.index_doc(id, s1) s2 = Student(u"Jane", u"USA") id = intids.register(s2) cat.index_doc(id, s2) s3 = Student(u"Ann", u"Hungary") id = intids.register(s3) cat.index_doc(id, s3) s4 = Student(u"Stewart", u"Hungary", m1) id = intids.register(s4) cat.index_doc(id, s4) o1 = Organization(u"Zope.org") id = intids.register(o1) cat.index_doc(id, o1) # cat2 = zc.relation.catalog.Catalog(dumpRelation, loadRelation) # cat2.addValueIndex(IProjectRelation['project'], dumpObj, loadObj, btree=BTrees.family32.OO) # cat2.addValueIndex(IProjectRelation['mentor'], dumpObj, loadObj, btree=BTrees.family32.OO) # cat2.addDefaultQueryFactory(zc.relation.queryfactory.TransposingTransitive('project','mentor')) # # rel = ProjectRelation(m1, p1) # cat2.index(rel) # component.provideUtility(cat2, zc.relation.interfaces.ICatalog, name='rel-catalog') component.provideUtility(cat, ICatalog, name='foo-catalog')
def setupCatalog(test, optCount=10, unoptCount=10, halfCount=10): intids = IntIds() component.provideUtility(intids, IIntIds) component.provideAdapter(SimpleKeyReference) cat = Catalog() cat['opt_name'] = FieldIndex('name', IOptimizedClass) cat['opt_value'] = FieldIndex('value', IOptimizedClass) cat['half_name'] = FieldIndex('name', IHalfOptimizedClass) cat['half_valueOpt'] = FieldIndex('value', IHalfOptimizedClass) cat['all_opt'] = AllIndex(IOptimizedClass) cat['all_unopt'] = AllIndex(IUnOptimizedClass) cat['all_half'] = AllIndex(IHalfOptimizedClass) for i in range(optCount): o = OptimizedClass() o.value = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) for i in range(unoptCount): o = UnOptimizedClass() o.value = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) for i in range(halfCount): o = HalfOptimizedClass() o.valueOpt = i o.valueNoOpt = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) component.provideUtility(cat, ICatalog, name='foo-catalog')
def setupCatalog(test, optCount=10, unoptCount=10, halfCount=10): intids = IntIds() component.provideUtility(intids, IIntIds) component.provideAdapter(SimpleKeyReference) cat = Catalog() cat['opt_name'] = FieldIndex('name', IOptimizedClass) cat['opt_value'] = FieldIndex('value', IOptimizedClass) cat['half_name'] = FieldIndex('name', IHalfOptimizedClass) cat['half_valueOpt'] = FieldIndex('value', IHalfOptimizedClass) cat['all_opt'] = AllIndex(IOptimizedClass) cat['all_unopt'] = AllIndex(IUnOptimizedClass) cat['all_half'] = AllIndex(IHalfOptimizedClass) for i in range(optCount): o = OptimizedClass() o.value = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) for i in range(unoptCount): o = UnOptimizedClass() o.value = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) for i in range(halfCount): o = HalfOptimizedClass() o.valueOpt = i o.valueNoOpt = i o.name = unicode(i) id = intids.register(o) cat.index_doc(id, o) component.provideUtility(cat, ICatalog, name='foo-catalog')
def test_len_items(self): u = IntIds() obj = P() obj._p_jar = ConnectionStub() self.assertEquals(len(u), 0) self.assertEquals(u.items(), []) self.assertEquals(list(u), []) uid = u.register(obj) ref = KeyReferenceToPersistent(obj) self.assertEquals(len(u), 1) self.assertEquals(u.items(), [(uid, ref)]) self.assertEquals(list(u), [uid]) obj2 = P() obj2.__parent__ = obj uid2 = u.register(obj2) ref2 = KeyReferenceToPersistent(obj2) self.assertEquals(len(u), 2) result = u.items() expected = [(uid, ref), (uid2, ref2)] result.sort() expected.sort() self.assertEquals(result, expected) result = list(u) expected = [uid, uid2] result.sort() expected.sort() self.assertEquals(result, expected) u.unregister(obj) u.unregister(obj2) self.assertEquals(len(u), 0) self.assertEquals(u.items(), [])
def setUp(test): root = setup.placefulSetUp(site=True) testing.setUp() provideAdapter(connectionOfPersistent, (IPersistent,), IConnection) provideAdapter(KeyReferenceToPersistent, (IPersistent,), IKeyReference) utility = IntIds() provideUtility(utility, IIntIds) root._p_jar = ConnectionStub() root['folder1'] = Folder() root['folder1']['folder1_1'] = Folder() root['folder1']['folder1_1']['folder1_1_1'] = Folder() root['folder2'] = Folder() root['folder2']['folder2_2'] = Folder() root['folder2']['folder2_2']['folder2_2_2'] = Folder() utility.register(root) utility.register(root['folder1']) utility.register(root['folder1']['folder1_1']) utility.register(root['folder1']['folder1_1']['folder1_1_1']) utility.register(root['folder2']) utility.register(root['folder2']['folder2_2']) utility.register(root['folder2']['folder2_2']['folder2_2_2']) test.globs['root'] = root