fix(windows): click external protocol prompt via UIAutomation

Author: rodrigo-fournier-immutable

sample/Tests/test/test_windows_helpers.py

@@ -79,6 +79,70 @@ def wait_for_authentication_success_in_unity_logs(timeout_seconds: int = 60) ->
     print("No authentication success detected in Unity logs within timeout.")
     return False
 
+
+def try_click_protocol_permission_dialog(timeout_seconds: int = 15) -> bool:
+    """
+    Try to click the browser-level external protocol permission prompt.
+
+    Important: this dialog is NOT part of the web page DOM, so Selenium cannot click it.
+    We use Windows UIAutomation to locate and invoke an "Open/Allow/Yes" button.
+    """
+    product_name = os.getenv("UNITY_APP_NAME", get_product_name())
+    print("Attempting to click external protocol permission dialog via UIAutomation...")
+
+    ps_script = f'''
+    $deadline = (Get-Date).AddSeconds({int(timeout_seconds)})
+    Add-Type -AssemblyName UIAutomationClient
+
+    function Find-And-Click($titlePart) {{
+      $procs = Get-Process -ErrorAction SilentlyContinue | Where-Object {{ $_.MainWindowHandle -ne 0 -and $_.MainWindowTitle -like "*$titlePart*" }}
+      foreach ($p in $procs) {{
+        try {{
+          $el = [Windows.Automation.AutomationElement]::FromHandle($p.MainWindowHandle)
+          if ($null -eq $el) {{ continue }}
+          $buttons = $el.FindAll([Windows.Automation.TreeScope]::Descendants,
+            [Windows.Automation.Condition]::new([Windows.Automation.AutomationElement]::ControlTypeProperty, [Windows.Automation.ControlType]::Button))
+
+          foreach ($b in $buttons) {{
+            $name = $b.Current.Name
+            if ($name -like "*Open*" -or $name -like "*Allow*" -or $name -like "*Yes*" -or $name -like "*{product_name}*") {{
+              $b.GetCurrentPattern([Windows.Automation.InvokePattern]::Pattern).Invoke()
+              Write-Host "Clicked protocol dialog button: $name"
+              return $true
+            }}
+          }}
+        }} catch {{}}
+      }}
+      return $false
+    }}
+
+    while ((Get-Date) -lt $deadline) {{
+      if (Find-And-Click "auth.immutable.com") {{ exit 0 }}
+      if (Find-And-Click "Immutable Passport") {{ exit 0 }}
+      if (Find-And-Click "Open") {{ exit 0 }}
+      Start-Sleep -Milliseconds 500
+    }}
+    Write-Host "No protocol dialog found"
+    exit 1
+    '''
+
+    try:
+        result = subprocess.run(
+            ["powershell", "-Command", ps_script],
+            capture_output=True,
+            text=True,
+            timeout=max(5, int(timeout_seconds) + 5),
+        )
+        if "Clicked protocol dialog button" in (result.stdout or ""):
+            print(result.stdout.strip())
+            print("Successfully clicked protocol dialog via UIAutomation.")
+            return True
+        print((result.stdout or "").strip() or "UIAutomation did not find/click any dialog button.")
+        return False
+    except Exception as e:
+        print(f"UIAutomation dialog click failed: {e}")
+        return False
+
 # Add chrome.exe to environment variable
 # Download chrome driver and add to environment variable
 
@@ -594,13 +658,23 @@ def login():
             except:
                 pass
 
-        # In CI, if protocol association is already configured, Chrome/Brave may auto-handle
-        # the external protocol WITHOUT showing any permission dialog. In that case there
-        # may be 0 buttons on /checking, and waiting for a dialog will just time out.
-        if "auth.immutable.com/checking" in current_url_lower and len(buttons) == 0:
-            print("No deep link permission dialog present on /checking (0 buttons).")
-            print("Assuming protocol is already permitted; waiting for Unity to process callback via Player.log...")
-            wait_for_authentication_success_in_unity_logs(timeout_seconds=75)
+        # The "open external protocol" prompt is typically browser UI (not DOM),
+        # so 0 buttons on /checking is expected. Try:
+        # - wait briefly for Unity success (maybe protocol already fired)
+        # - click the browser-level prompt via UIAutomation
+        # - wait again for Unity success
+        if "auth.immutable.com/checking" in current_url_lower:
+            if wait_for_authentication_success_in_unity_logs(timeout_seconds=10):
+                return
+
+            clicked = try_click_protocol_permission_dialog(timeout_seconds=20)
+            if clicked:
+                # Give Unity a moment to receive and process the callback
+                wait_for_authentication_success_in_unity_logs(timeout_seconds=75)
+                return
+
+            print("No protocol dialog clicked; Unity may never receive the callback.")
+            wait_for_authentication_success_in_unity_logs(timeout_seconds=30)
             return
 
         # Otherwise, try to click the deep link dialog if it appears.