GeoDNS/targeting_test.go

package main

import (
	"net"

	"github.com/abh/geodns/geo"
	. "gopkg.in/check.v1"
)

type TargetingSuite struct {
}

var _ = Suite(&TargetingSuite{})

func (s *TargetingSuite) SetUpSuite(c *C) {
	Config.GeoIP.Directory = "db"
}

func (s *TargetingSuite) TestTargetString(c *C) {
	var tgt TargetOptions
	tgt = TargetGlobal + TargetCountry + TargetContinent

	str := tgt.String()
	c.Check(str, Equals, "@ continent country")
}

func (s *TargetingSuite) TestTargetParse(c *C) {

	tgt, err := parseTargets("@ foo country")
	str := tgt.String()
	c.Check(str, Equals, "@ country")
	c.Check(err.Error(), Equals, "Unknown targeting option 'foo'")

	tgt, err = parseTargets("@ continent country asn")
	c.Assert(err, IsNil)
	str = tgt.String()
	c.Check(str, Equals, "@ continent country asn")
}

func (s *TargetingSuite) TestGetTargets(c *C) {
	ip := net.ParseIP("207.171.1.1")

	if geoIP == nil {
		geoIP = geo.New()
	}

	err := geoIP.SetupCity()
	if err != nil {
		c.Log("City GeoIP database requred for these tests")
		return
	}
	geoIP.SetupCountry()
	geoIP.SetupASN()

	tgt, _ := parseTargets("@ continent country")
	targets, _ := tgt.GetTargets(ip)
	c.Check(targets, DeepEquals, []string{"us", "north-america", "@"})

	tgt, _ = parseTargets("@ continent country region ")
	targets, _ = tgt.GetTargets(ip)
	c.Check(targets, DeepEquals, []string{"us-ca", "us", "north-america", "@"})

	tgt, _ = parseTargets("@ continent regiongroup country region ")
	targets, _ = tgt.GetTargets(ip)
	c.Check(targets, DeepEquals, []string{"us-ca", "us-west", "us", "north-america", "@"})

	tgt, _ = parseTargets("@ continent regiongroup country region asn ip")
	targets, _ = tgt.GetTargets(ip)
	c.Check(targets, DeepEquals, []string{"[207.171.1.1]", "[207.171.1.0]", "as7012", "us-ca", "us-west", "us", "north-america", "@"})

	ip = net.ParseIP("2607:f238:2:0::ff:4")
	tgt, _ = parseTargets("ip")
	targets, _ = tgt.GetTargets(ip)
	c.Check(targets, DeepEquals, []string{"[2607:f238:2::ff:4]", "[2607:f238:2::]"})

}