Add support for rewrite rules

Author: rslifka

lib/browsermob/proxy/client.rb

@@ -105,6 +105,15 @@ def clear_blacklist
         @resource['blacklist'].delete
       end
 
+      def rewrite(match_regex, replace)
+        regex = Regexp === match_regex ? match_regex.source : match_regex.to_s
+        @resource['rewrite'].put :matchRegex => regex, :replace => replace
+      end
+
+      def clear_rewrites
+        @resource['rewrite'].delete
+      end
+
       def header(hash)
         @resource['headers'].post hash.to_json, :content_type => "application/json"
       end

spec/unit/client_spec.rb

@@ -19,7 +19,8 @@ module Proxy
           "headers"              => double("resource[headers]"),
           "auth/basic/#{DOMAIN}" => double("resource[auth/basic/#{DOMAIN}]"),
           "hosts"                => double("resource[hosts]"),
-          "timeout"              => double("resource[timeout]")
+          "timeout"              => double("resource[timeout]"),
+          "rewrite"              => double("resource[rewrite]")
         }.each do |path, mock|
           resource.stub(:[]).with(path).and_return(mock)
         end
@@ -231,6 +232,19 @@ module Proxy
         client.remap_dns_hosts(DOMAIN => '1.2.3.4')
       end
 
+      it 'sets a rewrite rule' do
+        resource['rewrite'].should_receive(:put).
+          with(:matchRegex => 'old.com', :replace => 'new.com')
+
+        client.rewrite(%r[old.com], 'new.com')
+      end
+
+      it 'clears the rewrite rules' do
+        resource['rewrite'].should_receive(:delete)
+
+        client.clear_rewrites
+      end
+
       context "#selenium_proxy" do
         it "defaults to HTTP proxy only" do
           proxy = client.selenium_proxy