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

begin
  require "mongo_mapper"
  require "awesome_print/ext/mongo_mapper"

  describe "AwesomePrint/MongoMapper" do
    before :all do
      class MongoUser
        include MongoMapper::Document

        key :first_name, String
        key :last_name, String
      end
    end

    after :all do
      Object.instance_eval{ remove_const :MongoUser }
      Object.instance_eval{ remove_const :Chamelion }
    end

    before do
      stub_dotfile!
      @ap = AwesomePrint::Inspector.new(:plain => true, :sort_keys => true)
    end

    describe "with the raw option set to true" do
      # before { @ap.options[:raw] = true }
      before { @ap = AwesomePrint::Inspector.new(:plain => true, :sort_keys => true, :raw => true) }

      it "should print class instance" do
        user = MongoUser.new(:first_name => "Al", :last_name => "Capone")
        out = @ap.send(:awesome, user)
        str = <<-EOS.strip
#<MongoUser:0x01234567
    @_new = true,
    attr_accessor :first_name = "Al",
    attr_accessor :last_name = "Capone",
    attr_reader :changed_attributes = {
        "first_name" => nil,
         "last_name" => nil
    },
    attr_reader :first_name_before_type_cast = "Al",
    attr_reader :last_name_before_type_cast = "Capone"
>
EOS
        out.gsub!(/0x([a-f\d]+)/, "0x01234567")
        out.should == str
      end

      it "should print the class" do
        @ap.send(:awesome, MongoUser).should == <<-EOS.strip
class MongoUser < Object {
           "_id" => :object_id,
    "first_name" => :string,
     "last_name" => :string
}
EOS
      end

      it "should print the class when type is undefined" do
        class Chamelion
          include MongoMapper::Document
          key :last_attribute
        end

        @ap.send(:awesome, Chamelion).should == <<-EOS.strip
class Chamelion < Object {
               "_id" => :object_id,
    "last_attribute" => :undefined
}
EOS
      end
    end

    describe "with associations" do
      before :all do
        class Child
          include MongoMapper::EmbeddedDocument
          key :data
        end

        class Sibling
          include MongoMapper::Document
          key :title
        end

        class Parent
          include MongoMapper::Document
          key :name

          one :child
          one :sibling
        end
      end

      describe "with show associations turned off (default)" do
        it "should render the class as normal" do
          @ap.send(:awesome, Parent).should == <<-EOS.strip
class Parent < Object {
     "_id" => :object_id,
    "name" => :undefined
}
EOS
        end

        it "should render an instance as normal" do
          parent = Parent.new(:name => 'test')
          out = @ap.send(:awesome, parent)
          str = <<-EOS.strip
#<Parent:0x01234567> {
     "_id" => BSON::ObjectId('4d9183739a546f6806000001'),
    "name" => "test"
}
EOS
          out.gsub!(/'([\w]+){23}'/, "'4d9183739a546f6806000001'")
          out.gsub!(/0x([a-f\d]+)/, "0x01234567")
          out.should == str
        end
      end

      describe "with show associations turned on and inline embedded turned off" do
        before :each do
          @ap = AwesomePrint::Inspector.new(:plain => true, :mongo_mapper => { :show_associations => true })
        end

        it "should render the class with associations shown" do
          @ap.send(:awesome, Parent).should == <<-EOS.strip
class Parent < Object {
        "_id" => :object_id,
       "name" => :undefined,
      "child" => embeds one Child,
    "sibling" => one Sibling
}
EOS
        end

        it "should render an instance with associations shown" do
          parent = Parent.new(:name => 'test')
          out = @ap.send(:awesome, parent)
          str = <<-EOS.strip
#<Parent:0x01234567> {
        "_id" => BSON::ObjectId('4d9183739a546f6806000001'),
       "name" => "test",
      "child" => embeds one Child,
    "sibling" => one Sibling
}
EOS
          out.gsub!(/'([\w]+){23}'/, "'4d9183739a546f6806000001'")
          out.gsub!(/0x([a-f\d]+)/, "0x01234567")
          out.should == str
        end
      end

      describe "with show associations turned on and inline embedded turned on" do
        before :each do
          @ap = AwesomePrint::Inspector.new(:plain => true,
              :mongo_mapper => {
                :show_associations => true,
                :inline_embedded   => true
            }
          )
        end

        it "should render an instance with associations shown and embeds there" do
          parent = Parent.new(:name => 'test', :child => Child.new(:data => 5))
          out = @ap.send(:awesome, parent)
          str = <<-EOS.strip
#<Parent:0x01234567> {
        "_id" => BSON::ObjectId('4d9183739a546f6806000001'),
       "name" => "test",
      "child" => embedded #<Child:0x01234567> {
         "_id" => BSON::ObjectId('4d9183739a546f6806000001'),
        "data" => 5
    },
    "sibling" => one Sibling
}
EOS
          out.gsub!(/'([\w]+){23}'/, "'4d9183739a546f6806000001'")
          out.gsub!(/0x([a-f\d]+)/, "0x01234567")
          out.should == str
        end
      end
    end
  end

rescue LoadError => error
  puts "Skipping MongoMapper specs: #{error}"
end