From c6acd66ee89f856aa1f49cc38bff76538bf6fbca Mon Sep 17 00:00:00 2001
From: Mike Dvorkin <mike@dvorkin.net>
Date: Sat, 3 Apr 2010 21:42:31 -0700
Subject: [PATCH] Added support for tableless ActiveRecord models

---
 lib/ap/mixin/rails.rb | 2 +-
 spec/rails_spec.rb    | 4 ----
 2 files changed, 1 insertion(+), 5 deletions(-)

diff --git a/lib/ap/mixin/rails.rb b/lib/ap/mixin/rails.rb
index 3a33a7a..598d4d6 100755
--- a/lib/ap/mixin/rails.rb
+++ b/lib/ap/mixin/rails.rb
@@ -38,7 +38,7 @@ module AwesomePrintRails
   # Format ActiveRecord class object.
   #------------------------------------------------------------------------------
   def awesome_active_record_class(object)
-    if object.table_exists?
+    if object.respond_to?(:columns)
       data = object.columns.inject(ActiveSupport::OrderedHash.new) do |hash, c|
         hash[c.name.to_sym] = c.type
         hash
diff --git a/spec/rails_spec.rb b/spec/rails_spec.rb
index 8a38d87..d4841b0 100644
--- a/spec/rails_spec.rb
+++ b/spec/rails_spec.rb
@@ -13,10 +13,6 @@ if defined?(::Rails)
       columns << ActiveRecord::ConnectionAdapters::Column.new(name.to_s, default, sql_type.to_s, null)
     end
 
-    def self.table_exists?
-      true
-    end
-
     column :id, :integer
     column :name, :string
     column :rank, :integer