added wrapper methods for getContextHandles and setContext

src/main/java/com/codeborne/selenide/appium/SelenideAppium.java

@@ -7,6 +7,7 @@
 
 import javax.annotation.CheckReturnValue;
 import javax.annotation.Nonnull;
+import java.util.Set;
 
 import static com.codeborne.selenide.WebDriverRunner.driver;
 
@@ -68,6 +69,14 @@ public static void back() {
     Selenide.back();
   }
 
+  public static void setContext(String contextName) {
+    new SelenideAppiumTargetLocator().setContext(contextName);
+  }
+
+  public static Set<String> getContextHandles() {
+    return new SelenideAppiumTargetLocator().getContextHandles();
+  }
+
   @CheckReturnValue
   @Nonnull
   public static SelenideAppiumElement $x(String xpathExpression) {

src/main/java/com/codeborne/selenide/appium/SelenideAppiumTargetLocator.java

@@ -0,0 +1,26 @@
+package com.codeborne.selenide.appium;
+
+import com.codeborne.selenide.WebDriverRunner;
+import com.codeborne.selenide.logevents.SelenideLogger;
+import org.openqa.selenium.ContextAware;
+
+import java.util.Set;
+
+public class SelenideAppiumTargetLocator {
+
+  public void setContext(String contextName) {
+    SelenideLogger.run("set context", contextName, () -> {
+      (WebdriverUnwrapper.cast(WebDriverRunner.getWebDriver(), ContextAware.class))
+        .map(contextAware -> contextAware.context(contextName))
+        .orElseThrow(() -> new UnsupportedOperationException("Context not found" + contextName));
+    });
+  }
+
+  public Set<String> getContextHandles() {
+    return (WebdriverUnwrapper.cast(WebDriverRunner.getWebDriver(), ContextAware.class))
+      .map(ContextAware::getContextHandles)
+      .orElseThrow(() -> new UnsupportedOperationException("Cannot get contexts from mobile driver"));
+  }
+
+
+}

src/test/java/integration/android/ContextTest.java

@@ -0,0 +1,24 @@
+package integration.android;
+
+import com.codeborne.selenide.appium.SelenideAppium;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import static com.codeborne.selenide.WebDriverRunner.closeWebDriver;
+import static org.assertj.core.api.Assertions.assertThat;
+
+class ContextTest extends BaseSwagLabsAndroidTest {
+
+  @BeforeEach
+  void setUp() {
+    closeWebDriver();
+  }
+
+  @Test
+  void contexts() {
+    SelenideAppium.openAndroidDeepLink("mydemoapprn://webview", "com.saucelabs.mydemoapp.rn");
+    SelenideAppium.setContext("NATIVE_APP");
+    assertThat(SelenideAppium.getContextHandles())
+      .anyMatch(context -> context.contentEquals("NATIVE_APP"));
+  }
+}