Merge pull request #52 from AutomateThePlanet/android-custom-driver-options-fix

Added custom capabilities on app restart

Author: teo-nikolov

bellatrix.android/src/main/java/solutions/bellatrix/android/infrastructure/DriverService.java

@@ -207,15 +207,13 @@ private static String getBuildName() {
     }
 
     private static <TOption extends MutableCapabilities> void addDriverConfigOptions(TOption chromeOptions) {
-        for (var optionKey : APP_CONFIGURATION.get().appiumOptions.keySet()) {
-            chromeOptions.setCapability(optionKey, APP_CONFIGURATION.get().appiumOptions.get(optionKey));
+        for (var optionEntry : APP_CONFIGURATION.get().appiumOptions.entrySet()) {
+            chromeOptions.setCapability(optionEntry.getKey(), optionEntry.getValue());
         }
     }
 
     private static <TOption extends MutableCapabilities> void addCustomDriverOptions(TOption mobileOptions) {
-        for (var optionKey : CUSTOM_DRIVER_OPTIONS.get().keySet()) {
-            mobileOptions.setCapability(optionKey, CUSTOM_DRIVER_OPTIONS.get().get(optionKey));
-        }
+        getCustomDriverOptions().forEach(mobileOptions::setCapability);
     }
 
     public static void close() {
@@ -225,7 +223,7 @@ public static void close() {
 
         if (WRAPPED_ANDROID_DRIVER.get() != null) {
             WRAPPED_ANDROID_DRIVER.get().quit();
-            CUSTOM_DRIVER_OPTIONS.get().clear();
+            // CUSTOM_DRIVER_OPTIONS.get().clear();
         }
 
         DISPOSED.set(true);