lib/grumlin/test/rspec/gremlin_context.rb in grumlin-0.12.3 vs lib/grumlin/test/rspec/gremlin_context.rb in grumlin-0.12.4

@@ -9,12 +9,11 @@
       ::RSpec.shared_context GremlinContext do
         include GremlinContext
         include Grumlin::Sugar
 
         before do
-          Grumlin::Sugar::HELPERS.each do |helper|
-            name = helper.name.split("::").last
-            stub_const(name, helper)
+          Grumlin::Tools.constants.each do |tool|
+            stub_const(tool.to_s, Grumlin::Tools.const_get(tool))
           end
         end
 
         after do
           Grumlin.close