Objectinspector now supports arbitary types

Author: Siva Gollapalli

lib/rspec/support/object_inspector.rb

@@ -14,6 +14,10 @@ def self.inspect(object)
         elsif RSpec::Support.is_a_matcher?(object) && object.respond_to?(:description)
           object.description
         else
+          registered_klasses.each do |klass, inspector|
+            return inspector.call(object) if klass === object
+          end
+
           object.inspect
         end
       end
@@ -43,6 +47,14 @@ def self.format_date_time(date_time)
           date_time.inspect
         end
       end
+
+      def self.register(klass, &block)
+        registered_klasses[klass] = block
+      end
+
+      def self.registered_klasses
+        @registered_klasses ||= {}
+      end
     end
   end
 end

spec/rspec/support/object_inspector_spec.rb

@@ -92,6 +92,35 @@ def with_date_loaded
             matcher_without_a_description.inspect)
         end
       end
+
+      context "with arbitary types" do
+        class User
+          attr_accessor :name
+
+          def initialize(name)
+            @name = name
+          end
+        end
+
+        before(:context) do
+          ObjectInspector.register User do |user|
+            "User (#{user.name})"
+          end
+        end
+
+        it "works with custom class" do
+          user = User.new("FooBar")
+          expect(ObjectInspector.inspect(user)).to eq("User (FooBar)")
+        end
+
+        it "works with subclasses" do
+          class Admin < User
+          end
+
+          admin = Admin.new("FooBar")
+          expect(ObjectInspector.inspect(admin)).to eq("User (FooBar)")
+        end
+      end
     end
   end
 end