package configuration

import (
	"reflect"
	"testing"
	"time"

	. "launchpad.net/gocheck"
)

// Hook up gocheck into the gotest runner.
func Test(t *testing.T) {
	TestingT(t)
}

type LoadConfigurationSuite struct{}

var _ = Suite(&LoadConfigurationSuite{})

func (self *LoadConfigurationSuite) TestConfig(c *C) {
	config := LoadConfiguration("config.toml")
	c.Assert(config.Hostname, Equals, "")

	c.Assert(config.LogFile, Equals, "influxdb.log")
	c.Assert(config.LogLevel, Equals, "info")

	c.Assert(config.AdminAssetsDir, Equals, "./admin")
	c.Assert(config.AdminHttpPort, Equals, 8083)

	c.Assert(config.ApiHttpPort, Equals, 0)
	c.Assert(config.ApiHttpSslPort, Equals, 8087)
	c.Assert(config.ApiHttpCertPath, Equals, "../cert.pem")
	c.Assert(config.ApiHttpPortString(), Equals, "")

	c.Assert(config.GraphiteEnabled, Equals, false)
	c.Assert(config.GraphitePort, Equals, 2003)
	c.Assert(config.GraphiteDatabase, Equals, "")

	c.Assert(config.UdpServers, HasLen, 1)
	c.Assert(config.UdpServers[0].Enabled, Equals, true)
	c.Assert(config.UdpServers[0].Port, Equals, 4444)
	c.Assert(config.UdpServers[0].Database, Equals, "test")

	c.Assert(config.RaftDir, Equals, "/tmp/influxdb/development/raft")
	c.Assert(config.RaftServerPort, Equals, 8090)
	c.Assert(config.RaftTimeout.Duration, Equals, time.Second)

	c.Assert(config.DataDir, Equals, "/tmp/influxdb/development/db")

	c.Assert(config.ProtobufPort, Equals, 8099)
	c.Assert(config.ProtobufHeartbeatInterval.Duration, Equals, 200*time.Millisecond)
	c.Assert(config.ProtobufMinBackoff.Duration, Equals, 100*time.Millisecond)
	c.Assert(config.ProtobufMaxBackoff.Duration, Equals, time.Second)
	c.Assert(config.ProtobufTimeout.Duration, Equals, 2*time.Second)
	c.Assert(config.SeedServers, DeepEquals, []string{"hosta:8090", "hostb:8090"})

	c.Assert(config.WalDir, Equals, "/tmp/influxdb/development/wal")
	c.Assert(config.WalFlushAfterRequests, Equals, 0)
	c.Assert(config.WalBookmarkAfterRequests, Equals, 0)
	c.Assert(config.WalIndexAfterRequests, Equals, 1000)
	c.Assert(config.WalRequestsPerLogFile, Equals, 10000)

	c.Assert(config.ClusterMaxResponseBufferSize, Equals, 5)
}

func (self *LoadConfigurationSuite) TestSizeParsing(c *C) {
	var s Size
	c.Assert(s.UnmarshalText([]byte("200m")), IsNil)
	c.Assert(int64(s), Equals, 200*ONE_MEGABYTE)
	if t := reflect.TypeOf(0); t.Size() > 4 {
		c.Assert(s.UnmarshalText([]byte("10g")), IsNil)
		c.Assert(int64(s), Equals, 10*ONE_GIGABYTE)
	}
}