Merge pull request #24 from sharethrough/master

Add support for rewrite rules

Author: jarib

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/e2e/selenium_spec.rb

@@ -128,6 +128,29 @@
     end
   end
 
+  describe 'rewrite rules' do
+
+    let(:uri) { URI.parse url_for('1.html') }
+
+    before do
+      proxy.rewrite(%r{1\.html}, '2.html')
+    end
+
+    it 'fetches the rewritten url' do
+      driver.get uri
+
+      wait.until { driver.title == '2' }
+    end
+
+    it 'can be cleared' do
+      proxy.clear_rewrites
+      driver.get uri
+
+      wait.until { driver.title == '1' }
+    end
+
+  end
+
   it 'should set timeouts' do
     proxy.timeouts(read: 0.001)
     driver.get url_for('slow')

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,34 @@ module Proxy
         client.remap_dns_hosts(DOMAIN => '1.2.3.4')
       end
 
+      describe 'rewrite rules' do
+
+        context 'when using a regular expression' do
+          it 'sets a rewrite rule' do
+            resource['rewrite'].should_receive(:put).
+              with(:matchRegex => 'old\.com', :replace => 'new.com')
+
+            client.rewrite('old\.com', 'new.com')
+          end
+        end
+
+        context 'when using a string' do
+          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
+        end
+
+        it 'clears the rewrite rules' do
+          resource['rewrite'].should_receive(:delete)
+
+          client.clear_rewrites
+        end
+      end
+
+
       context "#selenium_proxy" do
         it "defaults to HTTP proxy only" do
           proxy = client.selenium_proxy