mirror of https://github.com/gorilla/websocket.git
64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
// Copyright 2014 The Gorilla WebSocket Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package websocket
|
|
|
|
import (
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
var parseURLTests = []struct {
|
|
s string
|
|
u *url.URL
|
|
}{
|
|
{"ws://example.com/", &url.URL{Scheme: "ws", Host: "example.com", Opaque: "/"}},
|
|
{"ws://example.com", &url.URL{Scheme: "ws", Host: "example.com", Opaque: "/"}},
|
|
{"ws://example.com:7777/", &url.URL{Scheme: "ws", Host: "example.com:7777", Opaque: "/"}},
|
|
{"wss://example.com/", &url.URL{Scheme: "wss", Host: "example.com", Opaque: "/"}},
|
|
{"wss://example.com/a/b", &url.URL{Scheme: "wss", Host: "example.com", Opaque: "/a/b"}},
|
|
{"ss://example.com/a/b", nil},
|
|
}
|
|
|
|
func TestParseURL(t *testing.T) {
|
|
for _, tt := range parseURLTests {
|
|
u, err := parseURL(tt.s)
|
|
if tt.u != nil && err != nil {
|
|
t.Errorf("parseURL(%q) returned error %v", tt.s, err)
|
|
continue
|
|
}
|
|
if tt.u == nil && err == nil {
|
|
t.Errorf("parseURL(%q) did not return error", tt.s)
|
|
continue
|
|
}
|
|
if !reflect.DeepEqual(u, tt.u) {
|
|
t.Errorf("parseURL(%q) returned %v, want %v", tt.s, u, tt.u)
|
|
continue
|
|
}
|
|
}
|
|
}
|
|
|
|
var hostPortNoPortTests = []struct {
|
|
u *url.URL
|
|
hostPort, hostNoPort string
|
|
}{
|
|
{&url.URL{Scheme: "ws", Host: "example.com"}, "example.com:80", "example.com"},
|
|
{&url.URL{Scheme: "wss", Host: "example.com"}, "example.com:443", "example.com"},
|
|
{&url.URL{Scheme: "ws", Host: "example.com:7777"}, "example.com:7777", "example.com"},
|
|
{&url.URL{Scheme: "wss", Host: "example.com:7777"}, "example.com:7777", "example.com"},
|
|
}
|
|
|
|
func TestHostPortNoPort(t *testing.T) {
|
|
for _, tt := range hostPortNoPortTests {
|
|
hostPort, hostNoPort := hostPortNoPort(tt.u)
|
|
if hostPort != tt.hostPort {
|
|
t.Errorf("hostPortNoPort(%v) returned hostPort %q, want %q", tt.u, hostPort, tt.hostPort)
|
|
}
|
|
if hostNoPort != tt.hostNoPort {
|
|
t.Errorf("hostPortNoPort(%v) returned hostNoPort %q, want %q", tt.u, hostNoPort, tt.hostNoPort)
|
|
}
|
|
}
|
|
}
|