From f17a5313cd2eb7a2326879ad35ce29071dd38f11 Mon Sep 17 00:00:00 2001
From: Luc Talatinian <lucix@amazon.com>
Date: Fri, 14 Jun 2024 10:55:24 -0400
Subject: [PATCH] wrong setupServer copy

---
 .../eventstream/eventstreamtest/testing.go    | 24 +++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/private/protocol/eventstream/eventstreamtest/testing.go b/private/protocol/eventstream/eventstreamtest/testing.go
index a6850596e1a..0388c3db4e1 100644
--- a/private/protocol/eventstream/eventstreamtest/testing.go
+++ b/private/protocol/eventstream/eventstreamtest/testing.go
@@ -6,6 +6,7 @@ package eventstreamtest
 import (
 	"bytes"
 	"context"
+	"crypto/tls"
 	"fmt"
 	"io"
 	"net/http"
@@ -34,9 +35,28 @@ const (
 )
 
 func setupServer(server *httptest.Server, useH2 bool) *http.Client {
-	server.Start()
+	server.Config.TLSConfig = &tls.Config{
+		InsecureSkipVerify: true,
+	}
 
-	return nil
+	clientTrans := &http.Transport{
+		TLSClientConfig: &tls.Config{
+			InsecureSkipVerify: true,
+		},
+	}
+
+	if useH2 {
+		server.Config.TLSConfig.NextProtos = []string{"h2"}
+		clientTrans.TLSClientConfig.NextProtos = []string{"h2"}
+		clientTrans.ForceAttemptHTTP2 = true
+	}
+	server.TLS = server.Config.TLSConfig
+
+	server.StartTLS()
+
+	return &http.Client{
+		Transport: clientTrans,
+	}
 }
 
 // ServeEventStream provides serving EventStream messages from a HTTP server to