New test with a websocket app

Author: Dariquest

src/code.cloudfoundry.org/gorouter/integration/route_services_test.go

@@ -21,7 +21,8 @@ var _ = Describe("Route services", func() {
 	var testState *testState
 
 	const (
-		appHostname = "app-with-route-service.some.domain"
+		appHostname   = "app-with-route-service.some.domain"
+		wsAppHostname = "ws-app-with-route-service.some.domain"
 	)
 
 	var (
@@ -84,6 +85,60 @@ var _ = Describe("Route services", func() {
 		return fmt.Sprintf("https://%s:%s", testState.trustedExternalServiceHostname, port)
 	}
 
+	Context("Happy Path with a web socket app with a route service", func() {
+		Context("When an app is registered with a simple route service", func() {
+			var (
+				wsApp *httptest.Server
+			)
+			BeforeEach(func() {
+				testState.StartGorouterOrFail()
+				routeService.Start()
+				wsApp = httptest.NewServer(http.HandlerFunc(
+					func(w http.ResponseWriter, r *http.Request) {
+						w.Header().Add("X-App-Instance", "ws-app")
+						w.Header().Set("Upgrade", "websocket")
+						w.Header().Set("Connection", "upgrade")
+						w.WriteHeader(200)
+						_, err := w.Write([]byte("I'm the websocket app"))
+						Expect(err).ToNot(HaveOccurred())
+					}))
+
+				testState.registerWithInternalRouteService(
+					wsApp,
+					routeService,
+					wsAppHostname,
+					testState.cfg.SSLPort,
+				)
+			})
+
+			It("succeeds", func() {
+				req := testState.newGetRequest(
+					fmt.Sprintf("https://%s", wsAppHostname),
+				)
+
+				res, err := testState.client.Do(req)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(res.StatusCode).To(Equal(http.StatusOK))
+				body, err := io.ReadAll(res.Body)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(body).To(Equal([]byte("I'm the route service")))
+			})
+
+			It("properly URL-encodes and decodes", func() {
+				req := testState.newGetRequest(
+					fmt.Sprintf("https://%s?%s", wsAppHostname, "param=a%0Ab"),
+				)
+
+				res, err := testState.client.Do(req)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(res.StatusCode).To(Equal(http.StatusOK))
+				body, err := io.ReadAll(res.Body)
+				Expect(err).ToNot(HaveOccurred())
+				Expect(body).To(Equal([]byte("I'm the route service")))
+			})
+		})
+	})
+
 	Context("Happy Path", func() {
 		Context("When an app is registered with a simple route service", func() {
 			BeforeEach(func() {