gotime-cassandra_object 4.3.1 → 4.3.2

data/gotime-cassandra_object.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name = 'gotime-cassandra_object'
-  s.version = '4.3.1'
+  s.version = '4.3.2'
   s.description = 'Cassandra ActiveModel'
   s.summary = 'Cassandra ActiveModel'
   s.authors = ["Michael Koziarski", "gotime"]

data/lib/cassandra_object/scope/query_methods.rb

@@ -1,13 +1,17 @@
 module CassandraObject
   class Scope
     module QueryMethods
-      def select!(value)
-        self.select_values += Array.wrap(value)
+      def select!(*values)
+        self.select_values += values.flatten
         self
       end
 
-      def select(value)
-        clone.select! value
+      def select(*values, &block)
+        if block_given?
+          to_a.select(&block)
+        else
+          clone.select! *values
+        end
       end
 
       def where!(*values)

data/test/unit/scope/query_methods_test.rb

@@ -1,6 +1,6 @@
 require 'test_helper'
 
-class CassandraObject::QueryMethodsTest < CassandraObject::TestCase
+class CassandraObject::Scope::QueryMethodsTest < CassandraObject::TestCase
   test "select" do
     issue = Issue.create title: 'foo', description: 'bar'
 
@@ -9,4 +9,11 @@ class CassandraObject::QueryMethodsTest < CassandraObject::TestCase
     assert_equal 'foo', issue.title
     assert_nil issue.description
   end
+
+  test "select with block" do
+    foo_issue = Issue.create title: 'foo'
+    bar_issue = Issue.create title: 'bar'
+
+    assert_equal [foo_issue], Issue.select { |issue| issue.title == 'foo' }
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gotime-cassandra_object
 version: !ruby/object:Gem::Version
-  version: 4.3.1
+  version: 4.3.2
   prerelease: 
 platform: ruby
 authors: