Added type

Andrew Kane
1 parent c0d07b65
Showing 2 changed files with 10 additions and 0 deletions Show diff stats
lib/searchkick/relation.rb
test/inheritance_test.rb
@@ -363,6 +363,15 @@ module Searchkick
       self
     end
  
+    def type(*args)
+      spawn.type!(*args)
+    end
+
+    def type!(*args)
+      options[:type] = Array(options[:type]) + args.flatten
+      self
+    end
+
     # same as Active Record
     def inspect
       entries = results.first(11).map!(&:inspect)
@@ -32,6 +32,7 @@ class InheritanceTest &lt; Minitest::Test
     store_names ["Green Bear"], Dog
     store_names ["Blue Bear"], Cat
     assert_equal ["Blue Bear"], Animal.search("bear", type: [Cat]).map(&:name)
+    assert_equal ["Blue Bear"], Animal.search("bear", relation: true).type(Cat).map(&:name)
   end
  
   def test_force_multiple_types
...	...	@@ -363,6 +363,15 @@ module Searchkick
363	363	self
364	364	end
365	365
	366	+ def type(*args)
	367	+ spawn.type!(*args)
	368	+ end
	369	+
	370	+ def type!(*args)
	371	+ options[:type] = Array(options[:type]) + args.flatten
	372	+ self
	373	+ end
	374	+
366	375	# same as Active Record
367	376	def inspect
368	377	entries = results.first(11).map!(&:inspect)
...	...
...	...	@@ -32,6 +32,7 @@ class InheritanceTest < Minitest::Test
32	32	store_names ["Green Bear"], Dog
33	33	store_names ["Blue Bear"], Cat
34	34	assert_equal ["Blue Bear"], Animal.search("bear", type: [Cat]).map(&:name)
	35	+ assert_equal ["Blue Bear"], Animal.search("bear", relation: true).type(Cat).map(&:name)
35	36	end
36	37
37	38	def test_force_multiple_types
...	...