package collector

import (
	"reflect"
	"testing"
)

// TestNetworkAddressSetsSeparatesIPv4AndIPv6 验证展平 IP 列表按地址族拆分。
func TestNetworkAddressSetsSeparatesIPv4AndIPv6(t *testing.T) {
	ipv4s, ipv6s := networkAddressSets(map[string]struct{}{
		"fe80::1/64":      {},
		"192.168.1.10/24": {},
		"10.0.0.2/8":      {},
		"not-an-ip":       {},
	})

	if want := []string{"10.0.0.2/8", "192.168.1.10/24"}; !reflect.DeepEqual(ipv4s, want) {
		t.Fatalf("ipv4s = %#v, want %#v", ipv4s, want)
	}
	if want := []string{"fe80::1/64"}; !reflect.DeepEqual(ipv6s, want) {
		t.Fatalf("ipv6s = %#v, want %#v", ipv6s, want)
	}
}