end-to-end-test: Set a timeout when invoking Handler

armsnyder · Nov 7, 2020 · fe6d04f · fe6d04f
1 parent d6db51e
commit fe6d04f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/server/server_suite_test.go b/pkg/server/server_suite_test.go
@@ -318,13 +318,18 @@ func (c *clientConnection) sendType(typ string, message interface{}) {
 	if err != nil {
 		panic(err)
 	}
-	_, err = Handler(c.ctx, events.APIGatewayWebsocketProxyRequest{
+
+	ctx, cancel := context.WithTimeout(c.ctx, time.Second)
+	defer cancel()
+
+	_, err = Handler(ctx, events.APIGatewayWebsocketProxyRequest{
 		Body: string(b),
 		RequestContext: events.APIGatewayWebsocketProxyRequestContext{
 			ConnectionID: c.connID,
 			EventType:    typ,
 		},
 	})
+
 	Expect(err).NotTo(HaveOccurred())
 }