rack_spec.rb in bullet-2.3.0

- old
+ new
@@ -1,40 +1,96 @@
-require File.dirname(__FILE__) + '/../spec_helper'
+require 'spec_helper'
 
+module Bullet
+  describe Rack do
+    let(:middleware) { Bullet::Rack.new app }
+    let(:app) { Support::AppDouble.new }
 
-describe Bullet::Rack do
-  let(:middleware) { Bullet::Rack.new app }
-  let(:app) { AppDouble.new }
+    context "#no_browser_cache" do
+      it "should add no cache meta in http headers" do
+        headers = {}
+        middleware.no_browser_cache(headers)
+        headers["Cache-Control"].should == "no-cache, no-store, max-age=0, must-revalidate"
+        headers["Pragma"].should == "no-cache"
+        headers["Expires"].should == "Wed, 09 Sep 2009 09:09:09 GMT"
+      end
+    end
 
-  describe "#call" do
-    context "when Bullet is enabled" do
-      before(:each) { Bullet.enable = true }
+    context "#html_request?" do
+      it "should be true if Content-Type is text/html and http body contains html tag" do
+        headers = {"Content-Type" => "text/html"}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should be_html_request(headers, response)
+      end
 
-      it "should invoke Bullet.start_request" do
-        Bullet.should_receive(:start_request)
-        middleware.call([])
+      it "should be false if there is no Content-Type header" do
+        headers = {}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_html_request(headers, response)
       end
 
-      it "should invoke Bullet.end_request" do
-        Bullet.should_receive(:end_request)
-        middleware.call([])
+      it "should be false if Content-Type is javascript" do
+        headers = {"Content-Type" => "text/javascript"}
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_html_request(headers, response)
       end
 
-      it "should return original response body" do
-        expected_response = ResponseDouble.new "Actual body"
-        app.response = expected_response
-        status, headers, response = middleware.call([])
-        response.should eq expected_response
+      it "should be false if response body doesn't contain html tag" do
+        headers = {"Content-Type" => "text/html"}
+        response = stub(:body => "<div>Partial</div>")
+        middleware.should_not be_html_request(headers, response)
       end
     end
 
-    context "when Bullet is disabled" do
-      before(:each) { Bullet.enable = false }
-      after(:each) { Bullet.enable = true }
+    context "empty?" do
+      it "should be false if response is a string and not empty" do
+        response = stub(:body => "<html><head></head><body></body></html>")
+        middleware.should_not be_empty(response)
+      end
 
-      it "should not call Bullet.start_request" do
-        Bullet.should_not_receive(:start_request)
-        middleware.call([])
+      it "should be tru if response is not found" do
+        response = ["Not Found"]
+        middleware.should be_empty(response)
+      end
+
+      it "should be true if response body is empty" do
+        response = stub(:body => "")
+        middleware.should be_empty(response)
+      end
+    end
+
+    context "#call" do
+      context "when Bullet is enabled" do
+        it "should invoke Bullet.start_request and Bullet.end_request" do
+          Bullet.should_receive(:start_request)
+          Bullet.should_receive(:end_request)
+          middleware.call([])
+        end
+
+        it "should return original response body" do
+          expected_response = Support::ResponseDouble.new "Actual body"
+          app.response = expected_response
+          status, headers, response = middleware.call([])
+          response.should == expected_response
+        end
+
+        it "should change response body if notification is active" do
+          Bullet.should_receive(:notification?).and_return(true)
+          Bullet.should_receive(:gather_inline_notifications).and_return("<bullet></bullet>")
+          Bullet.should_receive(:perform_out_of_channel_notifications)
+          status, headers, response = middleware.call([200, {"Content-Type" => "text/html"}])
+          headers["Content-Length"].should == "56"
+          response.should == ["<html><head></head><body></body></html><bullet></bullet>"]
+        end
+      end
+
+      context "when Bullet is disabled" do
+        before(:each) { Bullet.stub(:enable?, false) }
+
+        it "should not call Bullet.start_request" do
+          Bullet.should_not_receive(:start_request)
+          middleware.call([])
+        end
       end
     end
   end
 end