diff options
Diffstat (limited to 'vendor/github.com/gorilla/websocket/server_test.go')
-rw-r--r-- | vendor/github.com/gorilla/websocket/server_test.go | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/vendor/github.com/gorilla/websocket/server_test.go b/vendor/github.com/gorilla/websocket/server_test.go index 0a28141d6..c43dbb267 100644 --- a/vendor/github.com/gorilla/websocket/server_test.go +++ b/vendor/github.com/gorilla/websocket/server_test.go @@ -49,3 +49,21 @@ func TestIsWebSocketUpgrade(t *testing.T) { } } } + +var checkSameOriginTests = []struct { + ok bool + r *http.Request +}{ + {false, &http.Request{Host: "example.org", Header: map[string][]string{"Origin": []string{"https://other.org"}}}}, + {true, &http.Request{Host: "example.org", Header: map[string][]string{"Origin": []string{"https://example.org"}}}}, + {true, &http.Request{Host: "Example.org", Header: map[string][]string{"Origin": []string{"https://example.org"}}}}, +} + +func TestCheckSameOrigin(t *testing.T) { + for _, tt := range checkSameOriginTests { + ok := checkSameOrigin(tt.r) + if tt.ok != ok { + t.Errorf("checkSameOrigin(%+v) returned %v, want %v", tt.r, ok, tt.ok) + } + } +} |