adding new headless mode for chromium (#614)

* adding new headless mode for chromium

* chromium mock for other tests

Author: snamilikonda

src/Microsoft.PowerApps.TestEngine.Tests/TestInfra/PlaywrightTestInfraFunctionTests.cs

@@ -3,6 +3,7 @@
 
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Threading.Tasks;
 using Microsoft.Extensions.Logging;
 using Microsoft.Playwright;
@@ -56,6 +57,10 @@ public PlaywrightTestInfraFunctionTests()
             MockElementHandle = new Mock<IElementHandle>(MockBehavior.Strict);
             MockUserManager = new Mock<IUserManager>(MockBehavior.Strict);
             MockTestWebProvider = new Mock<ITestWebProvider>(MockBehavior.Strict);
+
+            // Mock Chromium behavior
+            MockPlaywrightObject.SetupGet(x => x.Chromium).Returns(new Mock<IBrowserType>(MockBehavior.Strict).Object);
+            MockPlaywrightObject.SetupGet(x => x.Chromium.Name).Returns("chromium");
         }
 
         [Theory]
@@ -154,6 +159,58 @@ public async Task SetupAsyncTest(string browser, string? device, int? screenWidt
             MockBrowser.Verify(x => x.NewContextAsync(It.Is<BrowserNewContextOptions>(y => verifyBrowserContextOptions(y))), Times.Once());
         }
 
+        [Theory]
+        [InlineData("chromium", null, true, true)]
+        [InlineData("Chromium", "msedge", true, true)]
+        [InlineData("Chromium", "chrome", true, true)]
+        [InlineData("Chromium", "msedge", false, false)]
+        [InlineData("cHromium", null, false, false)]
+        [InlineData("webkit", null, true, false)]
+        [InlineData("firefox", null, false, false)]
+        public async Task SetupAsync_HeadlessModeWithChromium_AddsNewHeadlessArgs(string browser, string? channel, bool headless, bool expectedResult)
+        {
+            // Arrange
+            MockTestState.Setup(x => x.GetTestEngineModules()).Returns(new List<Microsoft.PowerApps.TestEngine.Modules.ITestEngineModule>() { });
+
+            var testSettings = new TestSettings
+            {
+                Headless = headless,
+                Timeout = 30000
+            };
+
+            var browserConfig = new BrowserConfiguration
+            {
+                Browser = browser,
+                Channel = channel,
+            };
+
+            MockTestState.Setup(ts => ts.GetTestSettings()).Returns(testSettings);
+            MockSingleTestInstanceState.Setup(st => st.GetBrowserConfig()).Returns(browserConfig);
+            MockSingleTestInstanceState.Setup(st => st.GetLogger()).Returns(MockLogger.Object);
+            LoggingTestHelper.SetupMock(MockLogger);
+
+            MockBrowserType.Setup(c => c.Name).Returns(browser);
+            MockBrowserType.Setup(c => c.LaunchAsync(It.IsAny<BrowserTypeLaunchOptions>()))
+               .ReturnsAsync(Mock.Of<IBrowser>());
+            MockPlaywrightObject.Setup(p => p[browser]).Returns(MockBrowserType.Object);
+
+            var functions = new PlaywrightTestInfraFunctions(
+               MockTestState.Object,
+               MockSingleTestInstanceState.Object,
+               MockFileSystem.Object,
+               MockPlaywrightObject.Object
+            );
+
+            // Act
+            await functions.SetupAsync(Mock.Of<IUserManager>());
+
+            // Assert
+            MockBrowserType.Verify(c => c.LaunchAsync(It.Is<BrowserTypeLaunchOptions>(options =>
+                options.Args != null &&
+                options.Args.Contains("--headless=new")
+            )), expectedResult ? Times.Once : Times.Never);
+        }
+
         [Theory]
         [InlineData("en-US")]
         [InlineData("fr-FR")]

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

@@ -107,6 +107,14 @@ public async Task SetupAsync(IUserManager userManager)
             staticContext.Headless = launchOptions.Headless;
             staticContext.Timeout = launchOptions.Timeout;
 
+            //this is added for the new headless mode in chromium
+            if (testSettings.Headless && PlaywrightObject?.Chromium?.Name != null && string.Equals(browserConfig.Browser, PlaywrightObject.Chromium.Name, StringComparison.OrdinalIgnoreCase))
+            {
+                var headlessArgs = new[] { "--headless=new" };
+                launchOptions.Args = (launchOptions.Args ?? Array.Empty<string>()).Concat(headlessArgs).ToArray();
+                staticContext.Args = (staticContext.Args ?? Array.Empty<string>()).Concat(headlessArgs).ToArray();
+            }
+
             var browser = PlaywrightObject[browserConfig.Browser];
             if (browser == null)
             {