test/unit/domain_test.rb in rails-erd-0.1.1 vs test/unit/domain_test.rb in rails-erd-0.2.0

@@ -5,16 +5,31 @@
   test "generate should return domain" do
     assert_kind_of Domain, Domain.generate
   end
   
   test "name should return rails application name" do
-    Object::Quux = Module.new
-    Object::Quux::Application = Class.new Rails::Application
-    assert_equal "Quux", Domain.generate.name
+    begin
+      Object::Quux = Module.new
+      Object::Quux::Application = Class.new
+      Object::Rails = Struct.new(:application).new(Object::Quux::Application.new)
+      assert_equal "Quux", Domain.generate.name
+    ensure
+      Object::Quux.send :remove_const, :Application
+      Object.send :remove_const, :Quux
+      Object.send :remove_const, :Rails
+    end
   end
   
   test "name should return nil outside rails" do
     assert_nil Domain.generate.name
+  end
+  
+  test "inspect should display relationships" do
+    create_model "Foo", :bar => :references do
+      belongs_to :bar
+    end
+    create_model "Bar"
+    assert_match %r{#<RailsERD::Domain:.* {Bar => Foo}>}, Domain.generate.inspect
   end
   
   # Entity processing ========================================================
   test "entity_for should return associated entity for given model" do
     create_model "Foo"