Function TestRegisterRateLimited

control/controlclient/controlclient_test.go:412–519

TestRegisterRateLimited verifies that the client correctly handles 429 responses to registration requests by parsing the Retry-After header and returning a rateLimitError.

(t *testing.T)

// responses to registration requests by parsing the Retry-After header
// and returning a rateLimitError.
func TestRegisterRateLimited(t *testing.T) {
	bakedroots.ResetForTest(t, tlstest.TestRootCA())

	bus := eventbustest.NewBus(t)

	controlLn, err := tls.Listen("tcp", "127.0.0.1:0", tlstest.ControlPlane.ServerTLSConfig())
	if err != nil {
		t.Fatal(err)
	}
	defer controlLn.Close()

	var registerAttempts atomic.Int64
	tc := &testcontrol.Server{
		Logf: tstest.WhileTestRunningLogger(t),
		MaybeRateLimitRegister: func() (bool, string, string) {
			if registerAttempts.Add(1) == 1 {
				return true, "30", "try again later"
			}
			return false, "", ""
		},
	}
	controlSrv := &http.Server{
		Handler:  tc,
		ErrorLog: logger.StdLogger(t.Logf),
	}
	go controlSrv.Serve(controlLn)

	const fakeControlIP = "1.2.3.4"

	dialer := &tsdial.Dialer{}
	dialer.SetNetMon(netmon.NewStatic())
	dialer.SetBus(bus)
	dialer.SetSystemDialerForTest(func(ctx context.Context, network, addr string) (net.Conn, error) {
		host, _, err := net.SplitHostPort(addr)
		if err != nil {
			return nil, fmt.Errorf("SplitHostPort(%q): %v", addr, err)
		}
		var d net.Dialer
		if host == fakeControlIP {
			return d.DialContext(ctx, network, controlLn.Addr().String())
		}
		return nil, fmt.Errorf("unexpected dial to %q", addr)
	})

	opts := Options{
		Persist: persist.Persist{},
		GetMachinePrivateKey: func() (key.MachinePrivate, error) {
			return key.NewMachine(), nil
		},
		ServerURL: "https://controlplane.tstest",
		Clock:     tstime.StdClock{},
		Hostinfo: &tailcfg.Hostinfo{
			BackendLogID: "test-backend-log-id",
		},
		DiscoPublicKey: key.NewDisco().Public(),
		Logf:           t.Logf,
		HealthTracker:  health.NewTracker(bus),
		PopBrowserURL: func(url string) {