diff options
Diffstat (limited to 'activesupport')
-rw-r--r-- | activesupport/lib/core_ext/object_and_class.rb | 24 | ||||
-rw-r--r-- | activesupport/test/core_ext/object_and_class_ext_test.rb | 21 |
2 files changed, 45 insertions, 0 deletions
diff --git a/activesupport/lib/core_ext/object_and_class.rb b/activesupport/lib/core_ext/object_and_class.rb new file mode 100644 index 0000000000..59a463ae29 --- /dev/null +++ b/activesupport/lib/core_ext/object_and_class.rb @@ -0,0 +1,24 @@ +class Object #:nodoc: + def remove_subclasses_of(superclass) + subclasses_of(superclass).each { |subclass| Object.send(:remove_const, subclass) rescue nil } + end + + def subclasses_of(superclass) + subclasses = [] + ObjectSpace.each_object(Class) do |k| + next if !k.ancestors.include?(superclass) || superclass == k || k.to_s.include?("::") || subclasses.include?(k.to_s) + subclasses << k.to_s + end + subclasses + end +end + +class Class #:nodoc: + def remove_subclasses + Object.remove_subclasses_of(self) + end + + def subclasses + Object.subclasses_of(self) + end +end diff --git a/activesupport/test/core_ext/object_and_class_ext_test.rb b/activesupport/test/core_ext/object_and_class_ext_test.rb new file mode 100644 index 0000000000..5a0ae58837 --- /dev/null +++ b/activesupport/test/core_ext/object_and_class_ext_test.rb @@ -0,0 +1,21 @@ +require 'test/unit' +require File.dirname(__FILE__) + '/../../lib/core_ext/object_and_class' + +class A; end +class B < A; end +class C < B; end +class D < A; end + +class ClassExtTest < Test::Unit::TestCase + def test_methods + assert defined?(B) + assert defined?(C) + assert defined?(D) + + A.remove_subclasses + + assert !defined?(B) + assert !defined?(C) + assert !defined?(D) + end +end |