test/test_helper.rb in redgraph-0.1.3 vs test/test_helper.rb in redgraph-0.1.4

@@ -17,5 +17,17 @@
 
 unless $REDIS_URL = ENV['TEST_REDIS_URL']
   puts "To run the tests you need to define the TEST_REDIS_URL environment variable"
   exit(1)
 end
+
+module TestHelpers
+  private
+
+  def quick_add_node(label:, properties:)
+    @graph.add_node(Redgraph::Node.new(label: label, properties: properties))
+  end
+
+  def quick_add_edge(type:, src:, dest:, properties:)
+    @graph.add_edge(Redgraph::Edge.new(type: type, src: src, dest: dest, properties: properties))
+  end
+end