1package statement 2 3import ( 4 "testing" 5 6 "github.com/influxdata/influxdb/stress/v2/stress_client" 7) 8 9func TestExecSetID(t *testing.T) { 10 e := newTestExec() 11 newID := "oaijnifo" 12 e.SetID(newID) 13 if e.StatementID != newID { 14 t.Errorf("Expected: %v\nGot: %v\n", newID, e.StatementID) 15 } 16} 17 18func TestExecRun(t *testing.T) { 19 e := newTestExec() 20 s, _, _ := stressClient.NewTestStressTest() 21 e.Run(s) 22 if e == nil { 23 t.Fail() 24 } 25} 26 27func TestExecReport(t *testing.T) { 28 e := newTestExec() 29 s, _, _ := stressClient.NewTestStressTest() 30 rep := e.Report(s) 31 if rep != "" { 32 t.Fail() 33 } 34} 35 36func newTestExec() *ExecStatement { 37 return &ExecStatement{ 38 StatementID: "fooID", 39 Script: "fooscript.txt", 40 } 41} 42