Extend GraphRunner for RSpec.

lib/pacer-neo4j.rb

@@ -7,3 +7,4 @@
 require 'pacer-neo4j/edge'
 require 'pacer-neo4j/index'
 require 'pacer-neo4j/graph'
+require 'pacer-neo4j/rspec' if defined? RSpec

lib/pacer-neo4j/rspec.rb

@@ -0,0 +1,36 @@
+class Rspec::GraphRunner
+  module Neo4j
+    def initialize(*graphs)
+      super
+      if use_graph?('neo4j')
+        path1 = File.expand_path('tmp/spec.neo4j')
+        dir = Pathname.new(path1)
+        dir.rmtree if dir.exist?
+        @neo_graph = Pacer.neo4j(path1)
+
+        path2 = File.expand_path('tmp/spec.neo4j.2')
+        dir = Pathname.new(path2)
+        dir.rmtree if dir.exist?
+        @neo_graph2 = Pacer.neo4j(path2)
+
+        path3 = File.expand_path('tmp/spec_no_indices.neo4j')
+        dir = Pathname.new(path3)
+        dir.rmtree if dir.exist?
+        @neo_graph_no_indices = Pacer.neo4j(path3)
+        @neo_graph_no_indices.drop_index :vertices
+        @neo_graph_no_indices.drop_index :edges
+      end
+    end
+
+    def all(usage_style = :read_write, indices = true, &block)
+      super
+      neo4j(usage_style, indices, &block)
+    end
+
+    def neo4j(usage_style = :read_write, indices = true, &block)
+      for_graph('neo4j', usage_style, indices, true, @neo_graph, @neo_graph2, @neo_graph_no_indices, block)
+    end
+  end
+
+  include Neo4j
+end