Added NuGet source mapping for unit tests. (#35)

* Ensures that unit tests are using ONLY the locally built packages.
    - This wasn't a real issue until the first pre-release was published to public NuGet feeds

Co-authored-by: smaillet <25911635+smaillet@users.noreply.github.com>

Author: smaillet

src/Ubiquity.Versioning.Build.Tasks.UT/ProjectCreatorLibraryExtensions.cs

@@ -7,6 +7,8 @@
 using System;
 using System.Collections.Generic;
 using System.Diagnostics.CodeAnalysis;
+using System.Linq;
+using System.Xml.Linq;
 
 using Microsoft.Build.Evaluation;
 using Microsoft.Build.Utilities.ProjectCreation;
@@ -15,6 +17,17 @@ namespace Ubiquity.Versioning.Build.Tasks.UT
 {
     internal static class ProjectCreatorLibraryExtensions
     {
+        public static PackageRepository SourceMapping( this PackageRepository pkgRepo, string pkgSourceKey, string pattern )
+        {
+            var nugetConfig = XDocument.Load(pkgRepo.NuGetConfigPath);
+            XElement configuration = GetOrCreateConfigurationElement( nugetConfig );
+            XElement sourceMapping = GetOrCreateSourceMappingElement( configuration );
+            XElement pkgSrcElement = GetOrCreatePackageSource( sourceMapping, pkgSourceKey );
+            _ = GetOrCreatePackageElement(pkgSrcElement, pattern);
+            nugetConfig.Save(pkgRepo.NuGetConfigPath);
+            return pkgRepo;
+        }
+
         [SuppressMessage( "Style", "IDE0060:Remove unused parameter", Justification = "Syntactical sugar" )]
         [SuppressMessage( "Style", "IDE0046:Convert to conditional expression", Justification = "Result is Less than 'simplified'" )]
         public static ProjectCreator VersioningProject(
@@ -30,9 +43,9 @@ public static ProjectCreator VersioningProject(
             string? treatAsLocalProperty = null,
             ProjectCollection? projectCollection = null,
             IDictionary<string, string>? globalProperties = null,
-            NewProjectFileOptions? projectFileOptions = NewProjectFileOptions.None)
+            NewProjectFileOptions? projectFileOptions = NewProjectFileOptions.None )
         {
-            ArgumentException.ThrowIfNullOrWhiteSpace(targetFramework);
+            ArgumentException.ThrowIfNullOrWhiteSpace( targetFramework );
 
             return templates.SdkCsproj(
                                 path,
@@ -46,13 +59,71 @@ public static ProjectCreator VersioningProject(
                                 projectCollection,
                                 projectFileOptions,
                                 globalProperties
-                                ).ItemPackageReference(PackageUnderTestId, version: packageVersion, privateAssets: "All")
-                                 .Property("Nullable", "disable")
-                                 .Property("ManagePackageVersionsCentrally", "false")
-                                 .Property("ImplicitUsings","disable")
+                                ).ItemPackageReference( PackageUnderTestId, version: packageVersion, privateAssets: "All" )
+                                 .Property( "Nullable", "disable" )
+                                 .Property( "ManagePackageVersionsCentrally", "false" )
+                                 .Property( "ImplicitUsings", "disable" )
                                  ;
         }
 
+        private static XElement GetOrCreateSourceMappingElement( XElement configuration )
+        {
+            XElement? sourceMapping = configuration.Element("packageSourceMapping");
+            if(sourceMapping is null)
+            {
+                sourceMapping = new XElement( "packageSourceMapping" );
+                configuration.Add( sourceMapping );
+            }
+
+            return sourceMapping;
+        }
+
+        private static XElement GetOrCreateConfigurationElement( XDocument nugetConfig )
+        {
+            XElement? configuration = nugetConfig.Element("configuration");
+            if(configuration is null)
+            {
+                configuration = new XElement( "configuration" );
+                nugetConfig.Add( configuration );
+            }
+
+            return configuration;
+        }
+
+        private static XElement GetOrCreatePackageSource( XElement sourceMapping, string pkgSource )
+        {
+            XElement? packageSource = ( from e in sourceMapping.Elements("packageSource")
+                                        from a in e.Attributes()
+                                        where a.Name == "key" && a.Value == pkgSource
+                                        select e
+                                      ).FirstOrDefault();
+
+            if(packageSource is null)
+            {
+                packageSource = new XElement( "packageSource", new XAttribute( "key", pkgSource ) );
+                sourceMapping.Add( packageSource );
+            }
+
+            return packageSource;
+        }
+
+        private static XElement GetOrCreatePackageElement( XElement pkgSrc, string pattern )
+        {
+            XElement? package = ( from e in pkgSrc.Elements("package")
+                                  from a in e.Attributes()
+                                  where a.Name == "pattern" && a.Value == pattern
+                                  select e
+                                ).FirstOrDefault();
+
+            if(package is null)
+            {
+                package = new XElement( "package", new XAttribute( "pattern", pattern) );
+                pkgSrc.Add( package );
+            }
+
+            return package;
+        }
+
         private const string PackageUnderTestId = @"Ubiquity.NET.Versioning.Build.Tasks";
     }
 }

src/Ubiquity.Versioning.Build.Tasks.UT/TestModuleFixtures.cs

@@ -63,7 +63,8 @@ public static void AssemblyInitialize( TestContext ctx )
                 ctx.TestRunDirectory,                            // '.nuget/packages' repo folder goes here
                 new Uri(Path.Combine(BuildOutputPath, "NuGet")), // Local feed (Contains location of the build of the package under test)
                 new Uri("https://api.nuget.org/v3/index.json") // standard NuGet Feed
-            );
+            ).SourceMapping("Local1", "Ubiquity.NET.Versioning*") // force all fetches of built package to locally built source only
+             .SourceMapping("api.nuget.org", "*"); // Everything else goes to NuGet public feed.
         }
 
         [AssemblyCleanup]