browsermob-proxy 0.0.3 → 0.0.4

data/Rakefile

@@ -2,6 +2,20 @@ require 'bundler'
 Bundler::GemHelper.install_tasks
 
 require 'rspec/core/rake_task'
-RSpec::Core::RakeTask.new
+
+namespace :spec do
+  desc 'Run unit specs.'
+  RSpec::Core::RakeTask.new(:unit) do |t|
+    t.pattern = "spec/unit/*_spec.rb"
+  end
+
+  desc 'Run end to end specs.'
+  RSpec::Core::RakeTask.new(:e2e) do |t|
+    t.pattern = "spec/e2e/*_spec.rb"
+  end
+end
+
+desc 'Run all specs.'
+task :spec => %w[spec:unit spec:e2e]
 
 task :default => :spec

data/lib/browsermob/proxy/client.rb

@@ -41,18 +41,42 @@ module BrowserMob
 
       def whitelist(regexp, status_code)
         regex = Regexp === regexp ? regexp.source : regexp.to_s
-        @resource['har/whitelist'].put :regex => regex, :status => status_code
+        @resource['whitelist'].put :regex => regex, :status => status_code
       end
 
       def blacklist(regexp, status_code)
         regex = Regexp === regexp ? regexp.source : regexp.to_s
-        @resource['har/blacklist'].put :regex => regex, :status => status_code
+        @resource['blacklist'].put :regex => regex, :status => status_code
+      end
+
+      LIMITS = {
+        :upstream_kbps   => 'upstreamKbps',
+        :downstream_kbps => 'downstreamKbps',
+        :latency         => 'latency'
+      }
+
+      def limit(opts)
+        params = {}
+
+        opts.each do |key, value|
+          unless LIMITS.member?(key)
+            raise ArgumentError, "invalid: #{key.inspect} (valid options: #{LIMITS.keys.inspect})"
+          end
+
+          params[LIMITS[key]] = Integer(value)
+        end
+
+        if params.empty?
+          raise ArgumentError, "must specify one of #{LIMITS.keys.inspect}"
+        end
+
+        @resource['limit'].put params
       end
 
       def close
         @resource.delete
       end
-    end
+    end # Client
 
-  end
-end
+  end # Proxy
+end # BrowserMob

data/lib/browsermob/proxy/version.rb

@@ -1,5 +1,5 @@
 module BrowserMob
   module Proxy
-    VERSION = "0.0.3"
+    VERSION = "0.0.4"
   end
 end

data/spec/e2e/listener_spec.rb

@@ -1,10 +1,11 @@
 require 'spec_helper'
 
 describe "Proxy + WebDriverListener" do
-  let(:wait) { Selenium::WebDriver::Wait.new }
+  let(:proxy) { new_proxy }
   let(:listener) { BrowserMob::Proxy::WebDriverListener.new(proxy) }
+
   let(:driver)  { Selenium::WebDriver.for :firefox, :profile => profile, :listener => listener }
-  let(:proxy) { new_proxy }
+  let(:wait) { Selenium::WebDriver::Wait.new }
 
   let(:profile) {
     pr = Selenium::WebDriver::Firefox::Profile.new
@@ -13,9 +14,7 @@ describe "Proxy + WebDriverListener" do
     pr
   }
 
-  after {
-    proxy.close
-  }
+  after { proxy.close }
 
   it "should record events" do
     driver.get url_for("1.html")
@@ -26,7 +25,5 @@ describe "Proxy + WebDriverListener" do
     hars.size.should == 1
 
     hars.first.pages.size.should == 2
-
-    hars.each_with_index { |e,i| e.save_to("/tmp/#{i}.har") }
   end
 end

data/spec/e2e/selenium_spec.rb

@@ -28,4 +28,14 @@ describe "Proxy + WebDriver" do
     har.should be_kind_of(HAR::Archive)
     har.pages.size.should == 2
   end
+
+  it "should set whitelist and blacklist" do
+    proxy.whitelist(/example\.com/, 201)
+    proxy.blacklist(/bad\.com/, 404)
+  end
+
+  it "should set limits" do
+    proxy.limit(:downstream_kbps => 100, :upstream_kbps => 100, :latency => 2)
+  end
+
 end

