Added take [skip ci]

Andrew Kane
1 parent d182f8f2
Showing 2 changed files with 12 additions and 0 deletions Show diff stats
lib/searchkick/relation.rb
test/relation_test.rb
@@ -114,6 +114,10 @@ module Searchkick
       self
     end
  
+    def take(limit = nil)
+      limit ? first(limit) : first
+    end
+
     # TODO make more efficient if loaded
     def pluck(*fields)
       result = load(false)
@@ -52,6 +52,14 @@ class RelationTest &lt; Minitest::Test
     assert Product.search.pluck
   end
  
+  def test_take
+    store_names ["Blue", "Red"]
+    assert_kind_of Product, Product.take
+    assert_kind_of Product, Product.search.take if defined?(ActiveRecord)
+    assert_equal 1, Product.take(1).size
+    assert_equal 1, Product.search.take(1).size if defined?(ActiveRecord)
+  end
+
   def test_parameters
     skip unless defined?(ActiveRecord)
     require "action_controller"
...	...	@@ -114,6 +114,10 @@ module Searchkick
114	114	self
115	115	end
116	116
	117	+ def take(limit = nil)
	118	+ limit ? first(limit) : first
	119	+ end
	120	+
117	121	# TODO make more efficient if loaded
118	122	def pluck(*fields)
119	123	result = load(false)
...	...
...	...	@@ -52,6 +52,14 @@ class RelationTest < Minitest::Test
52	52	assert Product.search.pluck
53	53	end
54	54
	55	+ def test_take
	56	+ store_names ["Blue", "Red"]
	57	+ assert_kind_of Product, Product.take
	58	+ assert_kind_of Product, Product.search.take if defined?(ActiveRecord)
	59	+ assert_equal 1, Product.take(1).size
	60	+ assert_equal 1, Product.search.take(1).size if defined?(ActiveRecord)
	61	+ end
	62	+
55	63	def test_parameters
56	64	skip unless defined?(ActiveRecord)
57	65	require "action_controller"
...	...