From 6e07d5568518419f2fe1f657653a051e17f3c207 Mon Sep 17 00:00:00 2001
From: Ankit Jain <radical@gmail.com>
Date: Fri, 12 Jul 2024 21:26:08 -0400
Subject: [PATCH] Add new target to generate a Packages.Versions.props with all
 the relevant properties, and remove old custom tasks

---
 tests/Directory.Build.props                  |  1 +
 tests/Shared/Aspire.Workload.Testing.props   |  1 +
 tests/Shared/Aspire.Workload.Testing.targets | 90 +-------------------
 3 files changed, 4 insertions(+), 88 deletions(-)

diff --git a/tests/Directory.Build.props b/tests/Directory.Build.props
index 60906cd5a0..5b0337a39f 100644
--- a/tests/Directory.Build.props
+++ b/tests/Directory.Build.props
@@ -8,6 +8,7 @@
     <TestArchiveTestsDirForWorkloadTests Condition="'$(TestArchiveTestsDirForWorkloadTests)' == ''">$([MSBuild]::NormalizeDirectory($(ArtifactsDir), 'helix', 'workload-tests'))</TestArchiveTestsDirForWorkloadTests>
     <TestArchiveTestsDirForEndToEndTests Condition="'$(TestArchiveTestsDirForEndToEndTests)' == ''">$([MSBuild]::NormalizeDirectory($(ArtifactsDir), 'helix', 'e2e-tests'))</TestArchiveTestsDirForEndToEndTests>
     <PlaywrightDependenciesDirectory>$(ArtifactsBinDir)playwright-deps</PlaywrightDependenciesDirectory>
+    <GeneratedPackagesVersionsPropsPath>$(IntermediateOutputPath)Directory.Packages.Versions.props</GeneratedPackagesVersionsPropsPath>
 
     <!-- This is useful for local runs -->
     <VSTestResultsDirectory>$(ArtifactsTestResultsDir)</VSTestResultsDirectory>
diff --git a/tests/Shared/Aspire.Workload.Testing.props b/tests/Shared/Aspire.Workload.Testing.props
index a8fdf2fa63..bf3a3b5274 100644
--- a/tests/Shared/Aspire.Workload.Testing.props
+++ b/tests/Shared/Aspire.Workload.Testing.props
@@ -6,6 +6,7 @@
     <InstallWorkloadUsingArtifactsAfterThisTarget />
     <TemplateNuGetConfigPath>$(RepoRoot)NuGet.config</TemplateNuGetConfigPath>
     <PatchedNuGetConfigPath>$(ArtifactsObjDir)nuget.workload.config</PatchedNuGetConfigPath>
+    <GeneratedPackagesVersionsPropsPath>$(IntermediateOutputPath)Directory.Packages.Versions.props</GeneratedPackagesVersionsPropsPath>
 
     <_GlobalJsonContent>$([System.IO.File]::ReadAllText('$(RepoRoot)global.json'))</_GlobalJsonContent>
     <_DotNetCliVersionFromGlobalJson>$([System.Text.RegularExpressions.Regex]::Match($(_GlobalJsonContent), '(%3F&lt;="dotnet": ").*(%3F=")'))</_DotNetCliVersionFromGlobalJson>
diff --git a/tests/Shared/Aspire.Workload.Testing.targets b/tests/Shared/Aspire.Workload.Testing.targets
index a6a8a64743..f214e35079 100644
--- a/tests/Shared/Aspire.Workload.Testing.targets
+++ b/tests/Shared/Aspire.Workload.Testing.targets
@@ -76,97 +76,11 @@
     </ItemGroup>
   </Target>
 
