fix: Use correct exception type for fallback at file/folder pulling (appium#1912)

valfirst · web-flow · commit 79106ef59cd4 · 2023-05-10T07:31:24.000+03:00
diff --git a/src/main/java/io/appium/java_client/PullsFiles.java b/src/main/java/io/appium/java_client/PullsFiles.java
@@ -17,6 +17,7 @@
 package io.appium.java_client;
 
 import com.google.common.collect.ImmutableMap;
+import org.openqa.selenium.UnsupportedCommandException;
 
 import java.nio.charset.StandardCharsets;
 import java.util.AbstractMap;
@@ -49,7 +50,7 @@ default byte[] pullFile(String remotePath) {
                         ImmutableMap.of("remotePath", remotePath)
                 )
             );
-        } catch (UnsupportedOperationException e) {
+        } catch (UnsupportedCommandException e) {
             // TODO: Remove the fallback
             base64String = checkNotNull(
                 CommandExecutionHelper.execute(markExtensionAbsence(extName),
@@ -81,7 +82,7 @@ default byte[] pullFolder(String remotePath) {
                         ImmutableMap.of("remotePath", remotePath)
                 )
             );
-        } catch (UnsupportedOperationException e) {
+        } catch (UnsupportedCommandException e) {
             // TODO: Remove the fallback
             base64String = checkNotNull(
                 CommandExecutionHelper.execute(markExtensionAbsence(extName),
