From 6cfa9fb55b6399f9535d5b33d7459f17e1dbfa36 Mon Sep 17 00:00:00 2001
From: Peter Ombwa <peter.ombwa@microsoft.com>
Date: Fri, 19 Nov 2021 16:17:52 -0800
Subject: [PATCH] Log warning headers into PS Warning stream.

---
 tools/Custom/Module.cs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/tools/Custom/Module.cs b/tools/Custom/Module.cs
index fef409a88b4..4c0422f24b7 100644
--- a/tools/Custom/Module.cs
+++ b/tools/Custom/Module.cs
@@ -105,6 +105,12 @@ private async Task OnResponseCreated(string id, CancellationToken cancellationTo
                 var response = eventData?.ResponseMessage as HttpResponseMessage;
                 if (response != null)
                 {
+                    if (response.Headers.Warning != null)
+                    {
+                        string warningHeader = response.Headers.Warning.ToString();
+                        await signal(Events.Warning, cancellationToken,
+                            () => EventFactory.CreateWarningEvent(warningHeader));
+                    }
                     // Log request after response since all our request header are set via middleware pipeline.
                     var request = response?.RequestMessage;
                     if (request != null)