def setUp(self): self.database_name = 'testcase_transaction_123' self.db = Database.create(name=self.database_name) self.operating_collection = 'foo_test' self.test_1_col = Collection.create(name=self.operating_collection)
def setUp(self): self.database_name = 'testcase_transaction_123' self.db = Database.create(name=self.database_name) self.operating_collection = 'foo_test' self.test_1_col = Collection.create(name=self.operating_collection)
def setUp(self): self.database_name = 'testcase_index_222_123' self.db = Database.create(name=self.database_name) self.operating_collection = 'bar_extra' self.test_1_col = Collection.create(name=self.operating_collection)
def init(cls): """ """ name = cls.get_collection_name() # Set type try: collection_type = getattr(cls, 'collection_type') except: collection_type = 2 # Create collection try: cls.collection_instance = Collection.create(name=name, type=collection_type) except: cls.collection_instance = Collection.get_loaded_collection(name=name) try: if not isinstance(cls.objects, CollectionModelManager): cls.objects = cls.objects(cls) except: pass # This is the case if init was called more than once # Create meta data for collection cls._model_meta_data = cls.MetaDataObj() # Go through all fields for attribute in cls.get_collection_fields(): attribute.on_init(cls)
def setUp(self): self.database_name = 'testcase_index_222_123' self.db = Database.create(name=self.database_name) self.operating_collection = 'bar_extra' self.test_1_col = Collection.create(name=self.operating_collection)
def test_create_and_delete_collection_without_extra_db(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) self.assertIsNotNone(col) Collection.remove(name=collection_name)
def test_create_and_delete_collection_without_extra_db(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) self.assertIsNotNone(col) Collection.remove(name=collection_name)
def init(cls): """ """ # TODO: Deal with super classes name = cls.get_collection_name() # Set type try: collection_type = getattr(cls, 'collection_type') except: collection_type = 2 # TODO: Database is not set for the collection # Create collection try: cls.collection_instance = Collection.create(name=name, type=collection_type) except: cls.collection_instance = Collection.get_loaded_collection(name=name) try: if not isinstance(cls.objects, CollectionModelManager): cls.objects = cls.objects(cls) except: pass # This is the case if init was called more than once # Create meta data for collection cls._model_meta_data = cls.MetaDataObj() # Go through all fields fields_dict = cls.get_collection_fields_dict() for attribute_name in fields_dict: attribute = fields_dict[attribute_name] # Trigger init event attribute.on_init(cls, attribute_name) # Go through all index model_index_list = cls.get_model_fields_index() for index_attribute_name in model_index_list: # Save created index index_obj = model_index_list[index_attribute_name] created_index = Index(collection=cls.collection_instance, index_type_obj=index_obj) # Reset class attribute setattr(cls, index_attribute_name, created_index) # Save index created_index.save() if not created_index.index_type_obj.is_new: created_index.overwrite()
def init(cls): """ """ name = cls.get_collection_name() # Set type try: collection_type = getattr(cls, 'collection_type') except: collection_type = 2 # TODO: Database is not set for the collection # Create collection try: cls.collection_instance = Collection.create(name=name, type=collection_type) except: cls.collection_instance = Collection.get_loaded_collection( name=name) try: if not isinstance(cls.objects, CollectionModelManager): cls.objects = cls.objects(cls) except: pass # This is the case if init was called more than once # Create meta data for collection cls._model_meta_data = cls.MetaDataObj() # Go through all fields for attribute in cls.get_collection_fields(): # Trigger init event attribute.on_init(cls) # Go through all index model_index_list = cls.get_model_fields_index() for index_attribute_name in model_index_list: # Save created index index_obj = model_index_list[index_attribute_name] created_index = Index(collection=cls.collection_instance, index_type_obj=index_obj) # Reset class attribute setattr(cls, index_attribute_name, created_index) # Save index created_index.save() if not created_index.index_type_obj.is_new: created_index.overwrite()
def on_init(self, model_class): """ """ if not self.related_name is None: relation_name = self._get_relation_collection_name(model_class) try: self.relation_collection = Collection.create(name=relation_name, database=Client.instance().database, type=3) except: self.relation_collection = Collection.get_loaded_collection(name=relation_name) fields = self.relation_class._model_meta_data._fields fields[self.related_name] = ManyToManyField(to=model_class, related_name=None)
def test_get_collection(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) self.assertIsNotNone(col) retrieved_col = Collection.get_loaded_collection(name=collection_name) self.assertEqual(col.id, retrieved_col.id) self.assertEqual(col.name, retrieved_col.name) self.assertEqual(col.type, retrieved_col.type) Collection.remove(name=collection_name)
def test_getting_new_info_for_collection(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) retrieved_col = Collection.get_loaded_collection(name=collection_name) retrieved_col.set_data(waitForSync=True) retrieved_col.save() col.get() self.assertEqual(col.waitForSync, True) Collection.remove(name=collection_name)
def test_get_collection(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) self.assertIsNotNone(col) retrieved_col = Collection.get_loaded_collection(name=collection_name) self.assertEqual(col.id, retrieved_col.id) self.assertEqual(col.name, retrieved_col.name) self.assertEqual(col.type, retrieved_col.type) Collection.remove(name=collection_name)
def test_getting_new_info_for_collection(self): collection_name = 'test_foo_123' col = Collection.create(name=collection_name) retrieved_col = Collection.get_loaded_collection(name=collection_name) retrieved_col.set_data(waitForSync=True) retrieved_col.save() col.get() self.assertEqual(col.waitForSync, True) Collection.remove(name=collection_name)
def test_remove_document_from_collection(self): collection_name = 'test_remove_document_from_collection' col = Collection.create(name=collection_name) doc1 = col.create_document() doc1.save() all_documents = col.documents() self.assertEqual(len(all_documents), 1) doc = all_documents[0] doc.delete() all_documents = col.documents() self.assertEqual(len(all_documents), 0) Collection.remove(name=collection_name)
def test_different_document_revisions(self): collection_name = 'test_revision_documents' col = Collection.create(name=collection_name) doc1 = col.create_document() doc1.save() all_documents = col.documents() self.assertEqual(len(all_documents), 1) doc = all_documents[0] self.assertEqual(doc.revision, doc1.revision) doc.foo = 'bar' doc.save() self.assertNotEqual(doc.revision, doc1.revision) Collection.remove(name=collection_name)
def on_init(self, model_class): """ """ if not self.related_name is None: relation_name = self._get_relation_collection_name(model_class) try: self.relation_collection = Collection.create(name=relation_name, database=Client.instance().database, type=3) except: self.relation_collection = Collection.get_loaded_collection(name=relation_name) fields = self.relation_class._model_meta_data._fields otherside_field = ManyToManyField(to=model_class, related_name=None) fields[self.related_name] = otherside_field # Configure other side field otherside_field.related_queryset = self.relation_class.objects.all() otherside_field.relation_collection = self.relation_collection self.related_queryset = self.relation_class.objects.all()
def init(cls): """ """ # TODO: Deal with super classes name = cls.get_collection_name() # Set type try: collection_type = getattr(cls, 'collection_type') except: collection_type = 2 # TODO: Database is not set for the collection # Create collection try: cls.collection_instance = Collection.create(name=name, type=collection_type) except: cls.collection_instance = Collection.get_loaded_collection( name=name) try: if not isinstance(cls.objects, CollectionModelManager): cls.objects = cls.objects(cls) except: pass # This is the case if init was called more than once cls._default_manager = cls.objects # Create meta data for collection cls._model_meta_data = cls.MetaDataObj() if hasattr(cls, 'Meta'): cls._meta = cls.Meta() cls._meta.model_name = name cls._meta.object_name = name # Giving other classes the chance to extend the meta data on init if hasattr(cls, 'extend_meta_data'): cls.extend_meta_data(cls, cls._meta) # Go through all fields fields_dict = cls.get_collection_fields_dict() for attribute_name in fields_dict: attribute = fields_dict[attribute_name] # Trigger init event attribute.on_init(cls, attribute_name) # Go through all index model_index_list = cls.get_model_fields_index() for index_attribute_name in model_index_list: # Save created index index_obj = model_index_list[index_attribute_name] created_index = Index(collection=cls.collection_instance, index_type_obj=index_obj) # Reset class attribute setattr(cls, index_attribute_name, created_index) # Save index created_index.save() if not created_index.index_type_obj.is_new: created_index.overwrite()
SimpleQuery.all(collection=big_collection) @timer_decorator2('Retrieving %s documents via aql from the collection took %s seconds') def retrieve_aql_documents(big_collection): """ """ all_aql_query = Query() all_aql_query.append_collection(collection_name=big_collection.name) all_aql_query.execute() # Everything needs to be in one try-catch so we can clean up afterwards try: big_collection_name = 'big_collection' big_collection = Collection.create(big_collection_name) print('All tests are using either %s documents, edges or models' % document_number) # Create first all documents create_big_number_of_documents(big_collection) # Get all documents normally retrieve_normally_documents(big_collection) # Get documents with simple query retrieve_simply_documents(big_collection) # Get documents with aql retrieve_aql_documents(big_collection)