From e19e311174cda37fe74d7315e630a28a9034d14f Mon Sep 17 00:00:00 2001
From: Ankit Jain <radical@gmail.com>
Date: Fri, 12 Jul 2024 22:00:30 -0400
Subject: [PATCH] restore workloads project

---
 tests/Directory.Build.targets | 2 ++
 tests/workloads.proj          | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/tests/Directory.Build.targets b/tests/Directory.Build.targets
index 5354d5a667..df4061d376 100644
--- a/tests/Directory.Build.targets
+++ b/tests/Directory.Build.targets
@@ -57,4 +57,6 @@
                       Lines="@(TestClassName->Distinct())"
                       Overwrite="true" />
   </Target>
+
+  <Import Project="Shared\Aspire.Workload.Testing.targets" Condition="'$(IsWorkloadTestProject)' == 'true'" />
 </Project>
diff --git a/tests/workloads.proj b/tests/workloads.proj
index efd6b7f0a3..50b9c3acde 100644
--- a/tests/workloads.proj
+++ b/tests/workloads.proj
@@ -6,6 +6,9 @@
       <Configuration Condition="'$(Configuration)' == ''">Debug</Configuration>
     </PropertyGroup>
 
+    <MSBuild Projects="Aspire.EndToEnd.Tests\Aspire.EndToEnd.Tests.csproj"
+             Targets="Restore"
+             Properties="Configuration=$(Configuration);TestUsingWorkloads=true" />
     <MSBuild Projects="Aspire.EndToEnd.Tests\Aspire.EndToEnd.Tests.csproj"
              Targets="InstallWorkloadUsingArtifacts"
              Properties="Configuration=$(Configuration);TestUsingWorkloads=true" />