mirror of
https://github.com/Oxalide/vsphere-influxdb-go.git
synced 2023-10-10 13:36:51 +02:00
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package statement
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/stress/v2/stress_client"
|
|
)
|
|
|
|
func TestInsertSetID(t *testing.T) {
|
|
e := newTestInsert()
|
|
newID := "oaijnifo"
|
|
e.SetID(newID)
|
|
if e.StatementID != newID {
|
|
t.Errorf("Expected: %v\nGot: %v\n", newID, e.StatementID)
|
|
}
|
|
}
|
|
|
|
func TestInsertRun(t *testing.T) {
|
|
i := newTestInsert()
|
|
s, packageCh, _ := stressClient.NewTestStressTest()
|
|
// Listen to the other side of the directiveCh
|
|
go func() {
|
|
for pkg := range packageCh {
|
|
countPoints := i.Timestamp.Count
|
|
batchSize := s.BatchSize
|
|
got := len(strings.Split(string(pkg.Body), "\n"))
|
|
switch got {
|
|
case countPoints % batchSize:
|
|
case batchSize:
|
|
default:
|
|
t.Errorf("countPoints: %v\nbatchSize: %v\ngot: %v\n", countPoints, batchSize, got)
|
|
}
|
|
pkg.Tracer.Done()
|
|
}
|
|
}()
|
|
i.Run(s)
|
|
}
|
|
|
|
func newTestInsert() *InsertStatement {
|
|
return &InsertStatement{
|
|
TestID: "foo_test",
|
|
StatementID: "foo_ID",
|
|
Name: "foo_name",
|
|
TemplateString: "cpu,%v %v %v",
|
|
Timestamp: newTestTimestamp(),
|
|
Templates: newTestTemplates(),
|
|
TagCount: 1,
|
|
}
|
|
}
|