Enhance "where" condition for memory adapter

lib/lotus/model/adapters/memory/query.rb

@@ -100,7 +100,17 @@ def all
           #        .where(framework: 'lotus')
           def where(condition)
             column, value = _expand_condition(condition)
-            conditions.push([:where, Proc.new{ find_all{|r| r.fetch(column, nil) == value} }])
+            conditions.push([:where, Proc.new{
+              find_all{|r|
+                case value
+                when Array,Set,Range
+                  value.include?(r.fetch(column, nil))
+                else
+                  r.fetch(column, nil) == value
+                end
+              }
+            }])
+
             self
           end
 
@@ -523,7 +533,7 @@ def _all_with_present_column(column)
           end
 
           def _expand_condition(condition)
-            Array(condition).flatten
+            Array(condition).flatten(1)
           end
         end
       end

test/model/adapters/memory_adapter_test.rb

@@ -324,6 +324,39 @@ class TestDeviceRepository
           result = @adapter.query(collection, &query).all
           result.must_equal [@user3]
         end
+
+        it 'accepts an array as condition' do
+          names = [@user1.name, @user2.name]
+
+          query = Proc.new {
+            where(name: names)
+          }
+
+          result = @adapter.query(collection, &query).all
+          result.must_equal [@user1, @user2]
+        end
+
+        it 'accepts a set as condition' do
+          names = Set.new([@user1.name, @user2.name])
+
+          query = Proc.new {
+            where(name: names)
+          }
+
+          result = @adapter.query(collection, &query).all
+          result.must_equal [@user1, @user2]
+        end
+
+        it 'accepts a range as condition' do
+          ages = @user3.age..@user2.age
+
+          query = Proc.new {
+            where(age: ages)
+          }
+
+          result = @adapter.query(collection, &query).all
+          result.must_equal [@user2, @user3]
+        end
       end
     end