nodes_test.go

package dnsconfig

import (
	. "launchpad.net/gocheck"
	"net"
)

type NodesSuite struct {
	Nodes Nodes
}

var _ = Suite(&NodesSuite{})

func (s *NodesSuite) SetUpSuite(c *C) {
	s.Nodes = NewNodes()
}

func (s *NodesSuite) TestNodes(c *C) {
	s.Nodes.Clear()
	s.Nodes.Set("foo", Node{Ip: net.ParseIP("10.0.0.1"), Active: true})
	c.Assert(s.Nodes.Count(), Equals, 1)
	node := s.Nodes.Get("foo")
	c.Assert(node, NotNil)
	c.Assert(node.Name, Equals, "foo")
}

func (s *NodesSuite) TestLoad(c *C) {
	s.Nodes.Clear()
	err := s.Nodes.LoadFile("testdata/nodes-small.json")
	c.Assert(err, IsNil)

	node := s.Nodes.Get("edge01.lax")
	c.Assert(node, NotNil)
	c.Assert(node.Ip.String(), Equals, "108.161.187.3")
	c.Assert(node.Active, Equals, true)

	node = s.Nodes.Get("edge01.sea")
	c.Assert(node, NotNil)
	c.Assert(node.Active, Equals, false)

}