1 files changed, 21 insertions, 0 deletions
diff --git a/spec/support/xapian_index.rb b/spec/support/xapian_index.rb
new file mode 100644
index 000000000..21f6192ef
--- /dev/null
+++ b/spec/support/xapian_index.rb
@@ -0,0 +1,21 @@
+# Rebuild the current xapian index
+def rebuild_xapian_index(terms = true, values = true, texts = true, dropfirst = true)
+    if dropfirst
+        begin
+            ActsAsXapian.readable_init
+            FileUtils.rm_r(ActsAsXapian.db_path)
+        rescue RuntimeError
+        end
+        ActsAsXapian.writable_init
+        ActsAsXapian.writable_db.close
+    end
+    parse_all_incoming_messages
+    # safe_rebuild=true, which involves forking to avoid memory leaks, doesn't work well with rspec.
+    # unsafe is significantly faster, and we can afford possible memory leaks while testing.
+    models = [PublicBody, User, InfoRequestEvent]
+    ActsAsXapian.rebuild_index(models, verbose=false, terms, values, texts, safe_rebuild=false)
+end
+
+def update_xapian_index
+    ActsAsXapian.update_index(flush_to_disk=false, verbose=false)
+end