test(BRIDGE-136): Download Bridge

tests/e2e/ui_tests/windows_os/TestSession.cs

@@ -1,11 +1,16 @@
 using System;
 using System.Threading;
+using System.Management.Automation;
+using System.Management.Automation.Runspaces;
 using FlaUI.Core.AutomationElements;
 using FlaUI.Core;
 using FlaUI.UIA3;
 using ProtonMailBridge.UI.Tests.TestsHelper;
 using FlaUI.Core.Input;
 using System.Diagnostics;
+using System.Drawing.Text;
+using System.Collections.ObjectModel;
+using FlaUI.Core.Tools;
 
 namespace ProtonMailBridge.UI.Tests
 {
@@ -17,13 +22,57 @@ namespace ProtonMailBridge.UI.Tests
         protected static Window Window;
         protected static Window ChromeWindow;
         protected static Window FileExplorerWindow;
+        protected static Runspace? myRunSpace;
+        protected static string? bridgeDownloadURL;
+        protected static string downloadEnvVariable = "BRIDGE_DOWNLOAD_URL";
+        private static readonly DebugTests debugTests = new();
+        protected static string atlasEnvironment = "https://mail-api.proton.pink/";
+
 
         protected static void ClientCleanup()
         {
-            App.Kill();
-            App.Dispose();
+            var outcome = TestContext.CurrentContext.Result.Outcome.Status;
+            if (outcome == NUnit.Framework.Interfaces.TestStatus.Failed)
+            {
+                try
+                {
+                    debugTests.TakeScreenshot();
+                }
+                catch (Exception ex)
+                {
+                    TestContext.Out.WriteLine(ex.ToString());
+                }
+            }
+
+            try
+            {
+                App.Kill();
+            }
+            catch (Exception ex)
+            {
+                TestContext.Out.WriteLine(ex.ToString());
+            }
+
+            try
+            {
+                App.Dispose();
+            }
+            catch (Exception ex)
+            {
+                TestContext.Out.WriteLine(ex.ToString());
+            }
+
             // Give some time to properly exit the app
             Thread.Sleep(10000);
+            try
+            {
+                RemoveBridgeCredentials();
+            }
+            catch (Exception ex)
+            {
+                TestContext.Out.WriteLine($"Failed to remove Bridge credentials: {ex}");
+            }
+
         }
 
         public static void switchToFileExplorerWindow()
@@ -60,28 +109,117 @@ namespace ProtonMailBridge.UI.Tests
             }
 
             // Cast the found element to a Window object
-             ChromeWindow = _chromeWindow.AsWindow();
+            ChromeWindow = _chromeWindow.AsWindow();
 
             // Focus on the Chrome window
             ChromeWindow.Focus();
         }
 
+        protected static void RefreshWindow(TimeSpan? timeout = null)
+        {
+            Window = null;
+            TimeSpan refreshTimeout = timeout ?? TestData.ThirtySecondsTimeout;
+            RetryResult<Window> retry = Retry.WhileNull(
+                () =>
+                {
+                    try
+                    {
+                        Window = App.GetMainWindow(new UIA3Automation(), refreshTimeout);
+                    }
+                    catch (System.TimeoutException)
+                    {
+                        // Ignore
+                    }
+                    return Window;
+                },
+                refreshTimeout, TestData.RetryInterval);
+
+            if (!retry.Success)
+            {
+                Assert.Fail($"Failed to refresh window in {refreshTimeout.TotalSeconds} seconds.");
+            }
+        }
+
 
         public static void LaunchApp()
         {
+            TestContext.Out.WriteLine($"[RUNNING TEST] {TestContext.CurrentContext.Test.FullName}");
+            System.Environment.SetEnvironmentVariable("BRIDGE_HOST_URL", $"{atlasEnvironment}");
             string appExecutable = TestData.AppExecutable;
             Application.Launch(appExecutable);
             Wait.UntilInputIsProcessed(TestData.FiveSecondsTimeout);
             App = Application.Attach("bridge-gui.exe");
+            RefreshWindow(TestData.OneMinuteTimeout);
+            Window.Focus();
+        }
 
-            try
+        private static RetryResult<bool> WaitUntilAppIsRunning()
+        {
+            RetryResult<bool> retry = Retry.WhileFalse(
+                () =>
+                {
+                    Process[] pname = Process.GetProcessesByName("Proton Mail Bridge");
+                    return pname.Length > 0;
+                },
+                TimeSpan.FromSeconds(30), TestData.RetryInterval);
+
+            return retry;
+        }
+        public static void CreateRunSpace()
+        {
+            bridgeDownloadURL = Environment.GetEnvironmentVariable($"{downloadEnvVariable}");
+            myRunSpace = RunspaceFactory.CreateRunspace();
+            myRunSpace.Open();
+            Pipeline cmd = myRunSpace.CreatePipeline($"New-Item env:{downloadEnvVariable} -Value {bridgeDownloadURL} -Force");
+            cmd.Invoke();
+            cmd = myRunSpace.CreatePipeline(@"Set-Location $env:CI_PROJECT_DIR\tests\e2e\ui_tests\windows_os\InstallerScripts");
+            cmd.Invoke();
+            cmd = myRunSpace.CreatePipeline(@"Set-ExecutionPolicy -ExecutionPolicy Unrestricted -Scope CurrentUser");
+            cmd.Invoke();
+        }
+
+        public static Collection<PSObject>? InstallBridge()
+        {
+            CreateRunSpace();
+            if (myRunSpace is not null)
             {
-                Window = App.GetMainWindow(new UIA3Automation(), TestData.ThirtySecondsTimeout);
+                Pipeline cmd = myRunSpace.CreatePipeline("Get-Location");
+                cmd.Invoke();
+                cmd = myRunSpace.CreatePipeline(@".\Get-BridgeInstaller.ps1");
+                var objects = cmd.Invoke();
+
+                return objects;
             }
-            catch (System.TimeoutException)
+
+            return null;
+        }
+
+        public static Collection<PSObject>? UninstallBridge()
+        {
+            CreateRunSpace();
+            if (myRunSpace is not null)
             {
-                Assert.Fail("Failed to get window of application!");
+                Pipeline cmd = myRunSpace.CreatePipeline(@".\Remove-Bridge.ps1");
+                var objects = cmd.Invoke();
+
+                return objects;
             }
+
+            return null;
+        }
+
+        public static Collection<PSObject>? RemoveBridgeCredentials()
+        {
+            CreateRunSpace();
+            if (myRunSpace is not null)
+            {
+                Pipeline cmd = myRunSpace.CreatePipeline(@".\Remove-BridgeCredentials.ps1");
+                var objects = cmd.Invoke();
+
+                return objects;
+            }
+
+            return null;
         }
     }
 }