test/unit/diagram_test.rb in rails-erd-1.5.2 vs test/unit/diagram_test.rb in rails-erd-1.6.0

@@ -134,9 +134,18 @@
     create_model "Author"
     create_model "Editor"
     assert_equal [Book], retrieve_entities(:exclude => [:Author, :Editor]).map(&:model)
   end
 
+  test "generate should filter excluded polymorphic entities" do
+    create_model "Cannon"
+    create_model "Galleon" do
+      has_many :cannons, as: :defensible
+    end
+    assert_equal ["Cannon", "Galleon"], retrieve_entities(polymorphism: true, exclude: :Defensible).map(&:name)
+  end
+
+
   test "generate should include only specified entity" do
     create_model "Book"
     create_model "Author"
     assert_equal [Book], retrieve_entities(:only => [:Book]).map(&:model)
   end