test/helper.rb in keyrack-0.3.0.pre vs test/helper.rb in keyrack-0.3.0

@@ -8,11 +8,11 @@
   $stderr.puts e.message
   $stderr.puts "Run `bundle install` to install missing gems"
   exit e.status_code
 end
 require 'test/unit'
-require 'mocha'
+require 'mocha/setup'
 
 $LOAD_PATH.unshift(File.dirname(__FILE__))
 $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 require 'keyrack'
 
@@ -30,7 +30,17 @@
 
   def teardown
     if @tmpnames
       @tmpnames.each { |t| FileUtils.rm_rf(t) }
     end
+  end
+end
+
+class SequenceHelper < Object
+  def initialize(name)
+    @seq = Mocha::Sequence.new(name)
+  end
+
+  def <<(object)
+    object.in_sequence(@seq)
   end
 end