Function TestExtractDERPLatency

tailnet/derpmap_test.go:167–239

(t *testing.T)

}

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

	derpMap := &tailcfg.DERPMap{
		Regions: map[int]*tailcfg.DERPRegion{
			1: {
				RegionID:   1,
				RegionName: "Region One",
				Nodes: []*tailcfg.DERPNode{
					{Name: "node1", RegionID: 1},
				},
			},
			2: {
				RegionID:   2,
				RegionName: "Region Two",
				Nodes: []*tailcfg.DERPNode{
					{Name: "node2", RegionID: 2},
				},
			},
		},
	}

	t.Run("Basic", func(t *testing.T) {
		t.Parallel()
		node := &tailnet.Node{
			DERPLatency: map[string]float64{
				"1-node1": 0.05,
				"2-node2": 0.1,
			},
		}
		latencyMs := tailnet.ExtractDERPLatency(node, derpMap)
		require.EqualValues(t, 50, latencyMs["Region One"].Milliseconds())
		require.EqualValues(t, 100, latencyMs["Region Two"].Milliseconds())
		require.Len(t, latencyMs, 2)
	})

	t.Run("UnknownRegion", func(t *testing.T) {
		t.Parallel()
		node := &tailnet.Node{
			DERPLatency: map[string]float64{
				"999-node999": 0.2,
			},
		}
		latencyMs := tailnet.ExtractDERPLatency(node, derpMap)
		require.EqualValues(t, 200, latencyMs["Unnamed 999"].Milliseconds())
		require.Len(t, latencyMs, 1)
	})

	t.Run("InvalidRegionFormat", func(t *testing.T) {
		t.Parallel()
		node := &tailnet.Node{
			DERPLatency: map[string]float64{
				"invalid":  0.3,
				"1-node1":  0.05,
				"abc-node": 0.15,
			},
		}
		latencyMs := tailnet.ExtractDERPLatency(node, derpMap)