lib/grumlin/test/rspec/gremlin_context.rb in grumlin-0.4.0 vs lib/grumlin/test/rspec/gremlin_context.rb in grumlin-0.5.0

@@ -6,11 +6,17 @@
       module GremlinContext
       end
 
       ::RSpec.shared_context GremlinContext do
         include GremlinContext
+        include Grumlin::Sugar
 
-        let(:g) { Grumlin::Traversal.new }
+        before do
+          Grumlin::Sugar::HELPERS.each do |helper|
+            name = helper.name.split("::").last
+            stub_const(name, helper)
+          end
+        end
 
         after do
           Grumlin.config.default_pool.close
           Grumlin.config.reset!
         end