spec/unit/client_spec.rb in browsermob-proxy-0.0.5 vs spec/unit/client_spec.rb in browsermob-proxy-0.0.6

@@ -11,11 +11,12 @@
         {
           "har"         => mock("resource[har]"),
           "har/pageRef" => mock("resource[har/pageRef]"),
           "whitelist"   => mock("resource[whitelist]"),
           "blacklist"   => mock("resource[blacklist]"),
-          "limit"       => mock("resource[limit]")
+          "limit"       => mock("resource[limit]"),
+          "headers"     => mock("resource[headers]")
         }.each do |path, mock|
           resource.stub!(:[]).with(path).and_return(mock)
         end
       end
 
@@ -92,9 +93,15 @@
       end
 
       it "raises ArgumentError on invalid options" do
         lambda { client.limit(:foo => 1) }.should raise_error(ArgumentError)
         lambda { client.limit({})        }.should raise_error(ArgumentError)
+      end
+
+      it "sets headers" do
+        resource['headers'].should_receive(:post).with('{"foo":"bar"}', :content_type => "application/json")
+
+        client.headers(:foo => "bar")
       end
     end
 
   end
 end
\ No newline at end of file