test for staging env acceptability

Author: NewAlexandria

.gitignore

@@ -12,3 +12,5 @@
 /test/dummy/tmp/development_secret.txt
 
 .byebug_history
+
+.DS_Store

app/controllers/rake_ui/application_controller.rb

@@ -4,10 +4,13 @@ module RakeUi
   class ApplicationController < ActionController::Base
     before_action :black_hole_production
 
+    STAGING_OK = (Rails.env.staging? && RakeUi.configuration.allow_staging)
+    PROD_OK = RakeUi.configuration.allow_production
+
     private
 
     def black_hole_production
-      return if Rails.env.test? || Rails.env.development? || RakeUi.configuration.allow_production
+      return if Rails.env.test? || Rails.env.development? || STAGING_OK || PROD_OK
 
       raise ActionController::RoutingError, "Not Found"
     end

lib/rake-ui.rb

@@ -4,7 +4,10 @@
 
 module RakeUi
   mattr_accessor :allow_production
+  mattr_accessor :allow_staging
+
   self.allow_production = false
+  self.allow_staging = true
 
   def self.configuration
     yield(self) if block_given?