131 lines
2.6 KiB
Go
131 lines
2.6 KiB
Go
package testutil
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"time"
|
|
)
|
|
|
|
type Point struct {
|
|
Measurement string
|
|
Tags map[string]string
|
|
Values map[string]interface{}
|
|
Time time.Time
|
|
}
|
|
|
|
type Accumulator struct {
|
|
Points []*Point
|
|
}
|
|
|
|
func (a *Accumulator) Add(measurement string, value interface{}, tags map[string]string) {
|
|
if tags == nil {
|
|
tags = map[string]string{}
|
|
}
|
|
a.Points = append(
|
|
a.Points,
|
|
&Point{
|
|
Measurement: measurement,
|
|
Values: map[string]interface{}{"value": value},
|
|
Tags: tags,
|
|
},
|
|
)
|
|
}
|
|
|
|
func (a *Accumulator) AddValuesWithTime(
|
|
measurement string,
|
|
values map[string]interface{},
|
|
tags map[string]string,
|
|
timestamp time.Time,
|
|
) {
|
|
a.Points = append(
|
|
a.Points,
|
|
&Point{
|
|
Measurement: measurement,
|
|
Values: values,
|
|
Tags: tags,
|
|
Time: timestamp,
|
|
},
|
|
)
|
|
}
|
|
|
|
func (a *Accumulator) Get(measurement string) (*Point, bool) {
|
|
for _, p := range a.Points {
|
|
if p.Measurement == measurement {
|
|
return p, true
|
|
}
|
|
}
|
|
|
|
return nil, false
|
|
}
|
|
|
|
func (a *Accumulator) CheckValue(measurement string, val interface{}) bool {
|
|
for _, p := range a.Points {
|
|
if p.Measurement == measurement {
|
|
return p.Values["value"] == val
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (a *Accumulator) CheckTaggedValue(measurement string, val interface{}, tags map[string]string) bool {
|
|
return a.ValidateTaggedValue(measurement, val, tags) == nil
|
|
}
|
|
|
|
func (a *Accumulator) ValidateTaggedValue(measurement string, val interface{}, tags map[string]string) error {
|
|
if tags == nil {
|
|
tags = map[string]string{}
|
|
}
|
|
for _, p := range a.Points {
|
|
if !reflect.DeepEqual(tags, p.Tags) {
|
|
continue
|
|
}
|
|
|
|
if p.Measurement == measurement {
|
|
if p.Values["value"] != val {
|
|
return fmt.Errorf("%v (%T) != %v (%T)", p.Values["value"], p.Values["value"], val, val)
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
|
|
return fmt.Errorf("unknown measurement %s with tags %v", measurement, tags)
|
|
}
|
|
|
|
func (a *Accumulator) ValidateValue(measurement string, val interface{}) error {
|
|
return a.ValidateTaggedValue(measurement, val, nil)
|
|
}
|
|
|
|
func (a *Accumulator) HasIntValue(measurement string) bool {
|
|
for _, p := range a.Points {
|
|
if p.Measurement == measurement {
|
|
_, ok := p.Values["value"].(int64)
|
|
return ok
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (a *Accumulator) HasUIntValue(measurement string) bool {
|
|
for _, p := range a.Points {
|
|
if p.Measurement == measurement {
|
|
_, ok := p.Values["value"].(uint64)
|
|
return ok
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (a *Accumulator) HasFloatValue(measurement string) bool {
|
|
for _, p := range a.Points {
|
|
if p.Measurement == measurement {
|
|
_, ok := p.Values["value"].(float64)
|
|
return ok
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|