fix(mock): Fix unit tests
parent
533b5c4ed6
commit
bb50560481
|
@ -9,6 +9,7 @@ import (
|
||||||
|
|
||||||
type GeneratorResultSet struct {
|
type GeneratorResultSet struct {
|
||||||
sg gen.SeriesGenerator
|
sg gen.SeriesGenerator
|
||||||
|
tags models.Tags
|
||||||
max int
|
max int
|
||||||
count int
|
count int
|
||||||
f floatTimeValuesGeneratorCursor
|
f floatTimeValuesGeneratorCursor
|
||||||
|
@ -55,6 +56,7 @@ func NewResultSetFromSeriesGenerator(sg gen.SeriesGenerator, opts ...GeneratorOp
|
||||||
}
|
}
|
||||||
|
|
||||||
func (g *GeneratorResultSet) Next() bool {
|
func (g *GeneratorResultSet) Next() bool {
|
||||||
|
g.tags = g.tags[:0]
|
||||||
remain := g.max - g.count
|
remain := g.max - g.count
|
||||||
return g.sg.Next() && (g.max == 0 || remain > 0)
|
return g.sg.Next() && (g.max == 0 || remain > 0)
|
||||||
}
|
}
|
||||||
|
@ -83,7 +85,24 @@ func (g *GeneratorResultSet) Cursor() cursors.Cursor {
|
||||||
return g.cur
|
return g.cur
|
||||||
}
|
}
|
||||||
|
|
||||||
func (g *GeneratorResultSet) Tags() models.Tags { return g.sg.Tags() }
|
func copyTags(dst, src models.Tags) models.Tags {
|
||||||
|
if cap(dst) < src.Len() {
|
||||||
|
dst = make(models.Tags, src.Len())
|
||||||
|
} else {
|
||||||
|
dst = dst[:src.Len()]
|
||||||
|
}
|
||||||
|
copy(dst, src)
|
||||||
|
return dst
|
||||||
|
}
|
||||||
|
|
||||||
|
func (g *GeneratorResultSet) Tags() models.Tags {
|
||||||
|
if len(g.tags) == 0 {
|
||||||
|
g.tags = copyTags(g.tags, g.sg.Tags())
|
||||||
|
g.tags.Set(models.MeasurementTagKeyBytes, g.sg.Name())
|
||||||
|
g.tags.Set(models.FieldKeyTagKeyBytes, g.sg.Field())
|
||||||
|
}
|
||||||
|
return g.tags
|
||||||
|
}
|
||||||
func (g *GeneratorResultSet) Close() {}
|
func (g *GeneratorResultSet) Close() {}
|
||||||
func (g *GeneratorResultSet) Err() error { return nil }
|
func (g *GeneratorResultSet) Err() error { return nil }
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue