diff options
author | Jeremy Kemper <jeremy@bitsweat.net> | 2008-01-05 13:34:15 +0000 |
---|---|---|
committer | Jeremy Kemper <jeremy@bitsweat.net> | 2008-01-05 13:34:15 +0000 |
commit | 139b92495fa7697cdd619c549d4c7b263562b761 (patch) | |
tree | 20589f9f23d3b19697fed12bb1bf5ce48ef595af /activesupport/lib/active_support/testing/setup_and_teardown.rb | |
parent | fe66397adfb5a8057db78afcabd4d7eb0f13a783 (diff) | |
download | rails-139b92495fa7697cdd619c549d4c7b263562b761.tar.gz rails-139b92495fa7697cdd619c549d4c7b263562b761.tar.bz2 rails-139b92495fa7697cdd619c549d4c7b263562b761.zip |
* Continue evolution toward ActiveSupport::TestCase and friends. #10679 [Josh Peek]
* TestCase: introduce declared setup and teardown callbacks. Pass a list of methods and an optional block to call before setup or after teardown. Setup callbacks are run in the order declared; teardown callbacks are run in reverse. [Jeremy Kemper]
git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@8570 5ecf4fe2-1ee6-0310-87b1-e25e094e27de
Diffstat (limited to 'activesupport/lib/active_support/testing/setup_and_teardown.rb')
-rw-r--r-- | activesupport/lib/active_support/testing/setup_and_teardown.rb | 127 |
1 files changed, 127 insertions, 0 deletions
diff --git a/activesupport/lib/active_support/testing/setup_and_teardown.rb b/activesupport/lib/active_support/testing/setup_and_teardown.rb new file mode 100644 index 0000000000..1639462fae --- /dev/null +++ b/activesupport/lib/active_support/testing/setup_and_teardown.rb @@ -0,0 +1,127 @@ +module ActiveSupport + module Testing + module SetupAndTeardown + def self.included(base) + base.extend ClassMethods + + begin + require 'mocha' + base.alias_method_chain :run, :callbacks_and_mocha + rescue LoadError + base.alias_method_chain :run, :callbacks + end + end + + module ClassMethods + def setup(*method_names, &block) + method_names << block if block_given? + (@setup_callbacks ||= []).concat method_names + end + + def teardown(*method_names, &block) + method_names << block if block_given? + (@teardown_callbacks ||= []).concat method_names + end + + def setup_callback_chain + @setup_callbacks ||= [] + + if superclass.respond_to?(:setup_callback_chain) + superclass.setup_callback_chain + @setup_callbacks + else + @setup_callbacks + end + end + + def teardown_callback_chain + @teardown_callbacks ||= [] + + if superclass.respond_to?(:teardown_callback_chain) + superclass.teardown_callback_chain + @teardown_callbacks + else + @teardown_callbacks + end + end + end + + # This redefinition is unfortunate but test/unit shows us no alternative. + def run_with_callbacks(result) #:nodoc: + return if @method_name.to_s == "default_test" + + yield(Test::Unit::TestCase::STARTED, name) + @_result = result + begin + run_callbacks :setup + setup + __send__(@method_name) + rescue Test::Unit::AssertionFailedError => e + add_failure(e.message, e.backtrace) + rescue *Test::Unit::TestCase::PASSTHROUGH_EXCEPTIONS + raise + rescue Exception + add_error($!) + ensure + begin + teardown + run_callbacks :teardown, :reverse_each + rescue Test::Unit::AssertionFailedError => e + add_failure(e.message, e.backtrace) + rescue *Test::Unit::TestCase::PASSTHROUGH_EXCEPTIONS + raise + rescue Exception + add_error($!) + end + end + result.add_run + yield(Test::Unit::TestCase::FINISHED, name) + end + + # Doubly unfortunate: mocha does the same so we have to hax their hax. + def run_with_callbacks_and_mocha(result) + return if @method_name.to_s == "default_test" + + yield(Test::Unit::TestCase::STARTED, name) + @_result = result + begin + mocha_setup + begin + run_callbacks :setup + setup + __send__(@method_name) + mocha_verify { add_assertion } + rescue Mocha::ExpectationError => e + add_failure(e.message, e.backtrace) + rescue Test::Unit::AssertionFailedError => e + add_failure(e.message, e.backtrace) + rescue StandardError, ScriptError + add_error($!) + ensure + begin + teardown + run_callbacks :teardown, :reverse_each + rescue Test::Unit::AssertionFailedError => e + add_failure(e.message, e.backtrace) + rescue StandardError, ScriptError + add_error($!) + end + end + ensure + mocha_teardown + end + result.add_run + yield(Test::Unit::TestCase::FINISHED, name) + end + + protected + def run_callbacks(kind, enumerator = :each) + self.class.send("#{kind}_callback_chain").send(enumerator) do |callback| + case callback + when Proc; callback.call(self) + when String, Symbol; send!(callback) + else raise ArgumentError, "Unrecognized callback #{callback.inspect}" + end + end + end + end + end +end |