diff --git a/app/models/enumeration.rb b/app/models/enumeration.rb index 826955c..d532c78 100644 --- a/app/models/enumeration.rb +++ b/app/models/enumeration.rb @@ -98,7 +98,7 @@ class Enumeration < ActiveRecord::Base # # Note: subclasses is protected in ActiveRecord def self.get_subclasses - @@subclasses[Enumeration] + subclasses end # Does the +new+ Hash override the previous Enumeration? diff --git a/test/unit/enumeration_test.rb b/test/unit/enumeration_test.rb index 5c03dc2..56be566 100644 --- a/test/unit/enumeration_test.rb +++ b/test/unit/enumeration_test.rb @@ -108,4 +108,15 @@ class EnumerationTest < ActiveSupport::TestCase enumeration.parent = Enumeration.find(5) assert enumeration.is_override? end + + def test_get_subclasses + classes = Enumeration.get_subclasses + assert_include IssuePriority, classes + assert_include DocumentCategory, classes + assert_include TimeEntryActivity, classes + + classes.each do |klass| + assert_equal Enumeration, klass.superclass + end + end end