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)
