elastic_record 0.2.2 → 0.2.3

data/lib/elastic_record/relation/search_methods.rb

@@ -60,16 +60,16 @@ module ElasticRecord
       clone.offset! value
     end
 
-    def select!(value)
-      self.select_values += Array.wrap(value)
+    def select!(*args)
+      self.select_values += args.flatten
       self
     end
 
-    def select(value, &block)
+    def select(*args, &block)
       if block_given?
         to_a.select(&block)
       else
-        clone.select! value
+        clone.select! *args
       end
     end
 

data/test/elastic_record/relation/search_methods_test.rb

@@ -158,6 +158,16 @@ class ElasticRecord::Relation::SearchMethodsTest < MiniTest::Spec
     assert_equal ['foo'], selectable_klass.instance_variable_get('@latest_select')
   end
 
+  def test_select_with_block
+    Widget.elastic_connection.index({'widget' => {'color' => 'red'}}, {index: 'widgets', type: 'widget', id: 5})
+    Widget.elastic_connection.index({'widget' => {'color' => 'blue'}}, {index: 'widgets', type: 'widget', id: 10})
+
+    records = relation.select { |record| record.id == '5' }
+
+    assert_equal 1, records.size
+    assert_equal '5', records.first.id
+  end
+
   def test_extending_with_block
     relation.extending! do
       def foo

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: elastic_record
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
   prerelease: 
 platform: ruby
 authors: