mcmire · October 16, 2021 11:09 · Apr 24, 2017 · Apr 24, 2017 · Apr 24, 2017
diff --git a/bypass_broken_images_middleware.rb b/bypass_broken_images_middleware.rb
@@ -2,7 +2,7 @@
 # ------------
 #
 # * Save this as app/middlewares/bypass_broken_images_middleware.rb
-# * Add the following line inside of the Rails.application.configure block
+# * Add the following inside of the Rails.application.configure block
 #   in config/environments/test.rb:
 #
 #     config.middleware.insert_before(

diff --git a/bypass_broken_images_middleware.rb b/bypass_broken_images_middleware.rb
@@ -25,6 +25,8 @@ def call(env)
       raise error
     end
   end
+
+  private
 
   def unknown_path?(error)
     error.is_a?(ActionController::RoutingError)

diff --git a/bypass_broken_images_middleware.rb b/bypass_broken_images_middleware.rb
@@ -0,0 +1,37 @@
+# Instructions
+# ------------
+#
+# * Save this as app/middlewares/bypass_broken_images_middleware.rb
+# * Add the following line inside of the Rails.application.configure block
+#   in config/environments/test.rb:
+#
+#     config.middleware.insert_before(
+#       ActionDispatch::DebugExceptions,
+#       BypassBrokenImagesMiddleware,
+#     )
+
+class BypassBrokenImagesMiddleware
+  def initialize(app)
+    @app = app
+  end
+
+  def call(env)
+    @app.(env)
+  rescue => error
+    if unknown_path?(error) && request_for_image?(env)
+      # nothing to see here, move along
+      [404, {}, ""]
+    else
+      raise error
+    end
+  end
+
+  def unknown_path?(error)
+    error.is_a?(ActionController::RoutingError)
+  end
+
+  def request_for_image?(env)
+    env["PATH_INFO"] =~ %r{\A/(assets|images)/} ||
+      env["PATH_INFO"] =~ /\.(jpg|jpeg|png)\Z/
+  end
+end