diff --git a/cayley_test.go b/cayley_test.go index b94389a..96461e7 100644 --- a/cayley_test.go +++ b/cayley_test.go @@ -358,17 +358,18 @@ func runBench(n int, b *testing.B) { b.Skip() } prepare(b) - ses := gremlin.NewSession(ts, cfg.Timeout, true) - _, err := ses.InputParses(benchmarkQueries[n].query) - if err != nil { - b.Fatalf("Failed to parse benchmark gremlin %s: %v", benchmarkQueries[n].message, err) - } + b.StopTimer() b.ResetTimer() for i := 0; i < b.N; i++ { c := make(chan interface{}, 5) + ses := gremlin.NewSession(ts, cfg.Timeout, true) + // Do the parsing we know works. + ses.InputParses(benchmarkQueries[n].query) + b.StartTimer() go ses.ExecInput(benchmarkQueries[n].query, c, 100) for _ = range c { } + b.StopTimer() } }