Merge 2620330 into 74f79d3

lib/xcake.rb

@@ -1,5 +1,5 @@
 require 'xcake/core_ext/array'
-require 'xcake/core_ext/object'
+require 'xcake/core_ext/class'
 require 'xcake/core_ext/string'
 
 require 'xcake/xcode/project'

lib/xcake/core_ext/class.rb

@@ -0,0 +1,9 @@
+module Xcake
+  module CoreExtensions
+    # Returns all descendants of a class
+    #
+    def descendants
+      ObjectSpace.each_object(singleton_class).select { |klass| klass < self }
+    end
+  end
+end

lib/xcake/generator.rb

@@ -3,6 +3,7 @@ class Generator
     include Dependency
     include Plugin
     include Visitor
+    Class.include CoreExtensions
 
     attr_accessor :context
 

spec/core_ext/object_spec.rb → spec/core_ext/class_spec.rb

@@ -1,6 +1,7 @@
 require 'spec_helper'
 
 class Parent
+  Class.include Xcake::CoreExtensions
 end
 
 class Child < Parent
@@ -9,10 +10,11 @@ class Child < Parent
 class Grandchild < Child
 end
 
-describe Object do
+describe Xcake::CoreExtensions do
   it 'should return all descendants' do
     descendants = Parent.descendants
 
+    expect(descendants.count).to eq(2)
     expect(descendants).to include(Child)
     expect(descendants).to include(Grandchild)
   end