func TestReceive(t *testing.T) {
store := store.NewMemStore()
recv := NewReceiver()
recv.MaxOutbox = 100
recv.MaxInbox = 100
recv.FlushInterval = 1
recv.NumOutlets = 2
recv.NumAcceptors = 2
recv.Store = store
recv.Start()
msg := []byte("81 <190>1 2013-03-27T20:02:24+00:00 hostname token shuttle - - measure=hello val=99\n")
recv.Receive("123", msg)
time.Sleep(time.Second * 2)
var buckets []*bucket.Bucket
for bucket := range store.Scan("not important") {
buckets = append(buckets, bucket)
}
if len(buckets) != 1 {
t.FailNow()
}
testBucket := buckets[0]
if testBucket.Id.Name != "hello" {
t.FailNow()
}
if testBucket.Sum() != 99 {
t.FailNow()
}
}
func TestReceiveMultiMetrics(t *testing.T) {
store, recv := makeReceiver()
recv.FlushInterval = time.Millisecond
recv.Start()
defer recv.Stop()
opts := map[string][]string{"resolution": []string{"1000"}}
msg := []byte("95 <190>1 2013-03-27T00:00:01+00:00 hostname token shuttle - - measure=hello val=10 measure.db=10\n")
recv.Receive("user", "pass", msg, opts)
time.Sleep(recv.FlushInterval * 2)
var buckets []*bucket.Bucket
for b := range store.Scan("not important") {
buckets = append(buckets, b)
}
expectedLength := 2
actualLength := len(buckets)
if actualLength != expectedLength {
t.Errorf("expected=%d actual=%d\n", expectedLength, actualLength)
}
//the log line above has two measurements with values of 10.
actualSum := float64(0)
for i := range buckets {
actualSum += buckets[i].Sum()
}
expectedSum := float64(20)
if actualSum != expectedSum {
t.Errorf("expected=%d actual=%d\n", expectedSum, actualSum)
}
}
func TestReceive(t *testing.T) {
store, recv := makeReceiver()
recv.FlushInterval = time.Millisecond
recv.Start()
defer recv.Stop()
opts := map[string][]string{}
msg := []byte("81 <190>1 2013-03-27T20:02:24+00:00 hostname token shuttle - - measure=hello val=99\n")
recv.Receive("user", "pass", msg, opts)
time.Sleep(recv.FlushInterval * 2)
var buckets []*bucket.Bucket
for b := range store.Scan("not important") {
buckets = append(buckets, b)
}
if len(buckets) != 1 {
t.FailNow()
}
testBucket := buckets[0]
expectedName := "hello"
actualName := testBucket.Id.Name
if actualName != expectedName {
t.Errorf("actual=%s expected=%s\n", actualName, expectedName)
}
if testBucket.Sum() != 99 {
t.FailNow()
}
}
func TestReceiveOpts(t *testing.T) {
store, recv := makeReceiver()
recv.FlushInterval = time.Millisecond
recv.Start()
defer recv.Stop()
opts := map[string][]string{"resolution": []string{"1"}}
msg := []byte("81 <190>1 2013-03-27T00:00:01+00:00 hostname token shuttle - - measure=hello val=99\n")
recv.Receive("user", "pass", msg, opts)
time.Sleep(recv.FlushInterval * 2)
var buckets []*bucket.Bucket
for b := range store.Scan("not important") {
buckets = append(buckets, b)
}
if len(buckets) != 1 {
t.FailNow()
}
testBucket := buckets[0]
expectedSum := float64(99)
actualSum := testBucket.Sum()
if actualSum != expectedSum {
t.Errorf("actual=%d expected=%d\n", actualSum, expectedSum)
}
expectedSecond := 1
actualSecond := testBucket.Id.Time.Second()
if actualSecond != expectedSecond {
t.Errorf("actual=%d expected=%d\n", actualSecond, expectedSecond)
}
}
func TestReceiveRouter(t *testing.T) {
store, recv := makeReceiver()
recv.FlushInterval = time.Millisecond
recv.Start()
defer recv.Stop()
opts := map[string][]string{"resolution": []string{"1"}}
msg := []byte("113 <190>1 2013-03-27T00:00:01+00:00 shuttle heroku router - - host=test.l2met.net service=10ms connect=10ms bytes=45")
recv.Receive("user", "pass", msg, opts)
time.Sleep(recv.FlushInterval * 2)
//There are 3 measurements in our logline.
var buckets []*bucket.Bucket
for b := range store.Scan("not important") {
buckets = append(buckets, b)
}
expectedLength := 3
actualLength := len(buckets)
if actualLength != expectedLength {
t.Errorf("expected=%d actual=%d\n", expectedLength, actualLength)
}
//the log line above has two measurements with values of 10.
actualSum := float64(0)
for i := range buckets {
actualSum += buckets[i].Sum()
}
expectedSum := float64(65)
if actualSum != expectedSum {
t.Errorf("expected=%d actual=%d\n", expectedSum, actualSum)
}
//The name of the metric will be router.service, router.connect.
//The source will include the host.
expectedSrc := "test.l2met.net"
for i := range buckets {
actualSrc := buckets[i].Id.Source
if actualSrc != expectedSrc {
t.Errorf("expected=%s actual=%s\n", expectedSrc, actualSrc)
}
}
}