From 21f661fba5f9e654b62b6c1c1a00b47a7729ebfb Mon Sep 17 00:00:00 2001 From: David Irvine Date: Mon, 2 Jan 2017 02:17:37 -0500 Subject: [PATCH] Fix up tests for develop branch --- context_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/context_test.go b/context_test.go index fdc896f2..fe22c492 100644 --- a/context_test.go +++ b/context_test.go @@ -817,7 +817,7 @@ func TestContextGolangContext(t *testing.T) { func TestWebsocketsRequired(t *testing.T) { // Example request from spec: https://tools.ietf.org/html/rfc6455#section-1.2 - c, _, _ := CreateTestContext() + c, _ := CreateTestContext(httptest.NewRecorder()) c.Request, _ = http.NewRequest("GET", "/chat", nil) c.Request.Header.Set("Host", "server.example.com") c.Request.Header.Set("Upgrade", "websocket") @@ -830,7 +830,7 @@ func TestWebsocketsRequired(t *testing.T) { assert.True(t, c.IsWebsocket()) // Normal request, no websocket required. - c, _, _ = CreateTestContext() + c, _ = CreateTestContext(httptest.NewRecorder()) c.Request, _ = http.NewRequest("GET", "/chat", nil) c.Request.Header.Set("Host", "server.example.com")