Function TestBasicDERPController_RewriteDERPMap

tailnet/controllers_test.go:2236–2280

(t *testing.T)

}

func TestBasicDERPController_RewriteDERPMap(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true}).Leveled(slog.LevelDebug)

	testDERPMap := &tailcfg.DERPMap{
		Regions: map[int]*tailcfg.DERPRegion{
			1: {
				RegionID: 1,
			},
		},
	}

	// Ensure the fake rewriter works as expected.
	rewriter := &fakeDERPMapRewriter{
		ctx:   ctx,
		calls: make(chan rewriteDERPMapCall, 16),
	}
	rewriter.RewriteDERPMap(testDERPMap)
	rewriteCall := testutil.TryReceive(ctx, t, rewriter.calls)
	require.Same(t, testDERPMap, rewriteCall.derpMap)

	derpClient := &fakeDERPClient{
		ch:  make(chan *tailcfg.DERPMap),
		err: nil,
	}
	defer derpClient.Close()

	derpSetter := &fakeDERPMapSetter{
		ctx:   ctx,
		calls: make(chan *setDERPMapCall, 16),
	}

	derpCtrl := tailnet.NewBasicDERPController(logger, rewriter, derpSetter)
	derpCtrl.New(derpClient)

	// Simulate receiving a new DERP map from the server, which should be passed
	// to the rewriter and setter.
	testDERPMap = testDERPMap.Clone() // make a new pointer
	derpClient.ch <- testDERPMap
	rewriteCall = testutil.TryReceive(ctx, t, rewriter.calls)
	require.Same(t, testDERPMap, rewriteCall.derpMap)
	setterCall := testutil.TryReceive(ctx, t, derpSetter.calls)
	require.Same(t, testDERPMap, setterCall.derpMap)
}

type fakeWorkspaceUpdatesController struct {
	ctx   context.Context