From 6078fd772777e1d9cca9ff1adccd7b86e1c96fa7 Mon Sep 17 00:00:00 2001
From: Lorenzo Mattei <lore.mattei@gmail.com>
Date: Mon, 13 May 2024 18:18:20 +0200
Subject: [PATCH] Update tests to handle new Settings UI (#2844)

---
 .maestro/release_tests/emailprotection.yaml | 23 +++++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/.maestro/release_tests/emailprotection.yaml b/.maestro/release_tests/emailprotection.yaml
index 4b2526dbbf..1b4f0a4326 100644
--- a/.maestro/release_tests/emailprotection.yaml
+++ b/.maestro/release_tests/emailprotection.yaml
@@ -14,10 +14,25 @@ tags:
     file: ../shared/onboarding.yaml
 
 - tapOn: Settings
-- scroll
-- scroll
-- assertVisible: Email Protection
-- tapOn: Email Protection
+# Handling two different flows because of the current experiment
+# TODO: Remove the unused flow when the experiment is completed.
+- runFlow:
+    # New Settings UI
+    when:
+      visible: "Email Protection"
+    commands:
+      - tapOn: Email Protection
+      - assertVisible: Enable Email Protection
+      - tapOn: Enable Email Protection
+- runFlow:
+    # Old Settings UI
+    when:
+      notVisible: "Email Protection"
+    commands:
+      - scroll
+      - scroll
+      - assertVisible: Email Protection
+      - tapOn: Email Protection
 - assertVisible: Email privacy, simplified.
 - assertVisible:
     id: searchEntry