juanfont.headscale/hscontrol/types/routes_test.go

package types

import (
	"fmt"
	"net/netip"
	"testing"

	"github.com/google/go-cmp/cmp"
	"github.com/juanfont/headscale/hscontrol/util"
)

func TestPrefixMap(t *testing.T) {
	ipp := func(s string) netip.Prefix { return netip.MustParsePrefix(s) }

	tests := []struct {
		rs   Routes
		want map[netip.Prefix][]Route
	}{
		{
			rs: Routes{
				Route{
					Prefix: ipp("10.0.0.0/24"),
				},
			},
			want: map[netip.Prefix][]Route{
				ipp("10.0.0.0/24"): Routes{
					Route{
						Prefix: ipp("10.0.0.0/24"),
					},
				},
			},
		},
		{
			rs: Routes{
				Route{
					Prefix: ipp("10.0.0.0/24"),
				},
				Route{
					Prefix: ipp("10.0.1.0/24"),
				},
			},
			want: map[netip.Prefix][]Route{
				ipp("10.0.0.0/24"): Routes{
					Route{
						Prefix: ipp("10.0.0.0/24"),
					},
				},
				ipp("10.0.1.0/24"): Routes{
					Route{
						Prefix: ipp("10.0.1.0/24"),
					},
				},
			},
		},
		{
			rs: Routes{
				Route{
					Prefix:  ipp("10.0.0.0/24"),
					Enabled: true,
				},
				Route{
					Prefix:  ipp("10.0.0.0/24"),
					Enabled: false,
				},
			},
			want: map[netip.Prefix][]Route{
				ipp("10.0.0.0/24"): Routes{
					Route{
						Prefix:  ipp("10.0.0.0/24"),
						Enabled: true,
					},
					Route{
						Prefix:  ipp("10.0.0.0/24"),
						Enabled: false,
					},
				},
			},
		},
	}

	for idx, tt := range tests {
		t.Run(fmt.Sprintf("test-%d", idx), func(t *testing.T) {
			got := tt.rs.PrefixMap()
			if diff := cmp.Diff(tt.want, got, util.Comparers...); diff != "" {
				t.Errorf("PrefixMap() unexpected result (-want +got):\n%s", diff)
			}
		})
	}
}