domain_test.rb in rails-erd-1.0.0

- old
+ new

@@ -3,11 +3,11 @@
 class DomainTest < ActiveSupport::TestCase
   # Domain ===================================================================
   test "generate should return domain" do
     assert_kind_of Domain, Domain.generate
   end
-  
+
   test "name should return rails application name" do
     begin
       Object::Quux = Module.new
       Object::Quux::Application = Class.new
       Object::Rails = Struct.new(:application).new(Object::Quux::Application.new)
@@ -16,34 +16,34 @@
       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 object id only" do
     create_model "Foo", :bar => :references do
       belongs_to :bar
     end
     create_model "Bar"
     assert_match %r{#<RailsERD::Domain:.*>}, Domain.generate.inspect
   end
-  
+
   # Entity processing ========================================================
   test "entity_by_name should return associated entity for given name" do
     create_model "Foo"
     assert_equal Foo, Domain.generate.entity_by_name("Foo").model
   end
-  
+
   test "entities should return domain entities" do
     create_models "Foo", "Bar"
     assert_equal [Domain::Entity] * 2, Domain.generate.entities.collect(&:class)
   end
-  
+
   test "entities should return all domain entities sorted by name" do
     create_models "Foo", "Bar", "Baz", "Qux"
     assert_equal [Bar, Baz, Foo, Qux], Domain.generate.entities.collect(&:model)
   end
 
@@ -61,26 +61,26 @@
     create_model "Galleon" do
       has_many :cannons, :as => :defensible
     end
     assert_equal ["Defensible", "Galleon", "Stronghold"], Domain.generate.entities.collect(&:name)
   end
-  
+
   test "entities should omit abstract models" do
     Object.const_set :Foo, Class.new(ActiveRecord::Base) { self.abstract_class = true }
     create_model "Bar", Foo do
-      set_table_name "bars"
+      self.table_name = "bars"
     end
     create_table "foos", {}, true
     create_table "bars", {}, true
     assert_equal ["Bar"], Domain.generate.entities.collect(&:name)
   end
-  
+
   # Relationship processing ==================================================
   test "relationships should return empty array for empty domain" do
     assert_equal [], Domain.generate.relationships
   end
-  
+
   test "relationships should return relationships in domain model" do
     create_models "Baz", "Qux"
     create_model "Foo", :bar => :references, :qux => :references do
       belongs_to :bar
       belongs_to :qux
@@ -88,11 +88,11 @@
     create_model "Bar", :baz => :references do
       belongs_to :baz
     end
     assert_equal [Domain::Relationship] * 3, Domain.generate.relationships.collect(&:class)
   end
-  
+
   test "relationships should count mutual relationship as one" do
     create_model "Foo", :bar => :references do
       belongs_to :bar
     end
     create_model "Bar" do
@@ -114,21 +114,21 @@
       belongs_to :wizard
       belongs_to :spell
     end
     assert_equal [Domain::Relationship], Domain.generate.relationships.select(&:indirect?).collect(&:class)
   end
-  
+
   test "relationships should count relationship between same models with distinct foreign key seperately" do
     create_model "Foo", :bar => :references, :special_bar => :references do
       belongs_to :bar
     end
     create_model "Bar" do
       has_many :foos, :foreign_key => :special_bar_id
     end
     assert_equal [Domain::Relationship] * 2, Domain.generate.relationships.collect(&:class)
   end
-  
+
   test "relationships should use model name first in alphabet as source for many to many relationships" do
     create_table "many_more", :many_id => :integer, :more_id => :integer
     create_model "Many" do
       has_and_belongs_to_many :more
     end
@@ -136,11 +136,11 @@
       has_and_belongs_to_many :many
     end
     relationship = Domain.generate.relationships.first
     assert_equal ["Many", "More"], [relationship.source.name, relationship.destination.name]
   end
-  
+
   # Specialization processing ================================================
   test "specializations should return empty array for empty domain" do
     assert_equal [], Domain.generate.specializations
   end
 
@@ -157,11 +157,11 @@
   test "specializations should return specializations of specializations in domain model" do
     create_specialization
     Object.const_set :BelgianBeer, Class.new(Beer)
     assert_equal [Domain::Specialization] * 2, Domain.generate.specializations.collect(&:class)
   end
-  
+
   test "specializations should return generalizations in domain model" do
     create_model "Post" do
       has_many :assets, :as => :attachable
     end
     create_model "Asset", :attachable => :references do
@@ -178,37 +178,37 @@
     create_model "Asset", :attachable => :references do
       belongs_to :attachable, :polymorphic => true
     end
     assert_equal [Domain::Specialization] * 2, Domain.generate.specializations.collect(&:class)
   end
-  
+
   # test "generalizations should ..." do
   #   # TODO
   #   create_model "Post" do
   #     has_many :assets, :as => :attachable
   #   end
   #   create_model "Asset", :attachable => :references do
   #     belongs_to :attachable, :polymorphic => true
   #   end
   #   assert_equal [], Domain.generate.relationships
   # end
-  
+
   # Erroneous associations ===================================================
   test "relationships should omit bad has_many associations" do
     create_model "Foo" do
       has_many :flabs
     end
     assert_equal [], Domain.generate(:warn => false).relationships
   end
-  
+
   test "relationships should omit bad has_many through association" do
     create_model "Foo" do
       has_many :flabs, :through => :bars
     end
     assert_equal [], Domain.generate(:warn => false).relationships
   end
-  
+
   test "relationships should omit association to model outside domain" do
     create_model "Foo" do
       has_many :bars
     end
     create_model "Bar", :foo => :references
@@ -256,10 +256,10 @@
     output = collect_stdout do
       Domain.generate(:warn => false).relationships
     end
     assert_equal "", output
   end
-  
+
   # Erroneous models =========================================================
   test "entities should omit bad models" do
     Object.const_set :Foo, Class.new(ActiveRecord::Base)
     assert_equal [], Domain.generate(:warn => false).entities
   end