steve
39aff83837
CI / Test (server-http) (pull_request) Successful in 5s
CI / Test (rest) (pull_request) Successful in 7s
CI / Test (store) (pull_request) Successful in 5s
CI / Build (windows/amd64) (pull_request) Successful in 7s
CI / Lint (pull_request) Failing after 10s
CI / Build (linux/arm64) (pull_request) Successful in 8s
CI / Build (linux/amd64) (pull_request) Successful in 24s
e2e / Playwright vs docker-compose (pull_request) Successful in 1m36s
47 lines
1.1 KiB
Go
47 lines
1.1 KiB
Go
package wsclient
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/coder/websocket"
|
|
)
|
|
|
|
func TestConnectOnceCleanDisconnectDoesNotPanic(t *testing.T) {
|
|
serverErr := make(chan error, 1)
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
conn, err := websocket.Accept(w, r, nil)
|
|
if err != nil {
|
|
serverErr <- err
|
|
return
|
|
}
|
|
defer conn.CloseNow() //nolint:errcheck
|
|
|
|
// Wait for the agent hello so Dial and the first client write have both
|
|
// completed before ending the connection normally.
|
|
if _, _, err := conn.Read(r.Context()); err != nil {
|
|
serverErr <- err
|
|
return
|
|
}
|
|
serverErr <- conn.Close(websocket.StatusNormalClosure, "test complete")
|
|
}))
|
|
defer srv.Close()
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
err := connectOnce(ctx, Config{
|
|
ServerURL: srv.URL,
|
|
AgentToken: "test-token",
|
|
HeartbeatPeriod: time.Hour,
|
|
}, nil)
|
|
if err == nil {
|
|
t.Fatal("connectOnce returned nil after server disconnected")
|
|
}
|
|
if err := <-serverErr; err != nil {
|
|
t.Fatalf("server websocket: %v", err)
|
|
}
|
|
}
|