42 lines
704 B
Go
42 lines
704 B
Go
package statement
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/influxdata/influxdb/stress/v2/stress_client"
|
|
)
|
|
|
|
func TestWaitSetID(t *testing.T) {
|
|
e := newTestWait()
|
|
newID := "oaijnifo"
|
|
e.SetID(newID)
|
|
if e.StatementID != newID {
|
|
t.Errorf("Expected: %v\ngott: %v\n", newID, e.StatementID)
|
|
}
|
|
}
|
|
|
|
func TestWaitRun(t *testing.T) {
|
|
e := newTestWait()
|
|
s, _, _ := stressClient.NewTestStressTest()
|
|
e.Run(s)
|
|
if e == nil {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func TestWaitReport(t *testing.T) {
|
|
e := newTestWait()
|
|
s, _, _ := stressClient.NewTestStressTest()
|
|
rpt := e.Report(s)
|
|
if !strings.Contains(rpt, "WAIT") {
|
|
t.Fail()
|
|
}
|
|
}
|
|
|
|
func newTestWait() *WaitStatement {
|
|
return &WaitStatement{
|
|
StatementID: "fooID",
|
|
}
|
|
}
|