Fixed issue where IOS/Android drivers werent being returned for remote, if they were set.

Author: jleeh

src/main/java/com/frameworkium/config/browsers/BrowserStackImpl.java

@@ -2,6 +2,8 @@
 
 import com.frameworkium.config.DriverType;
 import com.frameworkium.config.remotes.BrowserStack;
+import io.appium.java_client.android.AndroidDriver;
+import io.appium.java_client.ios.IOSDriver;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.remote.DesiredCapabilities;
 import org.openqa.selenium.remote.RemoteWebDriver;
@@ -68,7 +70,14 @@ public DesiredCapabilities getDesiredCapabilities() {
     }
 
     public WebDriver getWebDriverObject(DesiredCapabilities capabilities) {
-        return new RemoteWebDriver(remoteURL, capabilities);
+        switch(supportedPlatform) {
+            case ANDROID:
+                return new AndroidDriver(remoteURL, capabilities);
+            case IOS:
+                return new IOSDriver(remoteURL, capabilities);
+            default:
+                return new RemoteWebDriver(remoteURL, capabilities);
+        }
     }
 
     private DesiredCapabilities setCapabilitiesBasedOnPlatform(DesiredCapabilities desiredCapabilities) {

src/main/java/com/frameworkium/config/browsers/SauceImpl.java

@@ -2,6 +2,8 @@
 
 import com.frameworkium.config.DriverType;
 import com.frameworkium.config.remotes.Sauce;
+import io.appium.java_client.android.AndroidDriver;
+import io.appium.java_client.ios.IOSDriver;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.remote.DesiredCapabilities;
 import org.openqa.selenium.remote.RemoteWebDriver;
@@ -65,7 +67,14 @@ public DesiredCapabilities getDesiredCapabilities() {
     }
 
     public WebDriver getWebDriverObject(DesiredCapabilities capabilities) {
-        return new RemoteWebDriver(remoteURL, capabilities);
+        switch(supportedPlatforms) {
+            case ANDROID:
+                return new AndroidDriver(remoteURL, capabilities);
+            case IOS:
+                return new IOSDriver(remoteURL, capabilities);
+            default:
+                return new RemoteWebDriver(remoteURL, capabilities);
+        }
     }
 
     private DesiredCapabilities setCapabilitiesBasedOnPlatform(DesiredCapabilities desiredCapabilities) {