lib/assert/context/test_dsl.rb in assert-2.15.0 vs lib/assert/context/test_dsl.rb in assert-2.15.1

@@ -1,5 +1,6 @@
+require 'assert/context_info'
 require 'assert/macro'
 require 'assert/suite'
 require 'assert/test'
 
 module Assert; end
@@ -12,21 +13,21 @@
         instance_eval(&desc_or_macro)
       elsif block_given?
         # create a test from the given code block
         self.suite.tests << Assert::Test.for_block(
           desc_or_macro.kind_of?(Assert::Macro) ? desc_or_macro.name : desc_or_macro,
-          Assert::Suite::ContextInfo.new(self, called_from, first_caller || caller.first),
+          Assert::ContextInfo.new(self, called_from, first_caller || caller.first),
           self.suite.config,
           &block
         )
       else
         test_eventually(desc_or_macro, called_from, first_caller || caller.first, &block)
       end
     end
 
     def test_eventually(desc_or_macro, called_from = nil, first_caller = nil, &block)
       # create a test from a proc that just skips
-      ci = Assert::Suite::ContextInfo.new(self, called_from, first_caller || caller.first)
+      ci = Assert::ContextInfo.new(self, called_from, first_caller || caller.first)
       self.suite.tests << Assert::Test.for_block(
         desc_or_macro.kind_of?(Assert::Macro) ? desc_or_macro.name : desc_or_macro,
         ci,
         self.suite.config,
         &proc { skip('TODO', ci.called_from) }