From 4f375d5b82705db9aeb652d2a562c2df369ca4ac Mon Sep 17 00:00:00 2001
From: Jeremy Kemper <jeremy@bitsweat.net>
Date: Tue, 18 Sep 2007 01:45:37 +0000
Subject: Support multiple config.after_initialize blocks so plugins and apps
 can more easily cooperate. Closes #9582.

git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@7507 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
---
 railties/test/initializer_test.rb | 76 +++++++++++++++++++++++++++++++++++----
 1 file changed, 69 insertions(+), 7 deletions(-)

(limited to 'railties/test')

diff --git a/railties/test/initializer_test.rb b/railties/test/initializer_test.rb
index cf333e878e..c474f0bc05 100644
--- a/railties/test/initializer_test.rb
+++ b/railties/test/initializer_test.rb
@@ -1,15 +1,16 @@
 require "#{File.dirname(__FILE__)}/abstract_unit"
 require 'initializer'
 
-class InitializerTest < Test::Unit::TestCase
-  class ConfigurationMock < Rails::Configuration
-    attr_reader :environment_path
+class ConfigurationMock < Rails::Configuration
+  attr_reader :environment_path
 
-    def initialize(envpath)
-      super()
-      @environment_path = envpath
-    end
+  def initialize(envpath)
+    super()
+    @environment_path = envpath
   end
+end
+
+class Initializer_load_environment_Test < Test::Unit::TestCase
 
   def test_load_environment_with_constant
     config = ConfigurationMock.new("#{File.dirname(__FILE__)}/fixtures/environment_with_constant.rb")
@@ -19,4 +20,65 @@ class InitializerTest < Test::Unit::TestCase
   ensure
     $initialize_test_set_from_env = nil
   end
+
+end
+
+class Initializer_after_initialize_with_blocks_environment_Test < Test::Unit::TestCase
+  def setup
+    config = ConfigurationMock.new("")
+    config.after_initialize do
+      $test_after_initialize_block1 = "success"
+    end
+    config.after_initialize do
+      $test_after_initialize_block2 = "congratulations"
+    end    
+    assert_nil $test_after_initialize_block1
+    assert_nil $test_after_initialize_block2    
+
+    Rails::Initializer.run(:after_initialize, config)
+  end
+  
+  def teardown
+    $test_after_initialize_block1 = nil
+    $test_after_initialize_block2 = nil    
+  end
+
+  def test_should_have_called_the_first_after_initialize_block
+    assert_equal "success", $test_after_initialize_block1
+  end
+  
+  def test_should_have_called_the_second_after_initialize_block
+    assert_equal "congratulations", $test_after_initialize_block2
+  end
+end
+  
+class Initializer_after_initialize_with_no_block_environment_Test < Test::Unit::TestCase
+
+  def setup
+    config = ConfigurationMock.new("")
+    config.after_initialize do
+      $test_after_initialize_block1 = "success"
+    end
+    config.after_initialize # don't pass a block, this is what we're testing!
+    config.after_initialize do
+      $test_after_initialize_block2 = "congratulations"
+    end    
+    assert_nil $test_after_initialize_block1
+
+    Rails::Initializer.run(:after_initialize, config)
+  end
+
+  def teardown
+    $test_after_initialize_block1 = nil
+    $test_after_initialize_block2 = nil    
+  end
+
+  def test_should_have_called_the_first_after_initialize_block
+    assert_equal "success", $test_after_initialize_block1, "should still get set"
+  end
+
+  def test_should_have_called_the_first_after_initialize_block
+    assert_equal "congratulations", $test_after_initialize_block2
+  end
+
 end
-- 
cgit v1.2.3