Extend browser user authentication provide to handle Condition Access

src/Microsoft.PowerApps.TestEngine/TestInfra/PlaywrightTestInfraFunctions.cs

@@ -105,6 +105,14 @@ public async Task SetupAsync(IUserManager userManager)
                 {
                     launchOptions.Channel = browserConfig.Channel;
                 }
+                else
+                {
+                    if (browserConfig.Browser?.ToLower() == "chromium")
+                    {
+                        // Default to msedge browser
+                        launchOptions.Channel = "msedge";
+                    }
+                }
 
                 Browser = await browser.LaunchAsync(launchOptions);
                 _singleTestInstanceState.GetLogger().LogInformation("Browser setup finished");
@@ -160,6 +168,15 @@ public async Task SetupAsync(IUserManager userManager)
                 {
                     staticContext.Channel = browserConfig.Channel;
                 }
+                else
+                {
+                    if (browserConfig.Browser?.ToLower() == "chromium")
+                    {
+                        // Default to msedge browser
+                        launchOptions.Channel = "msedge";
+                    }
+                }
+
                 BrowserContext = await browser.LaunchPersistentContextAsync(location, staticContext);
             }
             else

src/testengine.user.browser.tests/BrowserUserManagerModuleTests.cs

@@ -12,7 +12,6 @@ namespace testengine.user.environment.tests
 {
     public class BrowserUserManagerModuleTests
     {
-        private Mock<IBrowserContext> MockBrowserState;
         private Mock<ITestInfraFunctions> MockTestInfraFunctions;
         private Mock<ITestState> MockTestState;
         private Mock<ISingleTestInstanceState> MockSingleTestInstanceState;
@@ -26,7 +25,6 @@ public class BrowserUserManagerModuleTests
 
         public BrowserUserManagerModuleTests()
         {
-            MockBrowserState = new Mock<IBrowserContext>(MockBehavior.Strict);
             MockTestInfraFunctions = new Mock<ITestInfraFunctions>(MockBehavior.Strict);
             MockTestState = new Mock<ITestState>(MockBehavior.Strict);
             MockSingleTestInstanceState = new Mock<ISingleTestInstanceState>(MockBehavior.Strict);
@@ -59,9 +57,24 @@ public async Task LoginWithBrowserState(bool exists, bool isDirectoryCreated, st
             userManager.GetFiles = (path) => files.Split(',');
             userManager.Page = MockPage.Object;
 
+            MockSingleTestInstanceState.Setup(x => x.GetLogger()).Returns(MockLogger.Object);
+
+            MockTestState.Setup(x => x.GetTestSuiteDefinition()).Returns(new TestSuiteDefinition() { Persona = "User1" });
+            MockEnvironmentVariable.Setup(x => x.GetVariable("User1")).Returns("[email protected]");
+
+            MockBrowserContext.Setup(x => x.Pages).Returns(new List<IPage>() { MockPage.Object });
+            MockPage.Setup(x => x.Url).Returns("https://localhost/test");
+
+            MockLogger.Setup(x => x.Log(
+                It.IsAny<LogLevel>(),
+                It.IsAny<EventId>(),
+                It.IsAny<It.IsAnyType>(),
+                It.IsAny<Exception>(),
+                (Func<It.IsAnyType, Exception, string>)It.IsAny<object>()));
+
             // Act
-            await userManager.LoginAsUserAsync("*",
-                MockBrowserState.Object,
+            await userManager.LoginAsUserAsync("https://localhost/test",
+                MockBrowserContext.Object,
                 MockTestState.Object,
                 MockSingleTestInstanceState.Object,
                 MockEnvironmentVariable.Object,

src/testengine.user.browser/BrowserUserManagerModule.cs

@@ -47,16 +47,15 @@
         {
             Context = context;
 
+            var logger = singleTestInstanceState.GetLogger();
+
             if (!DirectoryExists(Location))
             {
+                logger.LogInformation($"Creating {Location}");
                 CreateDirectory(Location);
             }
 
-            if (GetFiles(Location).Count() == 0)
-            {
-                ValidatePage();
-                await Page.PauseAsync();
-            }
+            await LoginComplete(desiredUrl, testState, environmentVariable, logger);
         }
 
         private void ValidatePage()
@@ -66,5 +65,71 @@
                 Page = Context.Pages.First();
             }
         }
+
+        public async Task LoginComplete(string desiredUrl, ITestState testState, IEnvironmentVariable environmentVariable, ILogger logger)
+        {
+            var complete = false;
+
+            var persona = testState.GetTestSuiteDefinition().Persona;
+            var personaEmail = environmentVariable.GetVariable(persona);
+
+            if (string.IsNullOrEmpty(personaEmail))
+            {
+                logger.LogInformation($"Missing user persona {persona} email. Prompting user");
+                Console.Write("Persona Email? ");
+                personaEmail = Console.ReadLine();
+            }
+
+            var started = DateTime.Now;
+
+            while (!complete)
+            {
+                foreach (var page in Context.Pages)
+                {
+                    var host = new Uri(desiredUrl).Host;
+                    if (page.Url.Contains($"https://{host}"))
+                    {
+                        logger.LogInformation($"Found destination url");
+                        complete = true;
+                        break;
+                    }
+
+                    if (page.Url.Contains("common/fido"))
+                    {
+                        logger.LogInformation($"Login required");
+                        Console.WriteLine("Login required");
+                        await page.PauseAsync();
+                    }
+
+                    if (page.Url.Contains("oauth2/authorize"))
+                    {
+                        if (await page.IsVisibleAsync($"[data-test-id=\"{personaEmail}\"]"))
+                        {
+                            logger.LogInformation($"Selecting {personaEmail}");
+                            await page.ClickAsync($"[data-test-id=\"{personaEmail}\"]");
+                        }
+                    }
+
+                    if (page.Url.Contains("mcas.ms/aad_login"))
+                    {
+                        // TODO: Handle localized values
+                        if (await page.GetByRole(AriaRole.Button, new() { Name = "Continue with current profile" }).IsVisibleAsync())
+                        {
+                            logger.LogInformation($"Continue with Microsoft Conditional Access");
+                            await page.GetByRole(AriaRole.Button, new() { Name = "Continue with current profile" }).ClickAsync();
+                        }
+                    }
+                }
+
+                if (!complete)
+                {
+                    if (DateTime.Now.Subtract(started).TotalMinutes > 5)
+                    {
+                        throw new Exception("Unable to complete login");
+                    }
+                    Thread.Sleep(500);
+                }
+            }
+        }
     }
 }