Merge pull request #325 from sbtqa/mobile-android

Record video on android
sbtqa · Nov 30, 2020 · f30da4d · f30da4d
2 parents 9ae3d5e + 71a27a3
commit f30da4d
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 9 deletions.
diff --git a/...-plugin/src/main/java/ru/sbtqa/tag/pagefactory/mobile/properties/MobileConfiguration.java b/...-plugin/src/main/java/ru/sbtqa/tag/pagefactory/mobile/properties/MobileConfiguration.java
@@ -124,6 +124,18 @@ public interface MobileConfiguration extends Configuration {
     @DefaultValue("10")
     int getAppiumVideoFps();
 
+    @Key("appium.video.bitrate")
+    @DefaultValue("200000000")
+    int getAppiumVideoBitRate();
+
+    @Key("appium.video.videoSize")
+    @DefaultValue("1280x720")
+    String getAppiumVideoSize();
+
+    @Key("appium.video.bugReport")
+    @DefaultValue("false")
+    boolean getAppiumVideoBugReport();
+
     @Key("appium.xcodeOrgId")
     @DefaultValue("")
     String getAppiumXcodeOrgId();

diff --git a/...obile-plugin/src/main/java/ru/sbtqa/tag/pagefactory/mobile/utils/AppiumVideoRecorder.java b/...obile-plugin/src/main/java/ru/sbtqa/tag/pagefactory/mobile/utils/AppiumVideoRecorder.java
@@ -1,6 +1,8 @@
 package ru.sbtqa.tag.pagefactory.mobile.utils;
 
 import cucumber.api.Scenario;
+import io.appium.java_client.android.AndroidDriver;
+import io.appium.java_client.android.AndroidStartScreenRecordingOptions;
 import io.appium.java_client.ios.IOSDriver;
 import io.appium.java_client.ios.IOSStartScreenRecordingOptions;
 import io.appium.java_client.ios.IOSStartScreenRecordingOptions.VideoQuality;
@@ -38,15 +40,27 @@ public void startRecord() {
             return;
         }
 
-        IOSStartScreenRecordingOptions startOptions = new IOSStartScreenRecordingOptions()
-                .withVideoType(PROPERTIES.getAppiumVideoType())
-                .withVideoScale(PROPERTIES.getAppiumVideoScale())
-                .withTimeLimit(Duration.ofSeconds(PROPERTIES.getAppiumTimeLimit()))
-                .withVideoQuality(VideoQuality.valueOf(PROPERTIES.getAppiumVideoQuality()))
-                .withFps(PROPERTIES.getAppiumVideoFps())
-                .enableForcedRestart();
+        if (PROPERTIES.getAppiumPlatformName() == PlatformName.IOS) {
+            IOSStartScreenRecordingOptions startOptions = new IOSStartScreenRecordingOptions()
+                    .withVideoType(PROPERTIES.getAppiumVideoType())
+                    .withVideoScale(PROPERTIES.getAppiumVideoScale())
+                    .withTimeLimit(Duration.ofSeconds(PROPERTIES.getAppiumTimeLimit()))
+                    .withVideoQuality(VideoQuality.valueOf(PROPERTIES.getAppiumVideoQuality()))
+                    .withFps(PROPERTIES.getAppiumVideoFps())
+                    .enableForcedRestart();
+            ((IOSDriver) Environment.getDriverService().getDriver()).startRecordingScreen(startOptions);
+        } else {
+            AndroidStartScreenRecordingOptions startOptions = new AndroidStartScreenRecordingOptions()
+                    .withBitRate(PROPERTIES.getAppiumVideoBitRate())
+                    .withVideoSize(PROPERTIES.getAppiumVideoSize())
+                    .withTimeLimit(Duration.ofSeconds(PROPERTIES.getAppiumTimeLimit()))
+                    .enableForcedRestart();
+            if (PROPERTIES.getAppiumVideoBugReport()) {
+                startOptions.enableBugReport();
+            }
+            ((AndroidDriver) Environment.getDriverService().getDriver()).startRecordingScreen(startOptions);
+        }
 
-        ((IOSDriver) Environment.getDriverService().getDriver()).startRecordingScreen(startOptions);
         isRecording = true;
     }
 
@@ -57,7 +71,9 @@ public byte[] stopRecord() {
         }
 
         // get Base64 encoded video content
-        String encodedString = ((IOSDriver) Environment.getDriverService().getDriver()).stopRecordingScreen();
+        String encodedString = PROPERTIES.getAppiumPlatformName() == PlatformName.IOS
+                ? ((IOSDriver) Environment.getDriverService().getDriver()).stopRecordingScreen()
+                : ((AndroidDriver) Environment.getDriverService().getDriver()).stopRecordingScreen();
 
         // convert to byte array
         byte[] decodedBytes = Base64.getDecoder().decode(encodedString.getBytes());