From 2796fcdd874bf17e757f2bbe8a2d38ea9d5c11db Mon Sep 17 00:00:00 2001
From: Ben Wells <benwells@github.com>
Date: Fri, 21 Jul 2023 15:55:35 -0400
Subject: [PATCH] Update FileCommandTestBase.cs to add additional heredoc edge
 cases

---
 src/Test/L0/Worker/FileCommandTestBase.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/Test/L0/Worker/FileCommandTestBase.cs b/src/Test/L0/Worker/FileCommandTestBase.cs
index f3bcd8e73..9e14101e7 100644
--- a/src/Test/L0/Worker/FileCommandTestBase.cs
+++ b/src/Test/L0/Worker/FileCommandTestBase.cs
@@ -238,6 +238,10 @@ namespace GitHub.Runner.Common.Tests.Worker
                     "MY_KEY_4<<EOF",
                     "EOF EOF",
                     "EOF",
+                    "MY_KEY_5=abc << def",
+                    "MY_KEY_6=    <<EOF",
+                    "white space test",
+                    "EOF"
                 };
                 TestUtil.WriteContent(stateFile, content);
                 _fileCmdExtension.ProcessCommand(_executionContext.Object, stateFile, null);
@@ -247,6 +251,8 @@ namespace GitHub.Runner.Common.Tests.Worker
                 Assert.Equal($"hello=two", _store["MY_KEY_2"]);
                 Assert.Equal($" EOF", _store["MY_KEY_3"]);
                 Assert.Equal($"EOF EOF", _store["MY_KEY_4"]);
+                Assert.Equal($"abc << def", _store["MY_KEY_5"]);
+                Assert.Equal($"white space test", _store["MY_KEY_6="]);
             }
         }