package net_test

import (
	"net"
	"testing"

	"v2ray.com/core/common"
	. "v2ray.com/core/common/net"
	. "v2ray.com/ext/assert"
)

func parseCIDR(str string) *net.IPNet {
	_, ipNet, err := net.ParseCIDR(str)
	common.Must(err)
	return ipNet
}

func TestIPNet(t *testing.T) {
	assert := With(t)

	ipNet := NewIPNetTable()
	ipNet.Add(parseCIDR(("0.0.0.0/8")))
	ipNet.Add(parseCIDR(("10.0.0.0/8")))
	ipNet.Add(parseCIDR(("100.64.0.0/10")))
	ipNet.Add(parseCIDR(("127.0.0.0/8")))
	ipNet.Add(parseCIDR(("169.254.0.0/16")))
	ipNet.Add(parseCIDR(("172.16.0.0/12")))
	ipNet.Add(parseCIDR(("192.0.0.0/24")))
	ipNet.Add(parseCIDR(("192.0.2.0/24")))
	ipNet.Add(parseCIDR(("192.168.0.0/16")))
	ipNet.Add(parseCIDR(("198.18.0.0/15")))
	ipNet.Add(parseCIDR(("198.51.100.0/24")))
	ipNet.Add(parseCIDR(("203.0.113.0/24")))
	ipNet.Add(parseCIDR(("8.8.8.8/32")))
	ipNet.AddIP(net.ParseIP("91.108.4.0"), 16)
	assert(ipNet.Contains(ParseIP("192.168.1.1")), IsTrue)
	assert(ipNet.Contains(ParseIP("192.0.0.0")), IsTrue)
	assert(ipNet.Contains(ParseIP("192.0.1.0")), IsFalse)
	assert(ipNet.Contains(ParseIP("0.1.0.0")), IsTrue)
	assert(ipNet.Contains(ParseIP("1.0.0.1")), IsFalse)
	assert(ipNet.Contains(ParseIP("8.8.8.7")), IsFalse)
	assert(ipNet.Contains(ParseIP("8.8.8.8")), IsTrue)
	assert(ipNet.Contains(ParseIP("2001:cdba::3257:9652")), IsFalse)
	assert(ipNet.Contains(ParseIP("91.108.255.254")), IsTrue)
}