Changeset 416

Timestamp:

02/05/2010 11:32:32 AM (7 months ago)

Author:

matt

Message:

Added unit tests.

Location:

projects/whoosh/trunk/tests

Files:

• test_indexing.py (modified) (1 diff)
• test_writing.py (modified) (3 diffs)

projects/whoosh/trunk/tests/test_indexing.py

@@ -224 +224 @@
         tr.close()
 
+    def test_writer_delete(self):
+        s = fields.Schema(key=fields.ID(stored=True), value=fields.TEXT(stored=True))
+        st = RamStorage()
+        ix = st.create_index(s)
+        
+        w = ix.writer()
+        w.add_document(key=u"1", value=u"alfa")
+        w.add_document(key=u"2", value=u"bravo")
+        w.add_document(key=u"3", value=u"charlie")
+        w.commit()
+        
+        s = ix.searcher()
+        self.assertEqual(s.document(key=u"1")["value"], "alfa")
+        self.assertEqual(s.document(key=u"2")["value"], "bravo")
+        self.assertEqual(s.document(key=u"3")["value"], "charlie")
+        s.close()
+        
+        from whoosh.filedb.filewriting import OPTIMIZE
+        w = ix.writer()
+        w.delete_by_term("key", u"2")
+        w.commit(OPTIMIZE)
+        
+        s = ix.searcher()
+        self.assertEqual(s.document(key=u"1")["value"], "alfa")
+        self.assertEqual(s.document(key=u"3")["value"], "charlie")
+        self.assertEqual(list(s.reader().lexicon("key")), ["1", "3"])
+        s.close()
+
     def test_update(self):
         # Test update with multiple unique keys

projects/whoosh/trunk/tests/test_writing.py

@@ -1 +1 @@
 import unittest
 
-import os, random
+import os, os.path, random
 
 from whoosh import fields, index, writing
@@ -20 +20 @@
         if os.path.exists(path):
             os.remove(path)
+    
+    def clean_dir(self, path):
+        for filename in os.listdir(path):
+            os.remove(os.path.join(path, filename))
+        os.rmdir(path)
     
     def test_asyncwriter(self):
@@ -46 +51 @@
         r = ix.reader()
         self.assertEqual(sorted([int(id) for id in r.lexicon("id")]), range(20))
+        r.close()
+        ix.close()
 
+        self.clean_dir("test_index")