diff --git a/cayley_test.go b/cayley_test.go index b94389a..15428a8 100644 --- a/cayley_test.go +++ b/cayley_test.go @@ -363,12 +363,18 @@ func runBench(n int, b *testing.B) { 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() } }