barak/cayley
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

Allow short tests

Browse source
This commit is contained in:
kortschak 2014-07-31 08:45:45 +09:30
parent b4887e3c30
commit 38f7b79761
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
cayley_test.go
View file

@ -26,6 +26,7 @@ import (
var benchmarkQueries = []struct { var benchmarkQueries = []struct {
message string message string
long bool
query string query string
tag string tag string
expect [][]interface{} expect [][]interface{}
@ -59,6 +60,7 @@ var benchmarkQueries = []struct {
// 2014-07-12: This one takes about 4 whole seconds in memory. This is a behemoth. // 2014-07-12: This one takes about 4 whole seconds in memory. This is a behemoth.
{ {
message: "three huge sets with small intersection", message: "three huge sets with small intersection",
long: true,
query: ` query: `
function getId(x) { return g.V(x).In("name") } function getId(x) { return g.V(x).In("name") }
var actor_to_film = g.M().In("/film/performance/actor").In("/film/film/starring") var actor_to_film = g.M().In("/film/performance/actor").In("/film/film/starring")
@ -87,6 +89,7 @@ var benchmarkQueries = []struct {
// as it has a fixed ID. Exercises Check(). // as it has a fixed ID. Exercises Check().
{ {
message: "the helpless checker", message: "the helpless checker",
long: true,
query: ` query: `
g.V().As("person").In("name").In().In().Out("name").Is("Casablanca").All() g.V().As("person").In("name").In().In().Out("name").Is("Casablanca").All()
`, `,
@ -146,6 +149,7 @@ var benchmarkQueries = []struct {
// who was in "The Net" // who was in "The Net"
{ {
message: "Keanu with other in The Net", message: "Keanu with other in The Net",
long: true,
query: common + `actor2.Follow(coStars1).Intersect(m1_actors).Out("name").All() query: common + `actor2.Follow(coStars1).Intersect(m1_actors).Out("name").All()
`, `,
expect: [][]interface{}{ expect: [][]interface{}{
@ -159,6 +163,7 @@ var benchmarkQueries = []struct {
// of Egypt", and KR starred with Steven Martin in "Parenthood". // of Egypt", and KR starred with Steven Martin in "Parenthood".
{ {
message: "Keanu and Bullock with other", message: "Keanu and Bullock with other",
long: true,
query: common + `actor1.Save("name","costar1_actor").Follow(coStars1).Intersect(actor2.Save("name","costar2_actor").Follow(coStars2)).Out("name").All() query: common + `actor1.Save("name","costar1_actor").Follow(coStars1).Intersect(actor2.Save("name","costar2_actor").Follow(coStars2)).Out("name").All()
`, `,
expect: [][]interface{}{ expect: [][]interface{}{
@ -308,6 +313,9 @@ func prepare(t testing.TB) {
func TestQueries(t *testing.T) { func TestQueries(t *testing.T) {
prepare(t) prepare(t)
for _, test := range benchmarkQueries { for _, test := range benchmarkQueries {
if testing.Short() && test.long {
continue
}
ses := gremlin.NewSession(ts, cfg.GremlinTimeout, true) ses := gremlin.NewSession(ts, cfg.GremlinTimeout, true)
_, err := ses.InputParses(test.query) _, err := ses.InputParses(test.query)
if err != nil { if err != nil {
@ -345,6 +353,9 @@ func TestQueries(t *testing.T) {
} }
func runBench(n int, b *testing.B) { func runBench(n int, b *testing.B) {
if testing.Short() && benchmarkQueries[n].long {
b.Skip()
}
prepare(b) prepare(b)
ses := gremlin.NewSession(ts, cfg.GremlinTimeout, true) ses := gremlin.NewSession(ts, cfg.GremlinTimeout, true)
_, err := ses.InputParses(benchmarkQueries[n].query) _, err := ses.InputParses(benchmarkQueries[n].query)