def test_reindex(self): """ Test that an existing document can be re-indexed, updating the content and metadata in the process. """ doc = self.index.index('test doc', foo='bar', baze='nug') doc_db = Document.select(Document.rowid, Document.content).get() self.assertTrue(doc_db.rowid is not None) self.assertEqual(doc_db.content, 'test doc') self.assertEqual(doc_db.metadata, {'foo': 'bar', 'baze': 'nug'}) updated_doc = self.index.index( 'updated doc', document=doc, foo='bazz', nug='x') self.assertEqual(Document.select().count(), 1) self.assertEqual(updated_doc.metadata, {'foo': 'bazz', 'nug': 'x'}) u_doc_db = Document.select(Document.rowid, Document.content).get() self.assertEqual(u_doc_db.content, 'updated doc') self.assertEqual(u_doc_db.rowid, doc_db.rowid) self.assertEqual(u_doc_db.metadata, {'foo': 'bazz', 'nug': 'x'}) # Verify through relationship properties. self.assertEqual(IndexDocument.select().count(), 1) idx_doc = IndexDocument.get() self.assertEqual(idx_doc._data['document'], u_doc_db.rowid) self.assertEqual(idx_doc._data['index'], self.index.id)
def test_reindex(self): """ Test that an existing document can be re-indexed, updating the content and metadata in the process. """ doc = self.index.index('test doc', foo='bar', baze='nug') doc_db = (Document .select(Document._meta.primary_key, Document.content) .get()) self.assertTrue(doc_db.get_id() is not None) self.assertEqual(doc_db.content, 'test doc') self.assertEqual(doc_db.metadata, {'foo': 'bar', 'baze': 'nug'}) updated_doc = self.index.index( 'updated doc', document=doc, foo='bazz', nug='x') self.assertEqual(Document.select().count(), 1) self.assertEqual(updated_doc.metadata, {'foo': 'bazz', 'nug': 'x'}) u_doc_db = (Document .select(Document._meta.primary_key, Document.content) .get()) self.assertEqual(u_doc_db.content, 'updated doc') self.assertEqual(u_doc_db.get_id(), doc_db.get_id()) self.assertEqual(u_doc_db.metadata, {'foo': 'bazz', 'nug': 'x'}) # Verify through relationship properties. self.assertEqual(IndexDocument.select().count(), 1) idx_doc = IndexDocument.get() self.assertEqual(idx_doc.__data__['document'], u_doc_db.get_id()) self.assertEqual(idx_doc.__data__['index'], self.index.id)
def test_index_document(self): """ Basic test case ensuring that content can be indexed and that the many-to-many relationship between documents and indexes is set up correctly. """ content = 'huey is a sweet little kitty' doc = self.index.index(content=content) # Verify document properties. self.assertEqual(doc.rowid, 1) self.assertEqual(doc.content, content) self.assertEqual(doc.metadata, {}) # Verify through relationship properties. self.assertEqual(IndexDocument.select().count(), 1) idx_doc = IndexDocument.get() self.assertEqual(idx_doc._data['document'], doc.rowid) self.assertEqual(idx_doc._data['index'], self.index.id)
def test_index_document(self): """ Basic test case ensuring that content can be indexed and that the many-to-many relationship between documents and indexes is set up correctly. """ content = 'huey is a sweet little kitty' doc = self.index.index(content=content) # Verify document properties. self.assertEqual(doc.get_id(), 1) self.assertEqual(doc.content, content) self.assertEqual(doc.metadata, {}) # Verify through relationship properties. self.assertEqual(IndexDocument.select().count(), 1) idx_doc = IndexDocument.get() self.assertEqual(idx_doc.__data__['document'], doc.get_id()) self.assertEqual(idx_doc.__data__['index'], self.index.id)