awesome_print/spec/ext/active_record_spec.rb

require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')

begin
  require 'active_record'
  require 'awesome_print/ext/active_record'

  # Create tableless ActiveRecord model.
  #------------------------------------------------------------------------------
  class User < ActiveRecord::Base
    def self.columns()
      @columns ||= []
    end

    def self.column(name, sql_type = nil, default = nil, null = true)
      columns << ActiveRecord::ConnectionAdapters::Column.new(name.to_s, default, sql_type.to_s, null)
    end

    column :id, :integer
    column :name, :string
    column :rank, :integer
    column :admin, :boolean
    column :created_at, :datetime

    def self.table_exists?
      true
    end
  end

  class SubUser < User
    def self.columns
      User.columns
    end
  end

  describe "AwesomePrint/ActiveRecord" do
    before(:each) do
      stub_dotfile!
    end

    #------------------------------------------------------------------------------
    describe "ActiveRecord instance" do
      before(:each) do
        ActiveRecord::Base.default_timezone = :utc
        @diana = User.new(:name => "Diana", :rank => 1, :admin => false, :created_at => "1992-10-10 12:30:00")
        @laura = User.new(:name => "Laura", :rank => 2, :admin => true,  :created_at => "2003-05-26 14:15:00")
        @ap = AwesomePrint.new(:plain => true)
      end

      it "display single record" do
        out = @ap.send(:awesome, @diana)
        str = <<-EOS.strip
#<User:0x01234567> {
            :id => nil,
          :name => "Diana",
          :rank => 1,
         :admin => false,
    :created_at => ?
}
EOS
        if RUBY_VERSION.to_f < 1.9
          str.sub!('?', 'Sat Oct 10 12:30:00 UTC 1992')
        else
          str.sub!('?', '1992-10-10 12:30:00 UTC')
        end

        out.gsub(/0x([a-f\d]+)/, "0x01234567").should == str
      end

      it "display multiple records" do
        out = @ap.send(:awesome, [ @diana, @laura ])
        str = <<-EOS.strip
[
    [0] #<User:0x01234567> {
                :id => nil,
              :name => "Diana",
              :rank => 1,
             :admin => false,
        :created_at => ?
    },
    [1] #<User:0x01234567> {
                :id => nil,
              :name => "Laura",
              :rank => 2,
             :admin => true,
        :created_at => !
    }
]
EOS
        if RUBY_VERSION.to_f < 1.9
          str.sub!('?', 'Sat Oct 10 12:30:00 UTC 1992')
          str.sub!('!', 'Mon May 26 14:15:00 UTC 2003')
        else
          str.sub!('?', '1992-10-10 12:30:00 UTC')
          str.sub!('!', '2003-05-26 14:15:00 UTC')
        end

        out.gsub(/0x([a-f\d]+)/, "0x01234567").should == str
      end
    end

    #------------------------------------------------------------------------------
    describe "ActiveRecord class" do
      it "should print the class" do
        @ap = AwesomePrint.new(:plain => true)
        @ap.send(:awesome, User).should == <<-EOS.strip
class User < ActiveRecord::Base {
            :id => :integer,
          :name => :string,
          :rank => :integer,
         :admin => :boolean,
    :created_at => :datetime
}
EOS
      end

      it "should print the class for non-direct subclasses of AR::Base" do
        @ap = AwesomePrint.new(:plain => true)
        @ap.send(:awesome, SubUser).should == <<-EOS.strip
class SubUser < User {
            :id => :integer,
          :name => :string,
          :rank => :integer,
         :admin => :boolean,
    :created_at => :datetime
}
EOS
      end
    end
  end

rescue LoadError
  puts "Skipping ActiveRecord specs..."
end