spec/rack/dev-mark/railtie_spec.rb in rack-dev-mark-0.6.4 vs spec/rack/dev-mark/railtie_spec.rb in rack-dev-mark-0.7.0

@@ -37,22 +37,22 @@
         expect(@app.middleware.middlewares).to include(Rack::DevMark::Middleware)
       end
     end
   end
   context "rack_dev_mark theme" do
-    let(:theme) { d = double setup: nil; allow(d).to receive(:insert_into){ |b| "#{b} dev-mark" }; d }
+    let(:theme) { d = double setup: nil; allow(d).to receive(:insert_into){ |b, _, _| "#{b} dev-mark" }; d }
     before do
       @app.config.rack_dev_mark.enable = true
       @app.config.rack_dev_mark.theme = [theme]
     end
     it 'inserts the middleware' do
       @app.initialize!
       expect(theme).to receive(:setup)
     end
   end
   context "rack_dev_mark custom_theme alias" do
-    let(:theme) { d = double setup: nil; allow(d).to receive(:insert_into){ |b| "#{b} dev-mark" }; d }
+    let(:theme) { d = double setup: nil; allow(d).to receive(:insert_into){ |b, _, _| "#{b} dev-mark" }; d }
     before do
       @app.config.rack_dev_mark.enable = true
       @app.config.rack_dev_mark.custom_theme = [theme]
     end
     it 'inserts the middleware' do
@@ -86,9 +86,19 @@
       @app.initialize!
       middlewares = @app.middleware.middlewares
       expect(middlewares).to include(Rack::DevMark::Middleware)
       expect(middlewares).to include(dummy_middleware)
       expect(middlewares.index(Rack::DevMark::Middleware)).to eq(middlewares.index(dummy_middleware) + 1)
+    end
+  end
+  context "rack_dev_mark env" do
+    before do
+      @app.config.rack_dev_mark.enable = true
+      @app.config.rack_dev_mark.env = 'something'
+    end
+    it 'sets the env string' do
+      @app.initialize!
+      expect(Rack::DevMark.env).to eq('something')
     end
   end
 end
 end