Add Page model search specs

Author: rossta

app/models/concerns/searchable.rb

@@ -31,8 +31,10 @@ module Searchable
     # part of the FTS5 extension.
     # https://www.sqlite.org/fts5.html#the_snippet_function
     #
+    klass = self
+
     scope :with_snippets, ->(*columns) do
-      snippet_columns = search_index_mapping.keys.map(&:to_s) & columns.map(&:to_s)
+      snippet_columns = klass.search_index_mapping.keys.map(&:to_s) & columns.map(&:to_s)
       scope = select("#{table_name}.*").join_search_index
       snippet_columns.each.with_index do |column, column_index|
         scope = scope.select("snippet(#{search_table_name}, #{column_index}, '<mark>', '</mark>', '…', 32) AS #{column}_snippet")
@@ -85,7 +87,6 @@ def search_index_mapping
 
       @search_index_mapping ||= search_index_definition.flat_map { |item| Array(item) }.map { |a, b = a| [a, b] }.to_h
     end
-    private :search_index_mapping
 
     def refresh_search_index
       find_each(&:update_in_search_index)

spec/models/page_spec.rb

@@ -30,4 +30,26 @@
       expect(Page.search("Joy of Rails")).to include page
     end
   end
+
+  describe ".rank" do
+    it "orders search results by rank" do
+      page_1 = Page.find_or_create_by!(request_path: "/articles/custom-color-schemes-with-ruby-on-rails")
+      page_2 = Page.find_or_create_by!(request_path: "/articles/mastering-custom-configuration-in-rails")
+
+      search = Page.search("custom con*")
+
+      expect(search.ranked.to_a).to eq([page_2, page_1])
+    end
+  end
+
+  describe ".with_snippets" do
+    it "orders search results by rank" do
+      Page.find_or_create_by!(request_path: "/articles/custom-color-schemes-with-ruby-on-rails")
+
+      page = Page.search("Color*").with_snippets(:title, :body).first
+
+      expect(page.title_snippet).to match(%r{<mark>Color</mark>})
+      expect(page.body_snippet).to match(%r{<mark>color</mark>})
+    end
+  end
 end