diff --git a/src/main/java/io/appium/java_client/mac/options/SupportsEnvironmentOption.java b/src/main/java/io/appium/java_client/mac/options/SupportsEnvironmentOption.java index 17a2d05f9..803561030 100644 --- a/src/main/java/io/appium/java_client/mac/options/SupportsEnvironmentOption.java +++ b/src/main/java/io/appium/java_client/mac/options/SupportsEnvironmentOption.java @@ -20,7 +20,7 @@ import io.appium.java_client.remote.options.CanSetCapability; import org.openqa.selenium.Capabilities; -import java.util.List; +import java.util.Map; import java.util.Optional; public interface SupportsEnvironmentOption> extends @@ -33,11 +33,11 @@ public interface SupportsEnvironmentOption> extends * the parent process. This capability is only going to be applied if the application * is not running on session startup. * - * @param arguments E.g. ["--help"]. + * @param env E.g. ["--help"]. * @return self instance for chaining. */ - default T setEnvironment(List arguments) { - return amend(ENVIRONMENT_OPTION, arguments); + default T setEnvironment(Map env) { + return amend(ENVIRONMENT_OPTION, env); } /** @@ -45,8 +45,8 @@ default T setEnvironment(List arguments) { * * @return Application environment mapping. */ - default Optional> getEnvironment() { + default Optional> getEnvironment() { //noinspection unchecked - return Optional.ofNullable((List) getCapability(ENVIRONMENT_OPTION)); + return Optional.ofNullable((Map) getCapability(ENVIRONMENT_OPTION)); } }