From 5dc7ec058f21d6fdaf4d1b3dd9ed8cfd69138980 Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@victoriametrics.com>
Date: Sat, 27 Jul 2024 13:44:49 +0200
Subject: [PATCH] app/vmauth: verify how backend response headers are
 propagated to vmauth client

---
 app/vmauth/main_test.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/vmauth/main_test.go b/app/vmauth/main_test.go
index beb91dc543..6414ef8034 100644
--- a/app/vmauth/main_test.go
+++ b/app/vmauth/main_test.go
@@ -71,6 +71,10 @@ unauthorized_user:
   url_prefix: {BACKEND}/foo?bar=baz`
 	requestURL := "http://some-host.com/abc/def?some_arg=some_value"
 	backendHandler := func(w http.ResponseWriter, r *http.Request) {
+		h := w.Header()
+		h.Set("Connection", "close")
+		h.Set("Foo", "bar")
+
 		var bb bytes.Buffer
 		if err := r.Header.Write(&bb); err != nil {
 			panic(fmt.Errorf("unexpected error when marshaling headers: %w", err))
@@ -79,6 +83,7 @@ unauthorized_user:
 	}
 	responseExpected := `
 statusCode=200
+Foo: bar
 requested_url={BACKEND}/foo/abc/def?bar=baz&some_arg=some_value
 Pass-Header: abc
 User-Agent: vmauth