mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-24 03:06:48 +01:00
9edeecabc8
This makes easier to read and debug these tests. This also reduces test lines count by 15% from 3K to 2.5K . See https://itnext.io/f-tests-as-a-replacement-for-table-driven-tests-in-go-8814a8b19e9e . While at it, consistently use t.Fatal* instead of t.Error*, since t.Error* usually leads to more complicated and fragile tests, while it doesn't bring any practical benefits over t.Fatal*.
79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
package stream
|
|
|
|
import (
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/fasttime"
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/protoparser/graphite"
|
|
)
|
|
|
|
func TestStreamContextRead(t *testing.T) {
|
|
f := func(s string, rowsExpected *graphite.Rows) {
|
|
t.Helper()
|
|
ctx := getStreamContext(strings.NewReader(s))
|
|
if !ctx.Read() {
|
|
t.Fatalf("expecting successful read")
|
|
}
|
|
uw := getUnmarshalWork()
|
|
callbackCalls := 0
|
|
uw.ctx = ctx
|
|
uw.callback = func(rows []graphite.Row) error {
|
|
callbackCalls++
|
|
if len(rows) != len(rowsExpected.Rows) {
|
|
t.Fatalf("different len of expected rows;\ngot\n%+v;\nwant\n%+v", rows, rowsExpected.Rows)
|
|
}
|
|
if !reflect.DeepEqual(rows, rowsExpected.Rows) {
|
|
t.Fatalf("unexpected rows;\ngot\n%+v;\nwant\n%+v", rows, rowsExpected.Rows)
|
|
}
|
|
return nil
|
|
}
|
|
uw.reqBuf = append(uw.reqBuf[:0], ctx.reqBuf...)
|
|
ctx.wg.Add(1)
|
|
uw.Unmarshal()
|
|
if callbackCalls != 1 {
|
|
t.Fatalf("unexpected number of callback calls; got %d; want 1", callbackCalls)
|
|
}
|
|
}
|
|
|
|
// Full line without tags
|
|
f("aaa 1123 345", &graphite.Rows{
|
|
Rows: []graphite.Row{{
|
|
Metric: "aaa",
|
|
Value: 1123,
|
|
Timestamp: 345 * 1000,
|
|
}},
|
|
})
|
|
// Full line with tags
|
|
f("aaa;x=y 1123 345", &graphite.Rows{
|
|
Rows: []graphite.Row{{
|
|
Metric: "aaa",
|
|
Tags: []graphite.Tag{{
|
|
Key: "x",
|
|
Value: "y",
|
|
}},
|
|
Value: 1123,
|
|
Timestamp: 345 * 1000,
|
|
}},
|
|
})
|
|
// missing timestamp.
|
|
// Note that this test may be flaky due to timing issues. TODO: fix it
|
|
f("aaa 1123", &graphite.Rows{
|
|
Rows: []graphite.Row{{
|
|
Metric: "aaa",
|
|
Value: 1123,
|
|
Timestamp: int64(fasttime.UnixTimestamp()) * 1000,
|
|
}},
|
|
})
|
|
// -1 timestamp. See https://github.com/VictoriaMetrics/VictoriaMetrics/issues/610
|
|
// Note that this test may be flaky due to timing issues. TODO: fix it.
|
|
f("aaa 1123 -1", &graphite.Rows{
|
|
Rows: []graphite.Row{{
|
|
Metric: "aaa",
|
|
Value: 1123,
|
|
Timestamp: int64(fasttime.UnixTimestamp()) * 1000,
|
|
}},
|
|
})
|
|
}
|