Merge pull request #36 from smccarthy/ieprotectedmode

Ieprotectedmode - Check if Windows OS

Author: dimacus

SeleniumGridExtras/pom.xml

@@ -20,6 +20,11 @@
     </repositories>
 
     <dependencies>
+		<dependency>
+		    <groupId>net.java.dev.jna</groupId>
+		    <artifactId>jna-platform</artifactId>
+		    <version>4.1.0</version>
+		</dependency>
         <dependency>
             <groupId>org.seleniumhq.selenium</groupId>
             <artifactId>selenium-server</artifactId>

SeleniumGridExtras/src/main/java/com/groupon/seleniumgridextras/tasks/IEProtectedMode.java

@@ -39,8 +39,7 @@
 
 import com.google.gson.JsonObject;
 
-import com.groupon.seleniumgridextras.ExecuteCommand;
-
+import com.groupon.seleniumgridextras.config.RuntimeConfig;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -106,8 +105,6 @@ public String getCurrentSettingForZone(String zoneId) {
     return String.format(regLocation, zoneId);
   }
 
-  ;
-
 
   @Override
   public JsonObject execute(Map<String, String> parameter) {
@@ -119,14 +116,24 @@ public JsonObject execute(Map<String, String> parameter) {
 
   @Override
   public JsonObject execute() {
-    return getAllProtectedStatus();
+    if (RuntimeConfig.getOS().isWindows()) {
+      return getAllProtectedStatus();
+    } else {
+      getJsonResponse().addKeyValues("error", "IE Protected Mode command is only implemented in Windows");
+      return getJsonResponse().getJson();
+    }
   }
 
   @Override
   public JsonObject execute(String status) {
-    setAllProtectedStatuses(status.equals("1") ? true : false);
-    getJsonResponse().addKeyValues("out", "IE needs to restart before you see the changes");
-    return getAllProtectedStatus();
+    if (RuntimeConfig.getOS().isWindows()) {
+      setAllProtectedStatuses(status.equals("1") ? true : false);
+      getJsonResponse().addKeyValues("out", "IE needs to restart before you see the changes");
+      return getAllProtectedStatus();
+    } else {
+      getJsonResponse().addKeyValues("error", "IE Protected Mode command is only implemented in Windows");
+      return getJsonResponse().getJson();
+    }
   }
 
   private void setAllProtectedStatuses(boolean value) {
@@ -142,8 +149,6 @@ private void setAllProtectedStatuses(boolean value) {
     }
   }
 
-  ;
-
 
   private Boolean getProtectedEnabledForZone(String zone) {
     int enabled =