github.com/insionng/yougam@v0.0.0-20170714101924-2bc18d833463/libraries/gorilla/websocket/server_test.go (about)

     1  // Copyright 2013 The Gorilla WebSocket Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package websocket
     6  
     7  import (
     8  	"net/http"
     9  	"reflect"
    10  	"testing"
    11  )
    12  
    13  var subprotocolTests = []struct {
    14  	h         string
    15  	protocols []string
    16  }{
    17  	{"", nil},
    18  	{"foo", []string{"foo"}},
    19  	{"foo,bar", []string{"foo", "bar"}},
    20  	{"foo, bar", []string{"foo", "bar"}},
    21  	{" foo, bar", []string{"foo", "bar"}},
    22  	{" foo, bar ", []string{"foo", "bar"}},
    23  }
    24  
    25  func TestSubprotocols(t *testing.T) {
    26  	for _, st := range subprotocolTests {
    27  		r := http.Request{Header: http.Header{"Sec-Websocket-Protocol": {st.h}}}
    28  		protocols := Subprotocols(&r)
    29  		if !reflect.DeepEqual(st.protocols, protocols) {
    30  			t.Errorf("SubProtocols(%q) returned %#v, want %#v", st.h, protocols, st.protocols)
    31  		}
    32  	}
    33  }
    34  
    35  var isWebSocketUpgradeTests = []struct {
    36  	ok bool
    37  	h  http.Header
    38  }{
    39  	{false, http.Header{"Upgrade": {"websocket"}}},
    40  	{false, http.Header{"Connection": {"upgrade"}}},
    41  	{true, http.Header{"Connection": {"upgRade"}, "Upgrade": {"WebSocket"}}},
    42  }
    43  
    44  func TestIsWebSocketUpgrade(t *testing.T) {
    45  	for _, tt := range isWebSocketUpgradeTests {
    46  		ok := IsWebSocketUpgrade(&http.Request{Header: tt.h})
    47  		if tt.ok != ok {
    48  			t.Errorf("IsWebSocketUpgrade(%v) returned %v, want %v", tt.h, ok, tt.ok)
    49  		}
    50  	}
    51  }