File I/O abstraction part 21: Migrate IConfigurationManager and IFeatureProvider (#17906)

shenglol · web-flow · commit 12ee932c2842 · 2025-08-27T10:47:48.000-07:00
Refactored `IConfigurationManager` and `IFeatureProvider` (along with their implementations and tests) to use the Bicep.IO APIs. ###### Microsoft Reviewers: [Open in CodeFlow](https://microsoft.github.io/open-pr/?codeflow=https://github.com/Azure/bicep/pull/17906)
diff --git a/src/Bicep.Cli/Commands/TestCommand.cs b/src/Bicep.Cli/Commands/TestCommand.cs
@@ -44,7 +44,7 @@ public async Task<int> RunAsync(TestArguments args)
         {
             var inputUri = this.inputOutputArgumentsResolver.ResolveInputArguments(args);
             ArgumentHelper.ValidateBicepFile(inputUri);
-            var features = featureProviderFactory.GetFeatureProvider(inputUri.ToUri());
+            var features = featureProviderFactory.GetFeatureProvider(inputUri);
 
             if (!features.TestFrameworkEnabled)
             {
diff --git a/src/Bicep.Core.IntegrationTests/Decorators/DecoratorTests.cs b/src/Bicep.Core.IntegrationTests/Decorators/DecoratorTests.cs
@@ -6,6 +6,7 @@
 using Bicep.Core.UnitTests.Assertions;
 using Bicep.Core.UnitTests.Features;
 using Bicep.Core.UnitTests.Utils;
+using Bicep.TextFixtures.Utils;
 using FluentAssertions;
 using FluentAssertions.Execution;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
@@ -39,6 +40,8 @@ public void ParameterDecorator_MissingDeclaration_ExpectedParameterDeclaration()
         [TestMethod]
         public void ParameterDecorator_AttachedToOtherKindsOfDeclarations_CannotBeUsedAsDecoratorSpecificToTheDeclarations()
         {
+            var testCompiler = TestCompiler.ForMockFileSystemCompilation();
+
             var mainUri = new Uri("file:///main.bicep");
             var moduleUri = new Uri("file:///module.bicep");
 
diff --git a/src/Bicep.Core.UnitTests/Configuration/ConfigurationManagerTests.cs b/src/Bicep.Core.UnitTests/Configuration/ConfigurationManagerTests.cs
diff --git a/src/Bicep.Core.UnitTests/Configuration/PatchingConfigurationManager.cs b/src/Bicep.Core.UnitTests/Configuration/PatchingConfigurationManager.cs
@@ -2,6 +2,7 @@
 // Licensed under the MIT License.
 
 using Bicep.Core.Configuration;
+using Bicep.IO.Abstraction;
 
 namespace Bicep.Core.UnitTests.Configuration;
 
@@ -16,5 +17,5 @@ public PatchingConfigurationManager(ConfigurationManager configurationManager, F
         this.patchFunc = patchFunc;
     }
 
-    public RootConfiguration GetConfiguration(Uri sourceFileUri) => patchFunc(configurationManager.GetConfiguration(sourceFileUri));
+    public RootConfiguration GetConfiguration(IOUri sourceFileUri) => patchFunc(configurationManager.GetConfiguration(sourceFileUri));
 }
diff --git a/src/Bicep.Core.UnitTests/Configuration/ProviderConfigurationTests.cs b/src/Bicep.Core.UnitTests/Configuration/ProviderConfigurationTests.cs
@@ -8,6 +8,7 @@
 using Bicep.Core.UnitTests.Assertions;
 using Bicep.IO.Abstraction;
 using Bicep.IO.FileSystem;
+using Bicep.TextFixtures.IO;
 using FluentAssertions;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 
@@ -66,23 +67,18 @@ public void ExtensionConfiguration_default_configuration_returns_known_list_of_b
     [TestMethod]
     public void ExtensionConfiguration_user_provided_configuration_overrides_default_configuration()
     {
-        var bicepConfigFileName = "bicepconfig.json";
-        var fs = new MockFileSystem(new Dictionary<string, MockFileData>
-        {
-            [bicepConfigFileName] = new("""
+        var fileSet = InMemoryTestFileSet.Create(("bicepconfig.json", """
             {
-                "extensions": {
-                    "foo": "br:example.azurecr.io/some/fake/path:1.0.0",
-                    "az": "br:mcr.microsoft.com/bicep/extensions/az:0.2.3"
-                }
+              "extensions": {
+                "foo": "br:example.azurecr.io/some/fake/path:1.0.0",
+                "az": "br:mcr.microsoft.com/bicep/extensions/az:0.2.3"
+              }
             }
-            """)
-        });
+            """));
+
+        var configManager = new ConfigurationManager(fileSet.FileExplorer);
+        var config = configManager.GetConfiguration(fileSet.GetUri("main.bicep"));
 
-        var fileExplorer = new FileSystemFileExplorer(fs);
-        var configManager = new ConfigurationManager(fileExplorer);
-        var testFilePath = fs.Path.GetFullPath(bicepConfigFileName);
-        var config = configManager.GetConfiguration(PathHelper.FilePathToFileUrl(testFilePath));
         config.Diagnostics.Should().BeEmpty();
         config.Should().NotBeNull();
         config!.Extensions.Should().NotBeNull();
diff --git a/src/Bicep.Core.UnitTests/Features/FeatureProviderTests.cs b/src/Bicep.Core.UnitTests/Features/FeatureProviderTests.cs
@@ -7,6 +7,7 @@
 using Bicep.Core.Features;
 using Bicep.Core.UnitTests.Assertions;
 using Bicep.IO.FileSystem;
+using Bicep.TextFixtures.IO;
 using FluentAssertions;
 using Microsoft.VisualStudio.TestTools.UnitTesting;
 
@@ -21,44 +22,47 @@ public class FeatureProviderTests
     [TestMethod]
     public void PropertyLookup_WithNothingConfigured_ReturnsDefault()
     {
-        var fileSystem = new MockFileSystem(new Dictionary<string, MockFileData>
-        {
-            [CreatePath("repo")] = new MockDirectoryData(),
-            [CreatePath("repo/bicepconfig.json")] = @"{""experimentalFeaturesEnabled"": {}}",
-        });
+        var fileSet = InMemoryTestFileSet.Create(("repo/bicepconfig.json", """
+            {
+              "experimentalFeaturesEnabled": {}
+            }
+            """));
 
-        var fileExplorer = new FileSystemFileExplorer(fileSystem);
-        var configManager = new ConfigurationManager(fileExplorer);
-        var configuration = configManager.GetConfiguration(new Uri(this.CreatePath("repo/main.bicep")));
-        var fpm = new FeatureProviderFactory(configManager, fileExplorer);
+        var configManager = new ConfigurationManager(fileSet.FileExplorer);
+        var configuration = configManager.GetConfiguration(fileSet.GetUri("repo/main.bicep"));
+        var fpm = new FeatureProviderFactory(configManager, fileSet.FileExplorer);
 
-        var control = fpm.GetFeatureProvider(new Uri("file:///main.bicep"));
-        var sut = fpm.GetFeatureProvider(new Uri(this.CreatePath("repo/main.bicep")));
+        var control = fpm.GetFeatureProvider(fileSet.GetUri("main.bicep"));
+        var sut = fpm.GetFeatureProvider(fileSet.GetUri("repo/main.bicep"));
         sut.SymbolicNameCodegenEnabled.Should().Be(control.SymbolicNameCodegenEnabled);
     }
 
     [TestMethod]
     public void PropertyLookup_WithFeatureEnabledViaBicepConfig_ReturnsTrue()
     {
-        var fileSystem = new MockFileSystem(new Dictionary<string, MockFileData>
-        {
-            [CreatePath("repo")] = new MockDirectoryData(),
-            [CreatePath("repo/bicepconfig.json")] = @"{""experimentalFeaturesEnabled"": {}}",
-            [CreatePath("repo/subdir")] = new MockDirectoryData(),
-            [CreatePath("repo/subdir/bicepconfig.json")] = @"{""experimentalFeaturesEnabled"": {""symbolicNameCodegen"": true}}",
-        });
-        var fileExplorer = new FileSystemFileExplorer(fileSystem);
-        var configManager = new ConfigurationManager(fileExplorer);
-        var configuration = configManager.GetConfiguration(new Uri(this.CreatePath("repo/main.bicep")));
-        var fpm = new FeatureProviderFactory(configManager, fileExplorer);
+        var fileSet = InMemoryTestFileSet.Create(
+            ("repo/bicepconfig.json", """
+                {
+                  "experimentalFeaturesEnabled": {}
+                }
+                """),
+            ("repo/subdir/bicepconfig.json", """
+                {
+                  "experimentalFeaturesEnabled": {
+                    "symbolicNameCodegen": true
+                  }
+                }
+                """));
 
-        var control = fpm.GetFeatureProvider(new Uri("file:///main.bicep"));
+        var configManager = new ConfigurationManager(fileSet.FileExplorer);
+        var configuration = configManager.GetConfiguration(fileSet.GetUri("repo/main.bicep"));
+        var fpm = new FeatureProviderFactory(configManager, fileSet.FileExplorer);
+
+        var control = fpm.GetFeatureProvider(fileSet.GetUri("main.bicep"));
         control.SymbolicNameCodegenEnabled.Should().BeFalse();
-        var mainDirFeatures = fpm.GetFeatureProvider(new Uri(this.CreatePath("repo/main.bicep")));
+        var mainDirFeatures = fpm.GetFeatureProvider(fileSet.GetUri("repo/main.bicep"));
         mainDirFeatures.SymbolicNameCodegenEnabled.Should().BeFalse();
-        var subDirFeatures = fpm.GetFeatureProvider(new Uri(this.CreatePath("repo/subdir/module.bicep")));
+        var subDirFeatures = fpm.GetFeatureProvider(fileSet.GetUri("repo/subdir/module.bicep"));
         subDirFeatures.SymbolicNameCodegenEnabled.Should().BeTrue();
     }
-
-    private string CreatePath(string path) => Path.Combine(this.TestContext.ResultsDirectory!, path.Replace('/', Path.DirectorySeparatorChar));
 }
diff --git a/src/Bicep.Core.UnitTests/Features/OverridableFeatureProviderFactory.cs b/src/Bicep.Core.UnitTests/Features/OverridableFeatureProviderFactory.cs
@@ -2,6 +2,7 @@
 // Licensed under the MIT License.
 
 using Bicep.Core.Features;
+using Bicep.IO.Abstraction;
 
 namespace Bicep.Core.UnitTests.Features;
 
@@ -16,6 +17,6 @@ public OverriddenFeatureProviderFactory(FeatureProviderFactory factory, FeatureP
         this.overrides = overrides;
     }
 
-    public IFeatureProvider GetFeatureProvider(Uri templateUri)
-        => new OverriddenFeatureProvider(factory.GetFeatureProvider(templateUri), overrides);
+    public IFeatureProvider GetFeatureProvider(IOUri sourceFileUri)
+        => new OverriddenFeatureProvider(factory.GetFeatureProvider(sourceFileUri), overrides);
 }
diff --git a/src/Bicep.Core.UnitTests/Registry/ArtifactDispatcherTests.cs b/src/Bicep.Core.UnitTests/Registry/ArtifactDispatcherTests.cs
@@ -138,7 +138,7 @@ public async Task GetModuleRestoreStatus_ConfigurationChanges_ReturnsCachedStatu
         {
             // Arrange.
             var configManagerMock = StrictMock.Of<IConfigurationManager>();
-            configManagerMock.SetupSequence(m => m.GetConfiguration(It.IsAny<Uri>()))
+            configManagerMock.SetupSequence(m => m.GetConfiguration(It.IsAny<IOUri>()))
                 .Returns(BicepTestConstants.CreateMockConfiguration())
                 .Returns(BicepTestConstants.CreateMockConfiguration())
                 .Returns(changedConfiguration);
diff --git a/src/Bicep.Core.UnitTests/Registry/OciModuleRegistryTests.cs b/src/Bicep.Core.UnitTests/Registry/OciModuleRegistryTests.cs
@@ -668,7 +668,7 @@ public async Task RestoreModuleWithSource_ShouldRestoreSourceToDisk(bool publish
             var template = BinaryData.FromString(jsonContentsV1);
 
             var featureProviderFactoryMock = StrictMock.Of<IFeatureProviderFactory>();
-            featureProviderFactoryMock.Setup(x => x.GetFeatureProvider(bicepFile.Uri)).Returns(bicepFile.Features);
+            featureProviderFactoryMock.Setup(x => x.GetFeatureProvider(bicepFile.FileHandle.Uri)).Returns(bicepFile.Features);
 
             SourceArchive? sourceArchive = null;
             if (publishSource)
@@ -746,7 +746,7 @@ private OciArtifactReference CreateModuleReference(BicepSourceFile referencingFi
             featureProviderMock.Setup(m => m.CacheRootDirectory).Returns(cacheRootDirectory);
 
             var featureProviderFactoryMock = StrictMock.Of<IFeatureProviderFactory>();
-            featureProviderFactoryMock.Setup(m => m.GetFeatureProvider(parentModuleUri)).Returns(featureProviderMock.Object);
+            featureProviderFactoryMock.Setup(m => m.GetFeatureProvider(parentModuleUri.ToIOUri())).Returns(featureProviderMock.Object);
 
             var parentModuleFile = new BicepFile(
                 BicepTestConstants.FileExplorer.GetFile(parentModuleUri.ToIOUri()),
diff --git a/src/Bicep.Core.UnitTests/Utils/OciRegistryHelper.cs b/src/Bicep.Core.UnitTests/Utils/OciRegistryHelper.cs
@@ -90,10 +90,7 @@ public static (OciArtifactRegistry, FakeRegistryBlobClient) CreateModuleRegistry
             clientFactory.Setup(m => m.CreateAuthenticatedBlobClient(It.IsAny<CloudConfiguration>(), registryUri, repository)).Returns(client);
 
             var containerRegistryManager = new AzureContainerRegistryManager(clientFactory.Object);
-            var configurationManager = BicepTestConstants.ConfigurationManager;
-
-            var parentUri = new Uri("http://test.bicep", UriKind.Absolute);
-            var configuration = configurationManager.GetConfiguration(parentUri);
+            var configuration = IConfigurationManager.GetBuiltInConfiguration();
 
             using var compiledStream = new BufferedMemoryStream();
 
diff --git a/src/Bicep.Core/Configuration/ConfigurationManager.cs b/src/Bicep.Core/Configuration/ConfigurationManager.cs
@@ -28,15 +28,14 @@ public ConfigurationManager(IFileExplorer fileExplorer)
             this.fileExplorer = fileExplorer;
         }
 
-        public RootConfiguration GetConfiguration(Uri sourceFileUri)
+        public RootConfiguration GetConfiguration(IOUri sourceFileUri)
         {
             if (!sourceFileUri.IsFile)
             {
                 return GetDefaultConfiguration();
             }
 
-            var sourceFileIOUri = sourceFileUri.ToIOUri();
-            var sourceDirectory = this.fileExplorer.GetFile(sourceFileIOUri).GetParent();
+            var sourceDirectory = this.fileExplorer.GetFile(sourceFileUri).GetParent();
 
             if (!configFileLookupCache.GetOrAdd(sourceDirectory, LookupConfigurationFile).IsSuccess(out var configFileHandle, out var lookupDiagnostic))
             {
@@ -81,9 +80,9 @@ public void PurgeCache()
             return returnVal;
         }
 
-        public void RemoveConfigCacheEntry(IOUri identifier)
+        public void RemoveConfigCacheEntry(IOUri configFileUri)
         {
-            var configFileHandle = this.fileExplorer.GetFile(identifier);
+            var configFileHandle = this.fileExplorer.GetFile(configFileUri);
             if (loadedConfigCache.TryRemove(configFileHandle, out _))
             {
                 // If a config file has been removed from a workspace, the lookup cache is no longer valid.
diff --git a/src/Bicep.Core/Configuration/IConfigurationManager.cs b/src/Bicep.Core/Configuration/IConfigurationManager.cs
@@ -4,6 +4,7 @@
 using System.Reflection;
 using System.Text.Json;
 using Bicep.Core.Json;
+using Bicep.IO.Abstraction;
 
 namespace Bicep.Core.Configuration
 {
@@ -17,7 +18,7 @@ public interface IConfigurationManager
         /// </summary>
         /// <param name="sourceFileUri">The URI of the source file to get configuration for.</param>
         /// <returns>The configuration for the source file.</returns>
-        RootConfiguration GetConfiguration(Uri sourceFileUri);
+        RootConfiguration GetConfiguration(IOUri sourceFileUri);
 
         /// <summary>
         /// Gets the built-in configuration.
@@ -54,7 +55,7 @@ internal ConstantConfigurationManager(RootConfiguration configuration)
                 this.configuration = configuration;
             }
 
-            public RootConfiguration GetConfiguration(Uri sourceFileUri) => configuration;
+            public RootConfiguration GetConfiguration(IOUri sourceFileUri) => configuration;
         }
     }
 }
diff --git a/src/Bicep.Core/Emit/TemplateWriter.cs b/src/Bicep.Core/Emit/TemplateWriter.cs
@@ -1464,9 +1464,9 @@ private void EmitModuleForLocalDeploy(PositionTrackingJsonTextWriter jsonWriter,
                     jsonWriter.AddNestedSourceMap(moduleJsonWriter.TrackingJsonWriter);
                     emitter.EmitProperty("template", moduleTextWriter.ToString());
 
-                    if (moduleBicepFile?.Uri is { } sourceUri)
+                    if (moduleBicepFile?.FileHandle.Uri is { } sourceUri)
                     {
-                        emitter.EmitProperty("sourceUri", sourceUri.AbsoluteUri);
+                        emitter.EmitProperty("sourceUri", sourceUri.ToUriString());
                     }
                 });
 
diff --git a/src/Bicep.Core/Features/FeatureProviderFactory.cs b/src/Bicep.Core/Features/FeatureProviderFactory.cs
@@ -1,6 +1,7 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
 using Bicep.Core.Configuration;
+using Bicep.Core.Extensions;
 using Bicep.IO.Abstraction;
 
 namespace Bicep.Core.Features;
@@ -16,5 +17,5 @@ public FeatureProviderFactory(IConfigurationManager configurationManager, IFileE
         this.fileExplorer = fileExplorer;
     }
 
-    public IFeatureProvider GetFeatureProvider(Uri templateUri) => new FeatureProvider(configurationManager.GetConfiguration(templateUri), this.fileExplorer);
+    public IFeatureProvider GetFeatureProvider(IOUri sourceFileUri) => new FeatureProvider(configurationManager.GetConfiguration(sourceFileUri), this.fileExplorer);
 }
diff --git a/src/Bicep.Core/Features/IFeatureProviderFactory.cs b/src/Bicep.Core/Features/IFeatureProviderFactory.cs
@@ -1,10 +1,12 @@
 // Copyright (c) Microsoft Corporation.
 // Licensed under the MIT License.
+using Bicep.IO.Abstraction;
+
 namespace Bicep.Core.Features;
 
 public interface IFeatureProviderFactory
 {
-    IFeatureProvider GetFeatureProvider(Uri templateUri);
+    IFeatureProvider GetFeatureProvider(IOUri sourceFileUri);
 
     static IFeatureProviderFactory WithStaticFeatureProvider(IFeatureProvider featureProvider)
         => new ConstantFeatureProviderFactory(featureProvider);
@@ -18,6 +20,6 @@ internal ConstantFeatureProviderFactory(IFeatureProvider features)
             this.features = features;
         }
 
-        public IFeatureProvider GetFeatureProvider(Uri templateUri) => features;
+        public IFeatureProvider GetFeatureProvider(IOUri sourceFileUri) => features;
     }
 }
diff --git a/src/Bicep.Core/SourceGraph/BicepSourceFile.cs b/src/Bicep.Core/SourceGraph/BicepSourceFile.cs
@@ -78,9 +78,9 @@ protected BicepSourceFile(BicepSourceFile original)
 
         public ISyntaxHierarchy Hierarchy { get; }
 
-        public RootConfiguration Configuration => this.configurationManager.GetConfiguration(this.Uri);
+        public RootConfiguration Configuration => this.configurationManager.GetConfiguration(this.FileHandle.Uri);
 
-        public IFeatureProvider Features => this.featureProviderFactory.GetFeatureProvider(this.Uri);
+        public IFeatureProvider Features => this.featureProviderFactory.GetFeatureProvider(this.FileHandle.Uri);
 
         public IDiagnosticLookup LexingErrorLookup { get; }
 
diff --git a/src/Bicep.Core/SourceGraph/SourceFileFactory.cs b/src/Bicep.Core/SourceGraph/SourceFileFactory.cs
@@ -88,7 +88,7 @@ public ISourceFile CreateSourceFile(IOUri fileUri, string fileContents, Type? so
 
         public BicepParamFile CreateBicepParamFile(IOUri fileUri, string fileContents)
         {
-            var parser = new ParamsParser(fileContents, this.featureProviderFactory.GetFeatureProvider(fileUri.ToUri()));
+            var parser = new ParamsParser(fileContents, this.featureProviderFactory.GetFeatureProvider(fileUri));
             var lineStarts = TextCoordinateConverter.GetLineStarts(fileContents);
             var fileHandle = this.CreateFileHandle(fileUri);
 
diff --git a/src/Bicep.IO/Abstraction/IFileHandle.cs b/src/Bicep.IO/Abstraction/IFileHandle.cs
@@ -22,11 +22,11 @@ public interface IFileHandle : IIOHandle
 
         string ReadAllText();
 
-        Task<string> ReadAllTextAsync();
+        Task<string> ReadAllTextAsync(CancellationToken cancellationToken = default);
 
         void WriteAllText(string text);
 
-        Task WriteAllTextAsync(string text);
+        Task WriteAllTextAsync(string text, CancellationToken cancellationToken = default);
 
         void Delete();
 
diff --git a/src/Bicep.IO/FileSystem/FileSystemFileHandle.cs b/src/Bicep.IO/FileSystem/FileSystemFileHandle.cs
@@ -45,19 +45,20 @@ public Stream OpenWrite()
 
         public string ReadAllText() => this.FileSystem.File.ReadAllText(this.FilePath);
 
-        public Task<string> ReadAllTextAsync() => this.FileSystem.File.ReadAllTextAsync(this.FilePath);
+        public Task<string> ReadAllTextAsync(CancellationToken cancellationToken = default) =>
+            this.FileSystem.File.ReadAllTextAsync(this.FilePath, cancellationToken);
 
         public void WriteAllText(string text)
         {
             this.GetParent().EnsureExists();
             this.FileSystem.File.WriteAllText(this.FilePath, text);
         }
 
-        public async Task WriteAllTextAsync(string text)
+        public async Task WriteAllTextAsync(string text, CancellationToken cancellationToken = default)
         {
             this.GetParent().EnsureExists();
 
-            await this.FileSystem.File.WriteAllTextAsync(this.FilePath, text);
+            await this.FileSystem.File.WriteAllTextAsync(this.FilePath, text, cancellationToken);
         }
 
         public void Delete() => this.FileSystem.File.Delete(this.FilePath);
diff --git a/src/Bicep.IO/InMemory/DummyFileHandle.cs b/src/Bicep.IO/InMemory/DummyFileHandle.cs
@@ -35,11 +35,11 @@ public DummyFileHandle(IOUri uri)
 
         public string ReadAllText() => throw new UnreachableException();
 
-        public Task<string> ReadAllTextAsync() => throw new UnreachableException();
+        public Task<string> ReadAllTextAsync(CancellationToken cancellationToken = default) => throw new UnreachableException();
 
         public void WriteAllText(string text) => throw new UnreachableException();
 
-        public Task WriteAllTextAsync(string text) => throw new UnreachableException();
+        public Task WriteAllTextAsync(string text, CancellationToken cancellationToken = default) => throw new UnreachableException();
 
         public IFileLock? TryLock() => throw new UnreachableException();
     }
diff --git a/src/Bicep.IO/InMemory/InMemoryFileHandle.cs b/src/Bicep.IO/InMemory/InMemoryFileHandle.cs
diff --git a/src/Bicep.LangServer.IntegrationTests/CompletionTests.cs b/src/Bicep.LangServer.IntegrationTests/CompletionTests.cs
diff --git a/src/Bicep.LangServer.IntegrationTests/HoverTests.cs b/src/Bicep.LangServer.IntegrationTests/HoverTests.cs
diff --git a/src/Bicep.LangServer.UnitTests/BicepExternalSourceRequestHandlerTests.cs b/src/Bicep.LangServer.UnitTests/BicepExternalSourceRequestHandlerTests.cs
diff --git a/src/Bicep.LangServer.UnitTests/Handlers/BicepExternalSourceDocumentLinkHandlerTests.cs b/src/Bicep.LangServer.UnitTests/Handlers/BicepExternalSourceDocumentLinkHandlerTests.cs
diff --git a/src/Bicep.LangServer/Handlers/ImportKubernetesManifestHandler.cs b/src/Bicep.LangServer/Handlers/ImportKubernetesManifestHandler.cs
diff --git a/src/Bicep.Local.Deploy/Extensibility/NestedDeploymentExtension.cs b/src/Bicep.Local.Deploy/Extensibility/NestedDeploymentExtension.cs

Original file line number	Diff line number	Diff line change
`@@ -44,7 +44,7 @@ public async Task<int> RunAsync(TestArguments args)`
`44`	`44`	`{`
`45`	`45`	`var inputUri = this.inputOutputArgumentsResolver.ResolveInputArguments(args);`
`46`	`46`	`ArgumentHelper.ValidateBicepFile(inputUri);`
`47`		`- var features = featureProviderFactory.GetFeatureProvider(inputUri.ToUri());`
	`47`	`+ var features = featureProviderFactory.GetFeatureProvider(inputUri);`
`48`	`48`
`49`	`49`	`if (!features.TestFrameworkEnabled)`
`50`	`50`	`{`
Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,7 @@`
`2`	`2`	`// Licensed under the MIT License.`
`3`	`3`
`4`	`4`	`using Bicep.Core.Configuration;`
	`5`	`+using Bicep.IO.Abstraction;`
`5`	`6`
`6`	`7`	`namespace Bicep.Core.UnitTests.Configuration;`
`7`	`8`
`@@ -16,5 +17,5 @@ public PatchingConfigurationManager(ConfigurationManager configurationManager, F`
`16`	`17`	`this.patchFunc = patchFunc;`
`17`	`18`	`}`
`18`	`19`
`19`		`- public RootConfiguration GetConfiguration(Uri sourceFileUri) => patchFunc(configurationManager.GetConfiguration(sourceFileUri));`
	`20`	`+ public RootConfiguration GetConfiguration(IOUri sourceFileUri) => patchFunc(configurationManager.GetConfiguration(sourceFileUri));`
`20`	`21`	`}`
Original file line number	Diff line number	Diff line change
`@@ -2,6 +2,7 @@`
`2`	`2`	`// Licensed under the MIT License.`
`3`	`3`
`4`	`4`	`using Bicep.Core.Features;`
	`5`	`+using Bicep.IO.Abstraction;`
`5`	`6`
`6`	`7`	`namespace Bicep.Core.UnitTests.Features;`
`7`	`8`
`@@ -16,6 +17,6 @@ public OverriddenFeatureProviderFactory(FeatureProviderFactory factory, FeatureP`
`16`	`17`	`this.overrides = overrides;`
`17`	`18`	`}`
`18`	`19`
`19`		`- public IFeatureProvider GetFeatureProvider(Uri templateUri)`
`20`		`- => new OverriddenFeatureProvider(factory.GetFeatureProvider(templateUri), overrides);`
	`20`	`+ public IFeatureProvider GetFeatureProvider(IOUri sourceFileUri)`
	`21`	`+ => new OverriddenFeatureProvider(factory.GetFeatureProvider(sourceFileUri), overrides);`
`21`	`22`	`}`
Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,7 @@ public async Task GetModuleRestoreStatus_ConfigurationChanges_ReturnsCachedStatu`
`138`	`138`	`{`
`139`	`139`	`// Arrange.`
`140`	`140`	`var configManagerMock = StrictMock.Of<IConfigurationManager>();`
`141`		`- configManagerMock.SetupSequence(m => m.GetConfiguration(It.IsAny<Uri>()))`
	`141`	`+ configManagerMock.SetupSequence(m => m.GetConfiguration(It.IsAny<IOUri>()))`
`142`	`142`	`.Returns(BicepTestConstants.CreateMockConfiguration())`
`143`	`143`	`.Returns(BicepTestConstants.CreateMockConfiguration())`
`144`	`144`	`.Returns(changedConfiguration);`