Function TestDrainConnDuringReconnect

test/drain_test.go:458–496

(t *testing.T)

}

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

	done := make(chan bool, 1)
	closedCb := func(nc *nats.Conn) {
		done <- true
	}

	nc, err := nats.Connect(nats.DefaultURL,
		nats.ClosedHandler(closedCb),
		nats.DrainTimeout(20*time.Millisecond))
	if err != nil {
		t.Fatalf("Failed to create default connection: %v", err)
	}
	defer nc.Close()

	// Shutdown the server.
	s.Shutdown()

	waitFor(t, time.Second, 10*time.Millisecond, func() error {
		if nc.IsReconnecting() {
			return nil
		}
		return errors.New("Not reconnecting yet")
	})

	// This should work correctly.
	if err := nc.Drain(); err != nats.ErrConnectionReconnecting {
		t.Fatalf("Unexpected error on drain: %v", err)
	}

	// Closed should still fire.
	select {
	case <-done:
	case <-time.After(2 * time.Second):
		t.Fatalf("Timeout waiting for closed state for connection")
	}
}

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