diff options
Diffstat (limited to 'spec')
-rw-r--r-- | spec/support/model.rb | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/spec/support/model.rb b/spec/support/model.rb index d5be35cbba..969af83cc6 100644 --- a/spec/support/model.rb +++ b/spec/support/model.rb @@ -1,6 +1,6 @@ module Arel module Testing - class Engine + class Engine < Arel::Memory::Engine attr_reader :rows def initialize @@ -12,7 +12,12 @@ module Arel end def read(relation) - @rows.dup.map { |r| Row.new(relation, r) } + case relation + when Arel::Take + relation.eval + else + @rows.dup.map { |r| Row.new(relation, r) } + end end def create(insert) |