test clean

cayley_test.go

@@ -298,24 +298,25 @@ var m2_actors = movie2.Save("name","movie2").Follow(filmToActor)
 var (
 	once sync.Once
 	cfg  = &config.Config{
-		DatabasePath: "30kmoviedata.nq.gz",
-		DatabaseType: "memstore",
-		Timeout:      300 * time.Second,
+		DatabasePath:    "30kmoviedata.nq.gz",
+		DatabaseType:    "memstore",
+		ReplicationType: "single",
+		Timeout:         300 * time.Second,
 	}
 
-	ts graph.TripleStore
+	handle *graph.Handle
 )
 
 func prepare(t testing.TB) {
 	var err error
 	once.Do(func() {
-		ts, err = db.Open(cfg)
+		handle, err = db.Open(cfg)
 		if err != nil {
 			t.Fatalf("Failed to open %q: %v", cfg.DatabasePath, err)
 		}
 
 		if !graph.IsPersistent(cfg.DatabaseType) {
-			err = load(ts, cfg, "", "cquad")
+			err = load(handle.QuadWriter, cfg, "", "cquad")
 			if err != nil {
 				t.Fatalf("Failed to load %q: %v", cfg.DatabasePath, err)
 			}
@@ -329,7 +330,7 @@ func TestQueries(t *testing.T) {
 		if testing.Short() && test.long {
 			continue
 		}
-		ses := gremlin.NewSession(ts, cfg.Timeout, true)
+		ses := gremlin.NewSession(handle.QuadStore, cfg.Timeout, true)
 		_, err := ses.InputParses(test.query)
 		if err != nil {
 			t.Fatalf("Failed to parse benchmark gremlin %s: %v", test.message, err)
@@ -374,7 +375,7 @@ func runBench(n int, b *testing.B) {
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
 		c := make(chan interface{}, 5)
-		ses := gremlin.NewSession(ts, cfg.Timeout, true)
+		ses := gremlin.NewSession(handle.QuadStore, cfg.Timeout, true)
 		// Do the parsing we know works.
 		ses.InputParses(benchmarkQueries[n].query)
 		b.StartTimer()