Manually merged expanded mongo_mapper support provided by gardelea

spec/ext/mongo_mapper_spec.rb

@@ -24,10 +24,14 @@ begin
       @ap = AwesomePrint::Inspector.new(:plain => true, :sort_keys => true)
     end
 
-    it "should print class instance" do
-      user = MongoUser.new(:first_name => "Al", :last_name => "Capone")
-      out = @ap.send(:awesome, user)
-      str = <<-EOS.strip
+    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",
@@ -40,32 +44,143 @@ begin
     attr_reader :last_name_before_type_cast = "Capone"
 >
 EOS
-      out.gsub!(/0x([a-f\d]+)/, "0x01234567")
-      out.should == str
-    end
+        out.gsub!(/0x([a-f\d]+)/, "0x01234567")
+        out.should == str
+      end
 
-    it "should print the class" do
-      @ap.send(:awesome, MongoUser).should == <<-EOS.strip
+      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
+      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