feat: add conversions from tailnet to proto (#10441)

Adds conversions from existing tailnet types to protobuf
2025-07-23 21:32:07 +00:00 · 2023-11-01 10:54:00 +04:00
parent f4026edd71
commit 6882e8e524
3 changed files with 293 additions and 2 deletions
--- a/tailnet/convert_test.go
+++ b/tailnet/convert_test.go
@ -0,0 +1,153 @@
+package tailnet_test
+
+import (
+	"net/netip"
+	"testing"
+	"time"
+
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/require"
+	"tailscale.com/tailcfg"
+	"tailscale.com/types/key"
+
+	"github.com/coder/coder/v2/coderd/database/dbtime"
+	"github.com/coder/coder/v2/tailnet"
+	"github.com/coder/coder/v2/tailnet/proto"
+)
+
+func TestNode(t *testing.T) {
+	t.Parallel()
+	testCases := []struct {
+		name string
+		node tailnet.Node
+	}{
+		{
+			name: "Zero",
+		},
+		{
+			name: "AllFields",
+			node: tailnet.Node{
+				ID:            33,
+				AsOf:          time.Now(),
+				Key:           key.NewNode().Public(),
+				DiscoKey:      key.NewDisco().Public(),
+				PreferredDERP: 12,
+				DERPLatency: map[string]float64{
+					"1":  0.2,
+					"12": 0.3,
+				},
+				DERPForcedWebsocket: map[int]string{
+					1: "forced",
+				},
+				Addresses: []netip.Prefix{
+					netip.MustParsePrefix("10.0.0.0/8"),
+					netip.MustParsePrefix("ff80::aa:1/128"),
+				},
+				AllowedIPs: []netip.Prefix{
+					netip.MustParsePrefix("10.0.0.0/8"),
+					netip.MustParsePrefix("ff80::aa:1/128"),
+				},
+				Endpoints: []string{
+					"192.168.0.1:3305",
+					"[ff80::aa:1]:2049",
+				},
+			},
+		},
+		{
+			name: "dbtime",
+			node: tailnet.Node{
+				AsOf: dbtime.Now(),
+			},
+		},
+	}
+
+	for _, tc := range testCases {
+		tc := tc
+		t.Run(tc.name, func(t *testing.T) {
+			t.Parallel()
+			p, err := tailnet.NodeToProto(&tc.node)
+			require.NoError(t, err)
+
+			inv, err := tailnet.ProtoToNode(p)
+			require.NoError(t, err)
+			require.Equal(t, tc.node.ID, inv.ID)
+			require.True(t, tc.node.AsOf.Equal(inv.AsOf))
+			require.Equal(t, tc.node.Key, inv.Key)
+			require.Equal(t, tc.node.DiscoKey, inv.DiscoKey)
+			require.Equal(t, tc.node.PreferredDERP, inv.PreferredDERP)
+			require.Equal(t, tc.node.DERPLatency, inv.DERPLatency)
+			require.Equal(t, len(tc.node.DERPForcedWebsocket), len(inv.DERPForcedWebsocket))
+			for k, v := range inv.DERPForcedWebsocket {
+				nv, ok := tc.node.DERPForcedWebsocket[k]
+				require.True(t, ok)
+				require.Equal(t, nv, v)
+			}
+			require.ElementsMatch(t, tc.node.Addresses, inv.Addresses)
+			require.ElementsMatch(t, tc.node.AllowedIPs, inv.AllowedIPs)
+			require.ElementsMatch(t, tc.node.Endpoints, inv.Endpoints)
+		})
+	}
+}
+
+func TestUUIDToByteSlice(t *testing.T) {
+	t.Parallel()
+	u := uuid.New()
+	b := tailnet.UUIDToByteSlice(u)
+	u2, err := uuid.FromBytes(b)
+	require.NoError(t, err)
+	require.Equal(t, u, u2)
+
+	b = tailnet.UUIDToByteSlice(uuid.Nil)
+	u2, err = uuid.FromBytes(b)
+	require.NoError(t, err)
+	require.Equal(t, uuid.Nil, u2)
+}
+
+func TestOnlyNodeUpdates(t *testing.T) {
+	t.Parallel()
+	node := &tailnet.Node{ID: tailcfg.NodeID(1)}
+	p, err := tailnet.NodeToProto(node)
+	require.NoError(t, err)
+	resp := &proto.CoordinateResponse{
+		PeerUpdates: []*proto.CoordinateResponse_PeerUpdate{
+			{
+				Uuid: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1},
+				Kind: proto.CoordinateResponse_PeerUpdate_NODE,
+				Node: p,
+			},
+			{
+				Uuid:   []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2},
+				Kind:   proto.CoordinateResponse_PeerUpdate_DISCONNECTED,
+				Reason: "disconnected",
+			},
+			{
+				Uuid:   []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 3},
+				Kind:   proto.CoordinateResponse_PeerUpdate_LOST,
+				Reason: "disconnected",
+			},
+			{
+				Uuid: []byte{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4},
+			},
+		},
+	}
+	nodes, err := tailnet.OnlyNodeUpdates(resp)
+	require.NoError(t, err)
+	require.Len(t, nodes, 1)
+	require.Equal(t, tailcfg.NodeID(1), nodes[0].ID)
+}
+
+func TestSingleNodeUpdate(t *testing.T) {
+	t.Parallel()
+	node := &tailnet.Node{ID: tailcfg.NodeID(1)}
+	u := uuid.New()
+	resp, err := tailnet.SingleNodeUpdate(u, node, "unit test")
+	require.NoError(t, err)
+	require.Len(t, resp.PeerUpdates, 1)
+	up := resp.PeerUpdates[0]
+	require.Equal(t, proto.CoordinateResponse_PeerUpdate_NODE, up.Kind)
+	u2, err := uuid.FromBytes(up.Uuid)
+	require.NoError(t, err)
+	require.Equal(t, u, u2)
+	require.Equal(t, "unit test", up.Reason)
+	require.EqualValues(t, 1, up.Node.Id)
+}