Function TestUpdateDiscoForNodeCallback

control/controlclient/map_test.go:792–915

(t *testing.T)

}

func TestUpdateDiscoForNodeCallback(t *testing.T) {
	t.Run("key_wired_through_to_updater", func(t *testing.T) {
		nu := &rememberLastNetmapUpdater{
			done: make(chan any, 1),
		}
		ms := newTestMapSession(t, nu)

		oldKey := key.NewDisco()

		// Insert existing node
		node := tailcfg.Node{
			ID:       1,
			Key:      key.NewNode().Public(),
			DiscoKey: oldKey.Public(),
			Online:   new(false),
			LastSeen: new(time.Unix(1, 0)),
		}

		if nm := ms.netmapForResponse(&tailcfg.MapResponse{
			Peers: []*tailcfg.Node{&node},
		}); len(nm.Peers) != 1 {
			t.Fatalf("node not inserted")
		}

		newKey := key.NewDisco()
		ms.updateDiscoForNode(node.ID, node.Key, newKey.Public(), time.Now(), false)
		<-nu.done

		if nu.lastTSMPKey != node.Key || nu.lastTSMPDisco != newKey.Public() {
			t.Fatalf("expected [%s]=%s, got [%s]=%s", node.Key, newKey.Public(),
				nu.lastTSMPKey, nu.lastTSMPDisco)
		}
	})
	// Even though key stays in list of update, the updater only triggers on TSMP.
	t.Run("key_not_wired_through_to_updater", func(t *testing.T) {
		nu := &rememberLastNetmapUpdater{
			done: make(chan any, 1),
		}
		ms := newTestMapSession(t, nu)

		oldKey := key.NewDisco()

		// Insert existing node
		node := tailcfg.Node{
			ID:       1,
			Key:      key.NewNode().Public(),
			DiscoKey: oldKey.Public(),
			Online:   new(false),
			LastSeen: new(time.Unix(1, 0)),
		}

		if nm := ms.netmapForResponse(&tailcfg.MapResponse{
			Peers: []*tailcfg.Node{&node},
		}); len(nm.Peers) != 1 {
			t.Fatalf("node not inserted")
		}

		newKey := key.NewDisco().Public()