Function TestSetNodeCapMap

tstest/integration/capmap_test.go:98–145

TestSetNodeCapMap tests that SetNodeCapMap updates are propagated to peers.

(t *testing.T)

// TestSetNodeCapMap tests that SetNodeCapMap updates are propagated to peers.
func TestSetNodeCapMap(t *testing.T) {
	tstest.Parallel(t)
	env := NewTestEnv(t)

	n1 := NewTestNode(t, env)
	d1 := n1.StartDaemon()
	n1.AwaitListening()
	n1.MustUp()
	n1.AwaitRunning()

	nodes := env.Control.AllNodes()
	if len(nodes) != 1 {
		t.Fatalf("expected 1 node, got %d nodes", len(nodes))
	}
	node1 := nodes[0]

	// Set initial CapMap.
	caps := make(tailcfg.NodeCapMap)
	caps["test:state"] = []tailcfg.RawMessage{`"initial"`}
	env.Control.SetNodeCapMap(node1.Key, caps)

	// Start second node and verify it sees the first node's CapMap.
	n2 := NewTestNode(t, env)
	d2 := n2.StartDaemon()
	n2.AwaitListening()
	n2.MustUp()
	n2.AwaitRunning()

	if err := tstest.WaitFor(5*time.Second, func() error {
		st := n2.MustStatus()
		if len(st.Peer) == 0 {
			return errors.New("no peers")
		}
		p := st.Peer[st.Peers()[0]]
		if p.CapMap == nil || p.CapMap["test:state"] == nil {
			return errors.New("peer CapMap not set")
		}
		if string(p.CapMap["test:state"][0]) != `"initial"` {
			return errors.New("wrong CapMap value")
		}
		return nil
	}); err != nil {
		t.Fatal(err)
	}

	d1.MustCleanShutdown(t)
	d2.MustCleanShutdown(t)
}