From 35ae5bf578db7385ae8f891f3cee82af437568d3 Mon Sep 17 00:00:00 2001
From: Eric Mikulin <eric@picovoice.ai>
Date: Wed, 15 Jan 2025 12:02:23 -0800
Subject: [PATCH] iOS and Android browserstack workflow fixes

---
 .github/workflows/ios-browserstack.yml |   5 +
 script/automation/browserstack.py      | 136 +++++++++++++++++++++++++
 2 files changed, 141 insertions(+)
 create mode 100644 script/automation/browserstack.py

diff --git a/.github/workflows/ios-browserstack.yml b/.github/workflows/ios-browserstack.yml
index b16367458..a007d223e 100644
--- a/.github/workflows/ios-browserstack.yml
+++ b/.github/workflows/ios-browserstack.yml
@@ -40,6 +40,11 @@ jobs:
       - name: Make build dir
         run: mkdir ddp
 
+      - name: Install resource script dependency
+        run: |
+          brew update
+          brew install convmv
+
       - name: Copy test_resources
         run: ./copy_test_resources.sh
 
diff --git a/script/automation/browserstack.py b/script/automation/browserstack.py
new file mode 100644
index 000000000..54c0e77a0
--- /dev/null
+++ b/script/automation/browserstack.py
@@ -0,0 +1,136 @@
+import argparse
+import requests
+import time
+
+APP_URI = 'https://api-cloud.browserstack.com/app-automate/{}/v2/app'
+TEST_URI = 'https://api-cloud.browserstack.com/app-automate/{}/v2/test-suite'
+BUILD_URI = 'https://api-cloud.browserstack.com/app-automate/{}/v2/build'
+STATUS_URI = 'https://api-cloud.browserstack.com/app-automate/{}/v2/builds/{}'
+
+devices_dict = {
+    'android-min-max': [
+        'Samsung Galaxy S8-7.0',
+        'Samsung Galaxy M52-11.0',
+        'Google Pixel 9-15.0'
+    ],
+    'android-perf': [
+        'Google Pixel 6 Pro-15.0'
+    ],
+    'ios-min-max': [
+        'iPhone SE 2022-15',
+        'iPhone 14 Plus-16',
+        'iPhone 14-18'
+    ],
+    'ios-perf': [
+        'iPhone 13-18',
+    ]
+}
+
+
+def main(args: argparse.Namespace) -> None:
+    app_files = {
+        'file': open(args.app_path, 'rb')
+    }
+
+    app_response = requests.post(
+        APP_URI.format(args.type),
+        files=app_files,
+        auth=(args.username, args.access_key)
+    )
+    app_response_json = app_response.json()
+
+    if not app_response.ok:
+        print('App Upload Failed', app_response_json)
+        exit(1)
+
+    test_files = {
+        'file': open(args.test_path, 'rb')
+    }
+    test_response = requests.post(
+        TEST_URI.format(args.type),
+        files=test_files,
+        auth=(args.username, args.access_key)
+    )
+    test_response_json = test_response.json()
+
+    if not test_response.ok:
+        print('Test Upload Failed', test_response_json)
+        exit(1)
+
+    build_headers = {
+        'Content-Type': 'application/json'
+    }
+    build_data = {
+        'app': app_response_json['app_url'],
+        'testSuite': test_response_json['test_suite_url'],
+        'project': args.project_name,
+        'devices': devices_dict[args.devices],
+        'deviceLogs': True
+    }
+
+    while True:
+        build_response = requests.post(
+            BUILD_URI.format(args.type),
+            headers=build_headers,
+            json=build_data,
+            auth=(args.username, args.access_key)
+        )
+        if (build_response is not None and 'message' in build_response.json() and '[BROWSERSTACK_ALL_PARALLELS_IN_USE]'
+                in build_response.json()['message']):
+            print('Parallel threads limit reached. Waiting...', flush=True)
+            time.sleep(60)
+        else:
+            break
+
+    if build_response is None:
+        print('Build Failed')
+        exit(1)
+
+    build_response_json = build_response.json()
+
+    if not build_response.ok:
+        print('Build Failed', build_response.json())
+        exit(1)
+
+    if build_response_json['message'] != 'Success':
+        print('Build Unsuccessful')
+        exit(1)
+
+    print(
+        'View build results at https://app-automate.browserstack.com/dashboard/v2/builds/{}'
+        .format(build_response_json['build_id']))
+
+    while True:
+        time.sleep(60)
+        status_response = requests.get(
+            STATUS_URI.format(args.type, build_response_json['build_id']),
+            auth=(args.username, args.access_key)
+        )
+        status_response_json = status_response.json()
+        status = status_response_json['status']
+
+        if not status_response.ok:
+            print('Status Request Failed', status_response_json)
+            exit(1)
+
+        if status != 'queued' and status != 'running':
+            break
+
+    print('Status:', status)
+    if status != 'passed':
+        exit(1)
+
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--type', choices=['espresso', 'xcuitest'], required=True)
+    parser.add_argument('--username', required=True)
+    parser.add_argument('--access_key', required=True)
+
+    parser.add_argument('--project_name', required=True)
+    parser.add_argument('--devices', choices=devices_dict.keys(), required=True)
+    parser.add_argument('--app_path', required=True)
+    parser.add_argument('--test_path', required=True)
+    args = parser.parse_args()
+
+    main(args)