fix: Get Context Handles Only in WebView Context to Avoid UnsupportedCommandException (#235)

* fix: Get Context Handles Only in WebView Context to Avoid UnsupportedCommandException

* Splotless apply

* Fix failing test and improve existing exception

Author: rjap-salesforce

utam-core/src/main/java/utam/core/selenium/appium/MobileDriverAdapter.java

@@ -100,9 +100,10 @@ final AppiumDriver switchToWebView(String title) {
     // For the Appium chromedriver limitation to handle multiple WebViews,
     // If switch to context fail to find the target WebView, then switch to
     // use window
-    if (mobilePlatform == MobilePlatformType.ANDROID
-        || mobilePlatform == MobilePlatformType.ANDROID_PHONE
-        || mobilePlatform == MobilePlatformType.ANDROID_TABLET) {
+    if ((mobilePlatform == MobilePlatformType.ANDROID
+            || mobilePlatform == MobilePlatformType.ANDROID_PHONE
+            || mobilePlatform == MobilePlatformType.ANDROID_TABLET)
+        && !isNativeContext()) {
       Set<String> windowHandles = appiumDriver.getWindowHandles();
       for (String windowHandle : windowHandles) {
         if (!windowHandle.equals(NATIVE_CONTEXT_HANDLE)) {

utam-core/src/test/java/utam/core/selenium/appium/MobileDriverAdapterTests.java

@@ -30,11 +30,13 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 import org.openqa.selenium.Platform;
 import org.openqa.selenium.TimeoutException;
+import org.openqa.selenium.UnsupportedCommandException;
 import org.openqa.selenium.WebDriver.Navigation;
 import org.openqa.selenium.WebDriver.TargetLocator;
 import org.testng.annotations.Test;
@@ -253,7 +255,6 @@ public void testIsAnyWebViewContextAvailableWithNoWebView() {
   @Test
   public void testIsAnyWebViewContextAvailable() {
     MockUtilities mock = new MockUtilities(AppiumDriver.class);
-    AppiumDriver driver = (AppiumDriver) mock.getWebDriverMock();
     SupportsContextSwitching contextSwitcher = mock.getContextSwitcherMock();
     MobileDriverAdapter provider = (MobileDriverAdapter) mock.getDriverAdapter();
 
@@ -405,14 +406,42 @@ public void testSwitchToWebViewWithMultipleWebViewsAndroid() {
             });
     when(driver.getWindowHandles()).thenReturn(windowHandles);
     when(driver.getTitle()).thenReturn("").thenReturn(testWebViewTitle);
-    when(contextSwitcher.getContext()).thenReturn(contextTracker.currentContext);
+    when(contextSwitcher.getContext())
+        .thenReturn(contextTracker.currentContext)
+        .thenReturn(testWebViewHandle);
     mock.setMobilePlatform(Platform.LINUX);
     MobileDriverAdapter adapter = mock.getMobileDriverAdapter();
     assertThat(
         adapter.waitFor(() -> adapter.switchToWebView(testWebViewTitle)), is(sameInstance(driver)));
     assertThat(windowHandleTracker.currentHandle, is(equalTo(testWindowHandle)));
   }
 
+  /**
+   * Tests that the expectation to switch to no windows and not attempt to get handles when only
+   * native context exists on Android platform, negative case
+   */
+  @Test
+  public void testSwitchToWebViewWithNoWebViewsAndroid() {
+    ContextTracker contextTracker = new ContextTracker();
+
+    MockUtilities mock = new MockUtilities(AppiumDriver.class);
+    AppiumDriver driver = mock.getAppiumDriverMock();
+    SupportsContextSwitching contextSwitcher = mock.getContextSwitcherMock();
+
+    Set<String> contextHandles = new HashSet<>(List.of(NATIVE_CONTEXT_HANDLE));
+
+    String testWebViewTitle = "Test Application";
+    when(driver.getCapabilities().getPlatformName()).thenReturn(Platform.ANDROID);
+    when(contextSwitcher.getContextHandles()).thenReturn(contextHandles);
+    when(driver.getWindowHandles())
+        .thenThrow(new UnsupportedCommandException("getWindowHandles {}"));
+    when(contextSwitcher.getContext()).thenReturn(contextTracker.currentContext);
+    mock.setMobilePlatform(Platform.LINUX);
+    MobileDriverAdapter adapter = mock.getMobileDriverAdapter();
+    assertThat(adapter.switchToWebView(testWebViewTitle), nullValue());
+    assertThat(contextTracker.currentContext, is(equalTo(NATIVE_CONTEXT_HANDLE)));
+  }
+
   @Test
   public void testSetPageContext() {
     MockUtilities mock = new MockUtilities(AppiumDriver.class);