data/spec/unit/client_spec.rb

@@ -9,10 +9,11 @@ module BrowserMob
 
       before do
         {
-          "har"           => mock("resource[har]"),
-          "har/pageRef"   => mock("resource[har/pageRef]"),
-          "har/whitelist" => mock("resource[har/whitelist]"),
-          "har/blacklist" => mock("resource[har/blacklist]")
+          "har"         => mock("resource[har]"),
+          "har/pageRef" => mock("resource[har/pageRef]"),
+          "whitelist"   => mock("resource[whitelist]"),
+          "blacklist"   => mock("resource[blacklist]"),
+          "limit"       => mock("resource[limit]")
         }.each do |path, mock|
           resource.stub!(:[]).with(path).and_return(mock)
         end
@@ -49,18 +50,51 @@ module BrowserMob
       end
 
       it "sets the blacklist" do
-        resource['har/blacklist'].should_receive(:put).
-                                  with(:regex => "http://example.com", :status => 401)
+        resource['blacklist'].should_receive(:put).
+                              with(:regex => "http://example.com", :status => 401)
 
         client.blacklist(%r[http://example.com], 401)
       end
 
       it "sets the whitelist" do
-        resource['har/whitelist'].should_receive(:put).
-                                  with(:regex => "http://example.com", :status => 401)
+        resource['whitelist'].should_receive(:put).
+                              with(:regex => "http://example.com", :status => 401)
 
         client.whitelist(%r[http://example.com], 401)
       end
+
+      it "sets the :downstream_kbps limit" do
+        resource['limit'].should_receive(:put).
+                          with('downstreamKbps' => 100)
+
+        client.limit(:downstream_kbps => 100)
+      end
+
+      it "sets the :upstream_kbps limit" do
+        resource['limit'].should_receive(:put).
+                          with('upstreamKbps' => 100)
+
+        client.limit(:upstream_kbps => 100)
+      end
+
+      it "sets the :latency limit" do
+        resource['limit'].should_receive(:put).
+                          with('latency' => 100)
+
+        client.limit(:latency => 100)
+      end
+
+      it "sets all limits" do
+        resource['limit'].should_receive(:put).
+                          with('latency' => 100, 'downstreamKbps' => 200, 'upstreamKbps' => 300)
+
+        client.limit(:latency => 100, :downstream_kbps => 200, :upstream_kbps => 300)
+      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
     end
 
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: browsermob-proxy
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -13,7 +13,7 @@ date: 2011-10-09 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rest-client
-  requirement: &2161286300 !ruby/object:Gem::Requirement
+  requirement: &2153771760 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2161286300
+  version_requirements: *2153771760
 - !ruby/object:Gem::Dependency
   name: childprocess
-  requirement: &2161279940 !ruby/object:Gem::Requirement
+  requirement: &2153770220 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2161279940
+  version_requirements: *2153770220
 - !ruby/object:Gem::Dependency
   name: json
-  requirement: &2161263620 !ruby/object:Gem::Requirement
+  requirement: &2153768980 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2161263620
+  version_requirements: *2153768980
 - !ruby/object:Gem::Dependency
   name: har
-  requirement: &2161262300 !ruby/object:Gem::Requirement
+  requirement: &2153767740 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *2161262300
+  version_requirements: *2153767740
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &2161259040 !ruby/object:Gem::Requirement
+  requirement: &2153751700 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,10 +65,10 @@ dependencies:
         version: 2.5.0
   type: :development
   prerelease: false
-  version_requirements: *2161259040
+  version_requirements: *2153751700
 - !ruby/object:Gem::Dependency
   name: selenium-webdriver
-  requirement: &2161253700 !ruby/object:Gem::Requirement
+  requirement: &2153750140 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ~>
@@ -76,7 +76,7 @@ dependencies:
         version: '2.7'
   type: :development
   prerelease: false
-  version_requirements: *2161253700
+  version_requirements: *2153750140
 description: Ruby client for the BrowserMob Proxy REST API
 email:
 - jari.bakken@gmail.com