-  <UsingTask TaskName="PrepareNuGetConfigForWorkloadTesting"
-      TaskFactory="RoslynCodeTaskFactory"
-      AssemblyFile="$(MSBuildToolsPath)\Microsoft.Build.Tasks.Core.dll">
-    <ParameterGroup>
-      <TemplateNuGetConfigPath ParameterType="System.String" Required="true" />
-      <OutputFile ParameterType="System.String" Required="true" />
-    </ParameterGroup>
-    <Task>
-      <Using Namespace="System.IO" />
-      <Using Namespace="System.Xml.Linq" />
-      <Using Namespace="System.Xml.XPath" />
-      <Code Type="Fragment" Language="cs">
-        <![CDATA[
-          if (!File.Exists(TemplateNuGetConfigPath))
-          {
-            Log.LogError($"Could not find nuget config template at '{TemplateNuGetConfigPath}' .");
-            return false;
-          }
-
-          // This task adds a nuget packageSource pointing to the locally built nugets.
-          // And then adds a packageSourceMapping for that source to `*Aspire*` so the
-          // locally built nugets can be restored from there
-
-          // add packageSource
-          XDocument doc = XDocument.Load(TemplateNuGetConfigPath);
-          string xpath = "/configuration/packageSources";
-          XElement packageSources = doc.XPathSelectElement(xpath);
-          if (packageSources is null)
-          {
-            Log.LogError($"Could not find {xpath} in {TemplateNuGetConfigPath}");
-            return false;
-          }
-
-          // %BUILT_NUGETS_PATH% is set when building the testproject
-          packageSources.LastNode.AddAfterSelf(
-            new XElement("add",
-              new XAttribute("key", "nuget-local"),
-              new XAttribute("value", "%BUILT_NUGETS_PATH%")));
-
-          // add packageSourceMapping
-          string mappingXpath = "/configuration/packageSourceMapping";
-          XElement packageSourceMapping = doc.XPathSelectElement(mappingXpath);
-          if (packageSourceMapping is null)
-          {
-            // if mapping has been removed completely then the task needs an update!
-            throw new InvalidOperationException($"Expected to find {mappingXpath} in {TemplateNuGetConfigPath}");
-          }
-
-          packageSourceMapping.FirstNode.AddBeforeSelf(
-            new XElement("packageSource",
-              new XAttribute("key", "nuget-local"),
-              new XElement("package", new XAttribute("pattern", "*Aspire*"))));
-
-          doc.Save(OutputFile);
-          Log.LogMessage(MessageImportance.Low, $"Generated patched nuget.config at {OutputFile}");
-        ]]>
-      </Code>
-    </Task>
-  </UsingTask>
-
-  <!-- For test projects -->
-
-  <Target Name="_PatchNuGetConfig" AfterTargets="GetCopyToOutputDirectoryItems" Inputs="$(TemplateNuGetConfigPath)" Outputs="$(PatchedNuGetConfigPath)">
-    <PrepareNuGetConfigForWorkloadTesting TemplateNuGetConfigPath="$(RepoRoot)NuGet.config" OutputFile="$(PatchedNuGetConfigPath)" />
-  </Target>
-
-  <Target Name="_AddPackageVersionsForUseWithHelix" BeforeTargets="ZipTestArchive">
-    <PropertyGroup>
-      <_TestAssetsOutputDir>$(OutputPath)\testassets\</_TestAssetsOutputDir>
-
-      <_PackagePropsContents>
-<![CDATA[
-      <Project>
-        <PropertyGroup>
-          <PackageVersion>$(PackageVersion)</PackageVersion>
-          <MicrosoftExtensionsHostingAbstractionsPackageVersion>$(MicrosoftExtensionsHostingAbstractionsPackageVersion)</MicrosoftExtensionsHostingAbstractionsPackageVersion>
-          <MicrosoftExtensionsHostingPackageVersion>$(MicrosoftExtensionsHostingPackageVersion)</MicrosoftExtensionsHostingPackageVersion>
-        </PropertyGroup>
-      </Project>
-]]>
-      </_PackagePropsContents>
-    </PropertyGroup>
-
-    <WriteLinesToFile Lines="$(_PackagePropsContents)" File="$(_TestAssetsOutputDir)\testproject\Directory.Build.Helix.props" Overwrite="true" />
-  </Target>
-
   <Target Name="CheckForWorkloadInstalled" BeforeTargets="VSTest" Condition="'$(TestsRunningOutsideOfRepo)' == 'true'">
     <Error Condition="'$(SdkWithWorkloadForTestingPath)' == '' or !Exists($(SdkWithWorkloadForTestingPath))"
       Text="Could not find %24(SdkWithWorkloadForTestingPath)=$(SdkWithWorkloadForTestingPath) . Prepare it by following:
-      1. .\build.cmd -pack
-      2. dotnet build tests/Aspire.EndToEnd.Tests/Aspire.EndToEnd.csproj /t:InstallWorkloadUsingArtifacts /p:Configuration=$(Configuration)" />
+      1. .\build.cmd -pack (or ./build.sh on unix)
+      2. dotnet build tests/workloads.proj /p:Configuration=$(Configuration)" />
   </Target>
 
   <Target Name="_ValidateExpectedSetOfPackagesExist" Condition="'$(SkipPackageCheckForWorkloadTesting)' != 'true'">