aboutsummaryrefslogblamecommitdiffstats
path: root/activesupport/lib/active_support/testing/setup_and_teardown.rb
blob: 18de7185d980847efd08937cf067ffb5db46faeb (plain) (tree)
1
2
3
4
5
6
7
                             
 

                               
 

                    












                                                                      
                           
                                   
 


                                          


                         
                                                                    



                                                      
                                                                      

                                                        


           
                                

                            
         
 
                                  


                               


       
# frozen_string_literal: true

require_relative "../concern"
require_relative "../callbacks"

module ActiveSupport
  module Testing
    # Adds support for +setup+ and +teardown+ callbacks.
    # These callbacks serve as a replacement to overwriting the
    # <tt>#setup</tt> and <tt>#teardown</tt> methods of your TestCase.
    #
    #   class ExampleTest < ActiveSupport::TestCase
    #     setup do
    #       # ...
    #     end
    #
    #     teardown do
    #       # ...
    #     end
    #   end
    module SetupAndTeardown
      extend ActiveSupport::Concern

      included do
        include ActiveSupport::Callbacks
        define_callbacks :setup, :teardown
      end

      module ClassMethods
        # Add a callback, which runs before <tt>TestCase#setup</tt>.
        def setup(*args, &block)
          set_callback(:setup, :before, *args, &block)
        end

        # Add a callback, which runs after <tt>TestCase#teardown</tt>.
        def teardown(*args, &block)
          set_callback(:teardown, :after, *args, &block)
        end
      end

      def before_setup # :nodoc:
        super
        run_callbacks :setup
      end

      def after_teardown # :nodoc:
        run_callbacks :teardown
        super
      end
    end
  end
end