From d78a0b791203c608bbf3c81618ef511336df87f7 Mon Sep 17 00:00:00 2001
From: Forgind <Forgind@users.noreply.github.com>
Date: Thu, 10 Aug 2023 09:19:55 -0700
Subject: [PATCH] Check for baseline manifest as well

---
 test/EndToEnd/ValidateInsertedManifests.cs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/test/EndToEnd/ValidateInsertedManifests.cs b/test/EndToEnd/ValidateInsertedManifests.cs
index cdede126c..ffd5dd329 100644
--- a/test/EndToEnd/ValidateInsertedManifests.cs
+++ b/test/EndToEnd/ValidateInsertedManifests.cs
@@ -31,7 +31,12 @@ namespace EndToEnd.Tests
 
                     string manifestFile = manifestDir.GetFile("WorkloadManifest.json").FullName;
 
-                    File.Exists(manifestFile).Should().BeTrue();
+                    var baselineFile = manifestDir.GetFile("Baseline.WorkloadSet.json").FullName;
+                    if (!(new FileInfo(baselineFile).Exists))
+                    {
+                        new FileInfo(manifestFile).Exists.Should().BeTrue();
+                    }
+
                     using var fileStream = new FileStream(manifestFile, FileMode.Open, FileAccess.Read);
                     Action readManifest = () => WorkloadManifestReader.ReadWorkloadManifest(manifestId, fileStream, manifestFile);
                     readManifest.ShouldNotThrow("manifestId:" + manifestId + " manifestFile:" + manifestFile + "is invalid");