From 04b3f0611c69532526d8b16799a7625b92dcf1fc Mon Sep 17 00:00:00 2001
From: Aaron Patterson <aaron.patterson@gmail.com>
Date: Fri, 30 Jul 2010 10:38:54 -0700
Subject: PERF: removing def engine

---
 spec/support/model.rb | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'spec/support')

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)
-- 
cgit v1.2.3