Fix src/trafficcontroller/internal/proxy/firehose_handler_test.go

This test was failing with the new gorilla/websocket.
cloudfoundry · Dec 15, 2023 · 3db0598 · 3db0598
1 parent 41df29f
commit 3db0598
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/trafficcontroller/internal/proxy/firehose_handler_test.go b/src/trafficcontroller/internal/proxy/firehose_handler_test.go
@@ -2,7 +2,6 @@ package proxy_test
 
 import (
 	"errors"
-	"fmt"
 	"net/http"
 	"net/http/httptest"
 	"strings"
@@ -178,11 +177,14 @@ var _ = Describe("FirehoseHandler", func() {
 		)
 		Expect(err).ToNot(HaveOccurred())
 
-		f := func() string {
+		f := func() error {
 			_, _, err := conn.ReadMessage()
-			return fmt.Sprintf("%s", err)
+			return err
 		}
-		Eventually(f).Should(ContainSubstring("websocket: close 1000"))
+		Eventually(f).Should(MatchError(&websocket.CloseError{
+			Code: websocket.CloseNormalClosure,
+			Text: "",
+		}))
 	})
 
 	It("emits the number of connections as a metric", func() {