package agent

import "testing"

func TestPeerSubnet(t *testing.T) {
	cases := []struct {
		peer string
		want string
	}{
		{"2602:817:3000:a25::1", "2602:817:3000:a25::/64"},
		{"2602:817:3000:a25::104", "2602:817:3000:a25::/64"},
		{"172.25.25.1", "172.25.25.0/24"},
		{"172.25.25.104", "172.25.25.0/24"},
		{"", ""},
		{"not-an-ip", ""},
	}
	for _, tc := range cases {
		t.Run(tc.peer, func(t *testing.T) {
			got := peerSubnet(tc.peer)
			if got != tc.want {
				t.Fatalf("peerSubnet(%q) = %q, want %q", tc.peer, got, tc.want)
			}
		})
	}
}