Function TestDrainClosedHandlerRace

test/drain_test.go:498–516

(t *testing.T)

}

func TestDrainClosedHandlerRace(t *testing.T) {
	s := RunDefaultServer()
	defer s.Shutdown()

	nc, err := nats.Connect(s.ClientURL(), nats.ClosedHandler(func(_ *nats.Conn) {}))

	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	defer nc.Close()

	if err := nc.Drain(); err != nil {
		t.Fatalf("Unexpected error on drain: %v", err)
	}
	go func() {
		nc.SetClosedHandler(nil)
	}()
	time.Sleep(500 * time.Millisecond)
}