158 lines
3.2 KiB
Go
158 lines
3.2 KiB
Go
// Copyright (c) 2018 Arista Networks, Inc.
|
|
// Use of this source code is governed by the Apache License 2.0
|
|
// that can be found in the COPYING file.
|
|
|
|
package influxlib
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func testFields(line string, fields map[string]interface{},
|
|
t *testing.T) {
|
|
for k, v := range fields {
|
|
formatString := "%s=%v"
|
|
|
|
if _, ok := v.(string); ok {
|
|
formatString = "%s=%q"
|
|
}
|
|
assert.Contains(t, line, fmt.Sprintf(formatString, k, v),
|
|
fmt.Sprintf(formatString+" expected in %s", k, v, line))
|
|
}
|
|
}
|
|
|
|
func testTags(line string, tags map[string]string,
|
|
t *testing.T) {
|
|
for k, v := range tags {
|
|
assert.Contains(t, line, fmt.Sprintf("%s=%s", k, v),
|
|
fmt.Sprintf("%s=%s expected in %s", k, v, line))
|
|
}
|
|
}
|
|
|
|
func TestBasicWrite(t *testing.T) {
|
|
testConn, _ := NewMockConnection()
|
|
|
|
measurement := "TestData"
|
|
tags := map[string]string{
|
|
"tag1": "Happy",
|
|
"tag2": "Valentines",
|
|
"tag3": "Day",
|
|
}
|
|
fields := map[string]interface{}{
|
|
"Data1": 1234,
|
|
"Data2": "apples",
|
|
"Data3": 5.34,
|
|
}
|
|
|
|
err := testConn.WritePoint(measurement, tags, fields)
|
|
assert.NoError(t, err)
|
|
|
|
line, err := GetTestBuffer(testConn)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, line, measurement,
|
|
fmt.Sprintf("%s does not appear in %s", measurement, line))
|
|
testTags(line, tags, t)
|
|
testFields(line, fields, t)
|
|
}
|
|
|
|
func TestConnectionToHostFailure(t *testing.T) {
|
|
assert := assert.New(t)
|
|
var err error
|
|
|
|
config := &InfluxConfig{
|
|
Port: 8086,
|
|
Protocol: HTTP,
|
|
Database: "test",
|
|
}
|
|
config.Hostname = "this is fake.com"
|
|
_, err = Connect(config)
|
|
assert.Error(err)
|
|
config.Hostname = "\\-Fake.Url.Com"
|
|
_, err = Connect(config)
|
|
assert.Error(err)
|
|
}
|
|
|
|
func TestWriteFailure(t *testing.T) {
|
|
con, _ := NewMockConnection()
|
|
|
|
measurement := "TestData"
|
|
tags := map[string]string{
|
|
"tag1": "hi",
|
|
}
|
|
data := map[string]interface{}{
|
|
"Data1": "cats",
|
|
}
|
|
err := con.WritePoint(measurement, tags, data)
|
|
assert.NoError(t, err)
|
|
|
|
fc, _ := con.Client.(*fakeClient)
|
|
fc.failAll = true
|
|
|
|
err = con.WritePoint(measurement, tags, data)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestQuery(t *testing.T) {
|
|
query := "SELECT * FROM 'system' LIMIT 50;"
|
|
|
|
con, _ := NewMockConnection()
|
|
|
|
_, err := con.Query(query)
|
|
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestAddAndWriteBatchPoints(t *testing.T) {
|
|
testConn, _ := NewMockConnection()
|
|
|
|
measurement := "TestData"
|
|
points := []Point{
|
|
Point{
|
|
Measurement: measurement,
|
|
Tags: map[string]string{
|
|
"tag1": "Happy",
|
|
"tag2": "Valentines",
|
|
"tag3": "Day",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"Data1": 1234,
|
|
"Data2": "apples",
|
|
"Data3": 5.34,
|
|
},
|
|
Timestamp: time.Now(),
|
|
},
|
|
Point{
|
|
Measurement: measurement,
|
|
Tags: map[string]string{
|
|
"tag1": "Happy",
|
|
"tag2": "New",
|
|
"tag3": "Year",
|
|
},
|
|
Fields: map[string]interface{}{
|
|
"Data1": 5678,
|
|
"Data2": "bananas",
|
|
"Data3": 3.14,
|
|
},
|
|
Timestamp: time.Now(),
|
|
},
|
|
}
|
|
|
|
err := testConn.RecordBatchPoints(points)
|
|
assert.NoError(t, err)
|
|
|
|
line, err := GetTestBuffer(testConn)
|
|
assert.NoError(t, err)
|
|
|
|
assert.Contains(t, line, measurement,
|
|
fmt.Sprintf("%s does not appear in %s", measurement, line))
|
|
for _, p := range points {
|
|
testTags(line, p.Tags, t)
|
|
testFields(line, p.Fields, t)
|
|
}
|
|
}
|