Add ability to set proxy at runtime (needs test coverage)

CHANGELOG.md

@@ -1,5 +1,8 @@
 ### Next release ###
 
+#### Features ####
+*   Added ability to set proxy at runtime (Dmitry Vorotilin)
+
 ### 1.9.0 ###
 
 #### Features ####

README.md

@@ -110,6 +110,7 @@ and the following optional features:
 * `page.driver.scroll_to(left, top)`
 * `page.driver.basic_authorize(user, password)`
 * `element.native.send_keys(*keys)`
+* `page.driver.set_proxy(ip, port, type, user, password)`
 * window API
 * cookie handling
 * drag-and-drop

lib/capybara/poltergeist/browser.rb

@@ -260,6 +260,13 @@ module Capybara::Poltergeist
       command('clear_network_traffic')
     end
 
+    def set_proxy(ip, port, type, user, password)
+      args = [ip, port, type]
+      args << user if user
+      args << password if password
+      command('set_proxy', *args)
+    end
+
     def equals(page_id, id, other_id)
       command('equals', page_id, id, other_id)
     end

lib/capybara/poltergeist/client/browser.coffee

@@ -400,6 +400,10 @@ class Poltergeist.Browser
     @currentPage.clearNetworkTraffic()
     @current_command.sendResponse(true)
 
+  set_proxy: (ip, port, type, user, password) ->
+    phantom.setProxy(ip, port, type, user, password)
+    @current_command.sendResponse(true)
+
   get_headers: ->
     @current_command.sendResponse(@currentPage.getCustomHeaders())
 

lib/capybara/poltergeist/client/compiled/browser.js

@@ -549,6 +549,11 @@ Poltergeist.Browser = (function() {
     return this.current_command.sendResponse(true);
   };
 
+  Browser.prototype.set_proxy = function(ip, port, type, user, password) {
+    phantom.setProxy(ip, port, type, user, password);
+    return this.current_command.sendResponse(true);
+  };
+
   Browser.prototype.get_headers = function() {
     return this.current_command.sendResponse(this.currentPage.getCustomHeaders());
   };

lib/capybara/poltergeist/driver.rb

@@ -224,6 +224,10 @@ module Capybara::Poltergeist
       browser.clear_network_traffic
     end
 
+    def set_proxy(ip, port, type = "http", user = nil, password = nil)
+      browser.set_proxy(ip, port, type, user, password)
+    end
+
     def headers
       browser.get_headers
     end