Function TestController_RetriesWrappedDeadlineExceeded

tailnet/controllers_test.go:1078–1115

(t *testing.T)

}

func TestController_RetriesWrappedDeadlineExceeded(t *testing.T) {
	t.Parallel()
	testCtx := testutil.Context(t, testutil.WaitShort)
	ctx, cancel := context.WithCancel(testCtx)
	defer cancel()

	logger := testutil.Logger(t)
	dialer := &scriptedDialer{
		attempts: make(chan int, 10),
		dialFn: func(ctx context.Context, attempt int) (tailnet.ControlProtocolClients, error) {
			if attempt == 1 {
				return tailnet.ControlProtocolClients{}, &net.OpError{
					Op:  "dial",
					Net: "tcp",
					Err: context.DeadlineExceeded,
				}
			}

			<-ctx.Done()
			return tailnet.ControlProtocolClients{}, ctx.Err()
		},
	}

	uut := tailnet.NewController(logger.Named("ctrl"), dialer)
	uut.Run(ctx)

	require.Equal(t, 1, testutil.TryReceive(testCtx, t, dialer.attempts))
	require.Equal(t, 2, testutil.TryReceive(testCtx, t, dialer.attempts))

	select {
	case <-uut.Closed():
		t.Fatal("controller exited after wrapped deadline exceeded")
	default:
	}

	cancel()
	_ = testutil.TryReceive(testCtx, t, uut.Closed())
}

func TestController_DoesNotRedialAfterCancel(t *testing.T) {
	t.Parallel()