Implementing ability to set proxy in Options classes

Author: barancev

java/client/src/org/openqa/selenium/chrome/ChromeOptions.java

@@ -32,6 +32,7 @@
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.PageLoadStrategy;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.SessionNotCreatedException;
 import org.openqa.selenium.UnexpectedAlertBehaviour;
 import org.openqa.selenium.remote.BrowserType;
@@ -242,6 +243,11 @@ public ChromeOptions setHeadless(boolean headless) {
     return this;
   }
 
+  public ChromeOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   /**
    * Returns DesiredCapabilities for Chrome with these options included as
    * capabilities. This does not copy the options. Further changes will be

java/client/src/org/openqa/selenium/edge/EdgeOptions.java

@@ -24,6 +24,7 @@
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.Platform;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.remote.BrowserType;
 import org.openqa.selenium.remote.CapabilityType;
 
@@ -76,6 +77,11 @@ public void setPageLoadStrategy(String strategy) {
     setCapability(PAGE_LOAD_STRATEGY, Objects.requireNonNull(strategy));
   }
 
+  public EdgeOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   /**
    * Returns DesiredCapabilities for Edge with these options included as capabilities. This does not
    * copy the options. Further changes will be reflected in the returned capabilities.

java/client/src/org/openqa/selenium/firefox/FirefoxOptions.java

@@ -32,6 +32,7 @@
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.PageLoadStrategy;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.UnexpectedAlertBehaviour;
 import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.firefox.internal.ProfilesIni;
@@ -296,6 +297,11 @@ public FirefoxOptions setHeadless(boolean headless) {
     return this;
   }
 
+  public FirefoxOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   @Override
   public void setCapability(String key, Object value) {
     switch (key) {

java/client/src/org/openqa/selenium/ie/InternetExplorerOptions.java

@@ -44,6 +44,7 @@
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.PageLoadStrategy;
 import org.openqa.selenium.Platform;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.UnexpectedAlertBehaviour;
 import org.openqa.selenium.internal.ElementScrollBehavior;
 import org.openqa.selenium.remote.BrowserType;
@@ -229,6 +230,11 @@ public InternetExplorerOptions setUnhandledPromptBehaviour(UnexpectedAlertBehavi
     return amend(UNHANDLED_PROMPT_BEHAVIOUR, behaviour);
   }
 
+  public InternetExplorerOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   private InternetExplorerOptions amend(String optionName, Object value) {
     setCapability(optionName, value);
     return this;

java/client/src/org/openqa/selenium/opera/OperaOptions.java

@@ -28,7 +28,9 @@
 import com.google.common.io.Files;
 
 import org.openqa.selenium.MutableCapabilities;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.WebDriverException;
+import org.openqa.selenium.remote.CapabilityType;
 
 import java.io.File;
 import java.io.IOException;
@@ -199,6 +201,11 @@ public Object getExperimentalOption(String name) {
     return experimentalOptions.get(checkNotNull(name));
   }
 
+  public OperaOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   @Override
   public Map<String, ?> asMap() {
     Map<String, Object> toReturn = new TreeMap<>();

java/client/src/org/openqa/selenium/safari/SafariOptions.java

@@ -22,6 +22,7 @@
 import org.openqa.selenium.Capabilities;
 import org.openqa.selenium.MutableCapabilities;
 import org.openqa.selenium.Platform;
+import org.openqa.selenium.Proxy;
 import org.openqa.selenium.WebDriverException;
 import org.openqa.selenium.remote.CapabilityType;
 
@@ -158,6 +159,11 @@ public SafariOptions setUseTechnologyPreview(boolean useTechnologyPreview) {
     return this;
   }
 
+  public SafariOptions setProxy(Proxy proxy) {
+    setCapability(CapabilityType.PROXY, proxy);
+    return this;
+  }
+
   // Getters
 
   /**