Function TestWSHeaders_StaticAppliedOnHandshake

ws_test.go:736–767

(t *testing.T)

}

func TestWSHeaders_StaticAppliedOnHandshake(t *testing.T) {
	addr, got, closeLn := startHeaderCatcher(t)
	defer closeLn()

	static := make(http.Header)
	static.Set("Authorization", "Bearer Random Token")
	static.Add("X-Multi", "v1")
	static.Add("X-Multi", "v2")

	// Intentionally connect to our fake server; it won't complete the upgrade.
	opts := GetDefaultOptions()
	opts.WebSocketConnectionHeaders = static
	opts.Url = "ws://" + addr
	_, err := opts.Connect()
	if err == nil {
		t.Fatalf("expected connect to fail because server does not reply")
	}

	var headers []string
	select {
	case headers = <-got:
	case <-time.After(2 * time.Second):
		t.Fatal("did not capture headers in time")
	}

	if !hasHeaderValue(headers, "Authorization", "Bearer Random Token") {
		t.Fatalf("Authorization header missing: %v", headers)
	}
	if !hasHeaderValue(headers, "X-Multi", "v1") || !hasHeaderValue(headers, "X-Multi", "v2") {
		t.Fatalf("X-Multi headers missing/combined incorrectly: %v", headers)
	}
}

func TestWSHeaders_HandlerAppliedOnHandshake(t *testing.T) {
	addr, got, closeLn := startHeaderCatcher(t)