Pause module changes

Author: v-raghulraja

src/Microsoft.PowerApps.TestEngine/Modules/TestEngineExtensionChecker.cs

@@ -334,6 +334,21 @@ public bool VerifyContainsValidNamespacePowerFxFunctions(TestSettingExtensions s
                 stream.Position = 0;
                 ModuleDefinition module = ModuleDefinition.ReadModule(stream);
 
+                // Check if PauseModule exists and inspect its IsPreviewNamespaceEnabled property
+                var pauseModule = module.Types.FirstOrDefault(t => t.Name == "PauseModule");
+                if (pauseModule != null)
+                {
+                    // Check if the PauseModule has IsPreviewNamespaceEnabled property
+                    var previewProperty = pauseModule.Properties.FirstOrDefault(p => p.Name == "IsPreviewNamespaceEnabled");
+                    if (previewProperty != null)
+                    {
+                        // If PauseModule has IsPreviewNamespaceEnabled property, enable Preview namespace
+                        // The property's value will be determined at runtime based on YAML settings
+                        settings.AllowPowerFxNamespaces.Add(NAMESPACE_PREVIEW);
+                        Logger?.LogInformation("Auto-enabled Preview namespace due to PauseModule.IsPreviewNamespaceEnabled property.");
+                    }
+                }
+
                 // Get the source code of the assembly as will be used to check Power FX Namespaces
                 var code = DecompileModuleToCSharp(assembly);
 
@@ -382,6 +397,13 @@ public bool VerifyContainsValidNamespacePowerFxFunctions(TestSettingExtensions s
                     // Extension Module Check are based on constructor
                     if (type.BaseType != null && type.BaseType.Name == "ReflectionFunction")
                     {
+                        // Special handling for PauseFunction - allow root namespace when PauseModule is present
+                        if (type.Name == "PauseFunction" && pauseModule != null)
+                        {
+                            Logger?.LogInformation($"Allowing PauseFunction in root namespace due to PauseModule presence.");
+                            continue; // Skip namespace validation for PauseFunction
+                        }
+
                         var constructors = type.GetConstructors();
 
                         if (constructors.Count() == 0)

src/Microsoft.PowerApps.TestEngine/PowerFx/Functions/PauseFunction.cs

@@ -102,7 +102,6 @@ public void Setup(TestSettings settings)
             powerFxConfig.AddFunction(new AssertNotErrorFunction(Logger));
             powerFxConfig.AddFunction(new SetPropertyFunction(_testWebProvider, Logger));
             powerFxConfig.AddFunction(new IsMatchFunction(Logger));
-            powerFxConfig.AddFunction(new PauseFunction(TestInfraFunctions, TestState, Logger));
 
             if (settings != null && settings.ExtensionModules != null && settings.ExtensionModules.Enable)
             {
@@ -113,11 +112,7 @@ public void Setup(TestSettings settings)
                 }
                 foreach (var module in modules)
                 {
-                    if (module.GetType().Name.Contains("Pause"))
-                    {
-                        Logger.LogInformation("Skipping pause module as core Pause function is enabled");
-                        continue;
-                    }
+                    // Register all modules including pause module
                     module.RegisterPowerFxFunction(powerFxConfig, TestInfraFunctions, _testWebProvider, SingleTestInstanceState, TestState, _fileSystem);
                 }
             }

src/Microsoft.PowerApps.TestEngine/PowerFx/PowerFxEngine.cs

@@ -5,7 +5,6 @@
 using Microsoft.PowerApps.TestEngine.Config;
 using Microsoft.PowerApps.TestEngine.TestInfra;
 using Microsoft.PowerFx;
-using Microsoft.PowerFx.Core.Utils;
 using Microsoft.PowerFx.Types;
 
 namespace testengine.module
@@ -18,19 +17,28 @@ public class PauseFunction : ReflectionFunction
         private readonly ITestInfraFunctions _testInfraFunctions;
         private readonly ITestState _testState;
         private readonly ILogger _logger;
+        private readonly PauseModule _pauseModule;
 
-        public PauseFunction(ITestInfraFunctions testInfraFunctions, ITestState testState, ILogger logger)
-            : base(DPath.Root.Append(new DName("Preview")), "Pause", FormulaType.Blank)
+        // Register in root (no Preview namespace)
+        public PauseFunction(ITestInfraFunctions testInfraFunctions, ITestState testState, ILogger logger, PauseModule pauseModule = null)
+            : base("Pause", FormulaType.Blank)
         {
             _testInfraFunctions = testInfraFunctions;
             _testState = testState;
             _logger = logger;
+            _pauseModule = pauseModule;
         }
 
         public BlankValue Execute()
         {
-            _logger.LogInformation("------------------------------\n\n" +
-                 "Executing Pause function.");
+            _logger.LogInformation("------------------------------\n\nExecuting Pause function.");
+
+            // Check if PauseModule's IsPreviewNamespaceEnabled property allows execution
+            if (_pauseModule != null && !_pauseModule.IsPreviewNamespaceEnabled)
+            {
+                _logger.LogInformation("Pause function is disabled - Preview namespace not enabled in PauseModule configuration.");
+                return FormulaValue.NewBlank();
+            }
 
             if (!_testState.GetTestSettings().Headless)
             {

src/testengine.module.pause/PauseFunction.cs

@@ -16,16 +16,31 @@ namespace testengine.module
     [Export(typeof(ITestEngineModule))]
     public class PauseModule : ITestEngineModule
     {
+        /// <summary>
+        /// True when YAML testSettings.extensionModules.allowPowerFxNamespaces contains "Preview".
+        /// Read-only externally; set during registration when TestSettings are available.
+        /// </summary>
+        public bool IsPreviewNamespaceEnabled { get; private set; } = false;
+
         public void ExtendBrowserContextOptions(BrowserNewContextOptions options, TestSettings settings)
         {
-
+            // If called first, try to initialize from provided settings as well
+            if (settings != null && settings.ExtensionModules != null && settings.ExtensionModules.AllowPowerFxNamespaces != null)
+            {
+                IsPreviewNamespaceEnabled = settings.ExtensionModules.AllowPowerFxNamespaces.Contains("Preview");
+            }
         }
 
         public void RegisterPowerFxFunction(PowerFxConfig config, ITestInfraFunctions testInfraFunctions, ITestWebProvider testWebProvider, ISingleTestInstanceState singleTestInstanceState, ITestState testState, IFileSystem fileSystem)
         {
+            // Initialize the read-only property from YAML via TestState
+            var testSettings = testState?.GetTestSettings();
+            IsPreviewNamespaceEnabled = testSettings?.ExtensionModules?.AllowPowerFxNamespaces?.Contains("Preview") == true;
+
             ILogger logger = singleTestInstanceState.GetLogger();
-            config.AddFunction(new PauseFunction(testInfraFunctions, testState, logger));
-            logger.LogInformation("Registered Pause()");
+            // Pass this PauseModule instance to PauseFunction so it can check IsPreviewNamespaceEnabled
+            config.AddFunction(new PauseFunction(testInfraFunctions, testState, logger, this));
+            logger.LogInformation($"Registered Pause() with Preview namespace enabled: {IsPreviewNamespaceEnabled}");
         }
 
         public async Task RegisterNetworkRoute(ITestState state, ISingleTestInstanceState singleTestInstanceState, IFileSystem fileSystem, IPage Page, NetworkRequestMock mock)