Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-10 04:36:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

test(BRIDGE-247): Automate Bridge 0% update rollout

Browse Source
This commit is contained in:
Gordana Zafirova
2024-12-30 16:01:10 +00:00
parent 16245a372e
commit 6878b3b5e0
2 changed files with 318 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

57
tests/e2e/ui_tests/windows_os/Tests/ZeroPercentUpdateTest.cs Normal file
View File

@ -0,0 +1,57 @@
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Text;
using System.Text.Encodings.Web;
using System.Text.Json;
using System.Text.Json.Nodes;
using System.Threading.Tasks;
using ProtonMailBridge.UI.Tests.Results;
using ProtonMailBridge.UI.Tests.TestsHelper;
using ProtonMailBridge.UI.Tests.Windows;
namespace ProtonMailBridge.UI.Tests.Tests
{
public class ZeroPercentUpdateTest : TestSession
{
private readonly LoginWindow _loginWindow = new();
private readonly SettingsMenuWindow _settingsMenuWindow = new();
private readonly HelpMenuWindow _helpMenuWindow = new();
private readonly ZeroPercentUpdateWindow _zeroPercentWindow = new();
[SetUp]
public void TestInitialize()
{
LaunchApp();
}
[Test]
[Category("ZeroPercentUpdateRollout")]
public void EnableBetaAccessVerifyBetaIsEnabledVerifyNotificationAndRestartBridge()
{
_zeroPercentWindow.ClickStartSetupButton();
_zeroPercentWindow.CLickCancelButton();
_helpMenuWindow.ClickHelpButton();
_zeroPercentWindow.SaveCurrentVersionAndTagNumber();
Thread.Sleep(2000);
ClientCleanup();
_zeroPercentWindow.editTheVault();
LaunchApp();
_loginWindow.SignIn(TestUserData.GetPaidUser());
_settingsMenuWindow.ClickSettingsButton();
_zeroPercentWindow.VerifyBetaAccessIsEnabled();
_zeroPercentWindow.RestartBridgeNotification();
_helpMenuWindow.ClickHelpButton();
Thread.Sleep(2000);
_zeroPercentWindow.VerifyVersionAndTagNumberOnRelaunch();
}
[TearDown]
public void TestCleanup()
{
ClientCleanup();
}
}
}

261
tests/e2e/ui_tests/windows_os/Windows/ZeroPercentUpdateWindow.cs Normal file
View File

