1 files changed, 26 insertions, 36 deletions
diff --git a/actionpack/test/abstract_unit.rb b/actionpack/test/abstract_unit.rb
index 56cebee678..b914bbce4d 100644
--- a/actionpack/test/abstract_unit.rb
+++ b/actionpack/test/abstract_unit.rb
@@ -85,39 +85,28 @@ module RenderERBUtils
   end
 end
 
-module SetupOnce
-  extend ActiveSupport::Concern
-
-  included do
-    cattr_accessor :setup_once_block
-    self.setup_once_block = nil
-
-    setup :run_setup_once
-  end
+SharedTestRoutes = ActionDispatch::Routing::RouteSet.new
 
-  module ClassMethods
-    def setup_once(&block)
-      self.setup_once_block = block
+module ActionDispatch
+  module SharedRoutes
+    def before_setup
+      @routes = SharedTestRoutes
+      super
     end
   end
 
-  private
-    def run_setup_once
-      if self.setup_once_block
-        self.setup_once_block.call
-        self.setup_once_block = nil
-      end
+  # Hold off drawing routes until all the possible controller classes
+  # have been loaded.
+  module DrawOnce
+    class << self
+      attr_accessor :drew
     end
-end
+    self.drew = false
 
-SharedTestRoutes = ActionDispatch::Routing::RouteSet.new
+    def before_setup
+      super
+      return if DrawOnce.drew
 
-module ActiveSupport
-  class TestCase
-    include SetupOnce
-    # Hold off drawing routes until all the possible controller classes
-    # have been loaded.
-    setup_once do
       SharedTestRoutes.draw do
         get ':controller(/:action)'
       end
@@ -125,10 +114,18 @@ module ActiveSupport
       ActionDispatch::IntegrationTest.app.routes.draw do
         get ':controller(/:action)'
       end
+
+      DrawOnce.drew = true
     end
   end
 end
 
+module ActiveSupport
+  class TestCase
+    include ActionDispatch::DrawOnce
+  end
+end
+
 class RoutedRackApp
   attr_reader :routes
 
@@ -159,9 +156,7 @@ class BasicController
 end
 
 class ActionDispatch::IntegrationTest < ActiveSupport::TestCase
-  setup do
-    @routes = SharedTestRoutes
-  end
+  include ActionDispatch::SharedRoutes
 
   def self.build_app(routes = nil)
     RoutedRackApp.new(routes || ActionDispatch::Routing::RouteSet.new) do |middleware|
@@ -290,10 +285,7 @@ module ActionController
 
   class TestCase
     include ActionDispatch::TestProcess
-
-    setup do
-      @routes = SharedTestRoutes
-    end
+    include ActionDispatch::SharedRoutes
   end
 end
 
@@ -304,9 +296,7 @@ module ActionView
   class TestCase
     # Must repeat the setup because AV::TestCase is a duplication
     # of AC::TestCase
-    setup do
-      @routes = SharedTestRoutes
-    end
+    include ActionDispatch::SharedRoutes
   end
 end