Function TestNodeUpdater_fillPeerDiagnostics

tailnet/node_internal_test.go:569–615

(t *testing.T)

}

func TestNodeUpdater_fillPeerDiagnostics(t *testing.T) {
	t.Parallel()
	ctx := testutil.Context(t, testutil.WaitShort)
	logger := testutil.Logger(t)
	id := tailcfg.NodeID(1)
	nodeKey := key.NewNode().Public()
	discoKey := key.NewDisco().Public()
	uut := newNodeUpdater(
		logger,
		func(n *Node) {},
		id, nodeKey, discoKey,
	)
	defer uut.close()

	// at start of day, filling diagnostics will not have derp and sentNode is false
	d := PeerDiagnostics{}
	uut.fillPeerDiagnostics(&d)
	require.Equal(t, 0, d.PreferredDERP)
	require.False(t, d.SentNode)

	dl := map[string]float64{"1": 0.025}
	uut.setNetInfo(&tailcfg.NetInfo{
		PreferredDERP: 1,
		DERPLatency:   dl,
	})

	// after node callback, we should get the derp and SentNode is true.
	// Use eventually since, there is a race between the callback completing
	// and the test checking
	require.Eventually(t, func() bool {
		d := PeerDiagnostics{}
		uut.fillPeerDiagnostics(&d)
		// preferred DERP should be set right away, even if the callback is not
		// complete.
		if !assert.Equal(t, 1, d.PreferredDERP) {
			return false
		}
		return d.SentNode
	}, testutil.WaitShort, testutil.IntervalFast)

	done := make(chan struct{})
	go func() {
		defer close(done)
		uut.close()
	}()
	_ = testutil.TryReceive(ctx, t, done)
}

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