Function TestConfigMaps_addRemoveHosts

tailnet/configmaps_internal_test.go:1159–1251

(t *testing.T)

}

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

	ctx := testutil.Context(t, testutil.WaitShort)
	logger := testutil.Logger(t)
	fEng := newFakeEngineConfigurable()
	nodePrivateKey := key.NewNode()
	nodeID := tailcfg.NodeID(5)
	discoKey := key.NewDisco()
	suffix := dnsname.FQDN("test.")
	uut := newConfigMaps(logger, fEng, nodeID, nodePrivateKey, discoKey.Public(), suffix)
	defer uut.close()

	addr1 := CoderServicePrefix.AddrFromUUID(uuid.New())
	addr2 := CoderServicePrefix.AddrFromUUID(uuid.New())
	addr3 := CoderServicePrefix.AddrFromUUID(uuid.New())
	addr4 := CoderServicePrefix.AddrFromUUID(uuid.New())

	// WHEN: we set two hosts
	uut.setHosts(map[dnsname.FQDN][]netip.Addr{
		"agent.myws.me.coder.": {
			addr1,
		},
		"dev.main.me.coder.": {
			addr2,
			addr3,
		},
	})

	// THEN: the engine is reconfigured with those same hosts
	_ = testutil.TryReceive(ctx, t, fEng.setNetworkMap)
	req := testutil.TryReceive(ctx, t, fEng.reconfig)
	require.Equal(t, req.dnsCfg, &dns.Config{
		Routes: map[dnsname.FQDN][]*dnstype.Resolver{
			suffix: nil,
		},
		// Note that host names and Routes are independent --- so we faithfully reproduce the hosts, even though
		// they don't match the route.
		Hosts: map[dnsname.FQDN][]netip.Addr{
			"agent.myws.me.coder.": {
				addr1,
			},
			"dev.main.me.coder.": {
				addr2,
				addr3,
			},
		},
		OnlyIPv6: true,
	})

	// WHEN: We replace the hosts with a new set
	uut.setHosts(map[dnsname.FQDN][]netip.Addr{
		"newagent.myws.me.coder.": {
			addr4,
		},
		"newagent2.main.me.coder.": {
			addr1,
		},