Function TestController_DoesNotRedialAfterCancel

tailnet/controllers_test.go:1117–1154

(t *testing.T)

}

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

	fClient := newFakeWorkspaceUpdateClient(testCtx, t)
	dialer := &scriptedDialer{
		attempts: make(chan int, 10),
		dialFn: func(_ context.Context, _ int) (tailnet.ControlProtocolClients, error) {
			return tailnet.ControlProtocolClients{
				WorkspaceUpdates: fClient,
				Closer:           fakeCloser{},
			}, nil
		},
	}
	fCtrl := newFakeUpdatesController(testCtx, t)

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

	require.Equal(t, 1, testutil.TryReceive(testCtx, t, dialer.attempts))
	call := testutil.TryReceive(testCtx, t, fCtrl.calls)
	require.Equal(t, fClient, call.client)
	testutil.RequireSend[tailnet.CloserWaiter](testCtx, t, call.resp, newFakeCloserWaiter())

	cancel()
	closeCall := testutil.TryReceive(testCtx, t, fClient.close)
	testutil.RequireSend(testCtx, t, closeCall, nil)
	_ = testutil.TryReceive(testCtx, t, uut.Closed())

	select {
	case attempt := <-dialer.attempts:
		t.Fatalf("unexpected redial attempt after cancel: %d", attempt)
	default:
	}
}

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