@ -0,0 +1,261 @@
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Text;
using System.Text.Encodings.Web;
using System.Text.Json.Nodes;
using System.Text.Json;
using System.Threading.Tasks;
using ProtonMailBridge.UI.Tests.TestsHelper;
using FlaUI.Core.Definitions;
using FlaUI.Core.AutomationElements;
using FlaUI.Core.Input;
// using System.Windows.Forms;
namespace ProtonMailBridge.UI.Tests.Windows
{
public class ZeroPercentUpdateWindow : UIActions
{
private TextBox BridgeUpdateIsReady => Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.Text).And(cf.ByName("Bridge update is ready"))).AsTextBox();
private Button StartSetupButton => Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.Button).And(cf.ByName("Start setup"))).AsButton();
private Button CancelButton => Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.Button).And(cf.ByName("Cancel"))).AsButton();
private TextBox VersionBuildNumberTextBox => Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.Pane)).FindAllDescendants(cf => cf.ByControlType(ControlType.Text))[13].AsTextBox();
public void editTheVault()
{
try
{
string executablePath = Environment.GetEnvironmentVariable("UI_TEST_VAULT_EDITOR_EXE_PATH");
if (string.IsNullOrEmpty(executablePath))
{
Console.WriteLine("Executable path is not set in the environment variables.");
return;
}
if (!System.IO.File.Exists(executablePath))
{
Console.WriteLine($"Executable not found at path: {executablePath}");
return;
}
string readArgument = "read";
string writeArgument = "write";
string vaultJson;
// start the read process
ProcessStartInfo readProcessInfo = new ProcessStartInfo
{
FileName = executablePath,
Arguments = readArgument,
RedirectStandardOutput = true, // Capture the output
UseShellExecute = false, // Required to redirect output
CreateNoWindow = true // Optional: Run without showing a console window
};
//read the output
using (Process readProcess = Process.Start(readProcessInfo))
{
if (readProcess == null)
{
Assert.Fail("Failed to start the read process.");
return;
}
string output = readProcess.StandardOutput.ReadToEnd();
readProcess.WaitForExit();
// Write the output to a file
string desktopPath = Environment.GetFolderPath(Environment.SpecialFolder.Desktop);
vaultJson = Path.Combine(desktopPath, "vault.json");
File.WriteAllText(vaultJson, output);
Console.WriteLine($"Vault data has been written to: {vaultJson}");
}
//update the json file, UpdateChannel to early and UpdateRollout to 0
string jsonContent = File.ReadAllText(vaultJson);
JsonNode json = JsonNode.Parse(jsonContent);
if (json == null)
{
Assert.Fail("Can not parse the vault.json file");
return;
}
else
{
var settingsNode = json["Settings"];
settingsNode["UpdateChannel"] = "early";
settingsNode["UpdateRollout"] = 0;
// must use UnsafeRelaxedJsonEscaping for the serialization of the + character in the timestamp
File.WriteAllText(vaultJson, json.ToJsonString(new JsonSerializerOptions { WriteIndented = true, Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping }));
}
// read the new content from the json file
jsonContent = File.ReadAllText(vaultJson);
// start the write process
Process writeProcess = new Process
{
StartInfo = new ProcessStartInfo
{
FileName = executablePath,
Arguments = writeArgument,
RedirectStandardInput = true, // Pass input via standard input
RedirectStandardOutput = true, // Capture standard output
RedirectStandardError = true, // Capture error output
UseShellExecute = false, // Required for redirection
CreateNoWindow = true // Run without showing a window
}
};
writeProcess.Start();
// pass the json file content to the process
using (StreamWriter writer = writeProcess.StandardInput)
{
if (writer.BaseStream.CanWrite)
{
writer.Write(jsonContent);
}
}
// read the output and errors
string writeOutput = writeProcess.StandardOutput.ReadToEnd();
string error = writeProcess.StandardError.ReadToEnd();
writeProcess.WaitForExit();
if (!string.IsNullOrWhiteSpace(writeOutput))
{
Console.WriteLine("Output from write process:");
Console.WriteLine(writeOutput);
}
if (!string.IsNullOrWhiteSpace(error))
{
Console.WriteLine("Error from write process:");
Console.WriteLine(error);
}
// Check exit code for success/failure
if (writeProcess.ExitCode != 0)
{
Assert.Fail("Write process exited with error code: " + writeProcess.ExitCode);
}
}
catch (Exception ex)
{
Assert.Fail(ex.Message);
}
}
public ZeroPercentUpdateWindow VerifyBetaAccessIsEnabled()
{
CheckBox BetaAccess = Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.CheckBox).And(cf.ByName("Beta access toggle"))).AsCheckBox();
Assert.That(BetaAccess.IsToggled, Is.True);
return this;
}
public ZeroPercentUpdateWindow RestartBridgeNotification()
{
WaitUntilElementIsVisible(() => BridgeUpdateIsReady, 60);
Assert.That(BridgeUpdateIsReady.IsAvailable, Is.True);
Button RestartBridge = Window.FindFirstDescendant(cf => cf.ByControlType(ControlType.Button).And(cf.ByName("Restart Bridge"))).AsButton();
RestartBridge.Click();
Thread.Sleep(5000);
LaunchApp();
Window.Focus();
Thread.Sleep(5000);
return this;
}
private static int[] PreviousVersionNumbers = null;
private static string PreviousTagNumber = null;
private int[] GetVersionNumbers()
{
string text = VersionBuildNumberTextBox.Name;
string versionString = text.Substring(text.IndexOf("v") + 1).Split(' ')[0];
return Array.ConvertAll(versionString.Split('.'), int.Parse);
}
private string GetTagNumber()
{
string text = VersionBuildNumberTextBox.Name;
return text.Substring(text.IndexOf("(br-") + 4).Split(')')[0];
}
public void SaveCurrentVersionAndTagNumber()
{
PreviousVersionNumbers = GetVersionNumbers();
PreviousTagNumber = GetTagNumber();
}
public void VerifyVersionAndTagNumberOnRelaunch()
{
int[] newVersionNumbers = GetVersionNumbers();
string newTagNumber = GetTagNumber();
if (PreviousVersionNumbers == null || PreviousTagNumber == null)
{
throw new Exception("Previous version or tag number is not set.");
}
bool isVersionGreater = false;
for (int i = 0; i < PreviousVersionNumbers.Length; i++)
{
if (newVersionNumbers[i] > PreviousVersionNumbers[i])
{
isVersionGreater = true;
break;
}
else if (newVersionNumbers[i] < PreviousVersionNumbers[i])
{
throw new Exception($"Current version {string.Join('.', newVersionNumbers)} is not greater than previous version {string.Join('.', PreviousVersionNumbers)}.");
}
}
if (!isVersionGreater)
{
throw new Exception($"Current version {string.Join('.', newVersionNumbers)} is not greater than previous version {string.Join('.', PreviousVersionNumbers)}.");
}
if (newTagNumber == PreviousTagNumber)
{
throw new Exception($"Current tag number (br-{newTagNumber}) is the same as the previous tag number (br-{PreviousTagNumber}).");
}
}
public ZeroPercentUpdateWindow ClickStartSetupButton()
{
StartSetupButton.Click();
return this;
}
public ZeroPercentUpdateWindow CLickCancelButton()
{
CancelButton.Click();
return this;
}
private void WaitUntilElementIsVisible(Func<AutomationElement> findElementFunc, int numOfSeconds)
{
TimeSpan timeout = TimeSpan.FromSeconds(numOfSeconds);
Stopwatch stopwatch = Stopwatch.StartNew();
while (stopwatch.Elapsed < timeout)
{
//if element is visible the processing is completed
var element = findElementFunc();
if (element != null)
{
return;
}
Wait.UntilInputIsProcessed();
Thread.Sleep(500);
}
}
}
}