New command: remove

1 files changed, 24 insertions, 0 deletions

diff --git a/tests/test_collection.py b/tests/test_collection.py
index 3959399..a37a696 100644
--- a/tests/test_collection.py
+++ b/tests/test_collection.py
@@ -230,6 +230,30 @@ class testCollectionCreation(unittest.TestCase):
         self.assertFalse(os.path.isfile(os.path.join(tmpdir, '.gitignore')))
         shutil.rmtree(tmpdir)
 
+    def test_deletion(self):
+        tmpdir = tempfile.mkdtemp()
+        shutil.copy('tests/data/simple/settings.yaml', tmpdir)
+        shutil.copytree(
+            'tests/data/simple/items',
+            os.path.join(tmpdir, 'items'),
+            )
+        collection = lesana.Collection.init(tmpdir)
+        # We start with one item indexed with the term "another"
+        collection.start_search('another')
+        mset = collection._enquire.get_mset(0, 10)
+        self.assertEqual(mset.get_matches_estimated(), 1)
+        # Then delete it
+        collection.remove_entries(['11189ee47ddf4796b718a483b379f976'])
+        # An now we should have none
+        self.assertFalse(os.path.exists(os.path.join(
+            tmpdir,
+            'items',
+            '11189ee47ddf4796b718a483b379f976.yaml'
+            )))
+        collection.start_search('another')
+        mset = collection._enquire.get_mset(0, 10)
+        self.assertEqual(mset.get_matches_estimated(), 0)
+
 
 if __name__ == '__main__':
     unittest.main()