1 files changed, 238 insertions, 0 deletions
diff --git a/spec/ruby/core/module/const_added_spec.rb b/spec/ruby/core/module/const_added_spec.rb
new file mode 100644
index 0000000000..90cd36551a
--- /dev/null
+++ b/spec/ruby/core/module/const_added_spec.rb
@@ -0,0 +1,238 @@
+require_relative '../../spec_helper'
+require_relative 'fixtures/classes'
+require_relative 'fixtures/const_added'
+
+describe "Module#const_added" do
+  it "is a private instance method" do
+    Module.should have_private_instance_method(:const_added)
+  end
+
+  it "returns nil in the default implementation" do
+    Module.new do
+      const_added(:TEST).should == nil
+    end
+  end
+
+  it "for a class defined with the `class` keyword, const_added runs before inherited" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(_)
+        ScratchPad << :const_added
+      end
+    end
+
+    parent = Class.new do
+      def self.inherited(_)
+        ScratchPad << :inherited
+      end
+    end
+
+    class mod::C < parent; end
+
+    ScratchPad.recorded.should == [:const_added, :inherited]
+  end
+
+  it "the superclass of a class assigned to a constant is set before const_added is called" do
+    ScratchPad.record []
+
+    parent = Class.new do
+      def self.const_added(name)
+        ScratchPad << name
+        ScratchPad << const_get(name).superclass
+      end
+    end
+
+    class parent::C < parent; end
+
+    ScratchPad.recorded.should == [:C, parent]
+  end
+
+  it "is called when a new constant is assigned on self" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+    end
+
+    mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+      TEST = 1
+    RUBY
+
+    ScratchPad.recorded.should == [:TEST]
+  end
+
+  it "is called when a new constant is assigned on self through const_set" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+    end
+
+    mod.const_set(:TEST, 1)
+
+    ScratchPad.recorded.should == [:TEST]
+  end
+
+  it "is called when a new module is defined under self" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+    end
+
+    mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+      module SubModule
+      end
+
+      module SubModule
+      end
+    RUBY
+
+    ScratchPad.recorded.should == [:SubModule]
+  end
+
+  it "is called when a new module is defined under a named module (assigned to a constant)" do
+    ScratchPad.record []
+
+    ModuleSpecs::ConstAddedSpecs::NamedModule = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+
+      module self::A
+        def self.const_added(name)
+          ScratchPad << name
+        end
+
+        module self::B
+        end
+      end
+    end
+
+    ScratchPad.recorded.should == [:A, :B]
+    ModuleSpecs::ConstAddedSpecs.send :remove_const, :NamedModule
+  end
+
+  it "is called when a new class is defined under self" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+    end
+
+    mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+      class SubClass
+      end
+
+      class SubClass
+      end
+    RUBY
+
+    ScratchPad.recorded.should == [:SubClass]
+  end
+
+  it "is called when a new class is defined under a named module (assigned to a constant)" do
+    ScratchPad.record []
+
+    ModuleSpecs::ConstAddedSpecs::NamedModuleB = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+
+      class self::A
+        def self.const_added(name)
+          ScratchPad << name
+        end
+
+        class self::B
+        end
+      end
+    end
+
+    ScratchPad.recorded.should == [:A, :B]
+    ModuleSpecs::ConstAddedSpecs.send :remove_const, :NamedModuleB
+  end
+
+  it "is called when an autoload is defined" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << name
+      end
+    end
+
+    mod.autoload :Autoload, "foo"
+    ScratchPad.recorded.should == [:Autoload]
+  end
+
+  it "is called with a precise caller location with the line of definition" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        location = caller_locations(1, 1)[0]
+        ScratchPad << location.lineno
+      end
+    end
+
+    line = __LINE__
+    mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+      TEST = 1
+
+      module SubModule
+      end
+
+      class SubClass
+      end
+    RUBY
+
+    mod.const_set(:CONST_SET, 1)
+
+    ScratchPad.recorded.should == [line + 2, line + 4, line + 7, line + 11]
+  end
+
+  it "is called when the constant is already assigned a value" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad.record const_get(name)
+      end
+    end
+
+    mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+      TEST = 123
+    RUBY
+
+    ScratchPad.recorded.should == 123
+  end
+
+  it "records re-definition of existing constants" do
+    ScratchPad.record []
+
+    mod = Module.new do
+      def self.const_added(name)
+        ScratchPad << const_get(name)
+      end
+    end
+
+    -> {
+      mod.module_eval(<<-RUBY, __FILE__, __LINE__ + 1)
+        TEST = 123
+        TEST = 456
+      RUBY
+    }.should complain(/warning: already initialized constant .+::TEST/)
+
+    ScratchPad.recorded.should == [123, 456]
+  end
+end