test_from_readme.rb in jellyfish-1.0.1

- old
+ new
@@ -1,37 +1,43 @@
 
 require 'jellyfish/test'
 require 'uri'
+require 'stringio'
 
 describe 'from README.md' do
   after do
     [:Tank, :Heater, :Protector].each do |const|
       Object.send(:remove_const, const) if Object.const_defined?(const)
     end
+    Muack.verify
   end
 
   readme = File.read(
              "#{File.dirname(File.expand_path(__FILE__))}/../README.md")
   codes  = readme.scan(
     /### ([^\n]+).+?``` ruby\n(.+?)\n```\n\n<!---(.+?)-->/m)
 
   codes.each.with_index do |(title, code, test), index|
-    if title =~ /NewRelic/i
-      warn "Skip NewRelic Test" unless Bacon.kind_of?(Bacon::TestUnitOutput)
-      next
-    end
+    next if title =~ /NewRelic/i
 
-    should "pass from README.md #%02d #{title}" % index do
+    would "pass from README.md #%02d #{title}" % index do
       method_path, expect = test.strip.split("\n", 2)
       method, path        = method_path.split(' ')
       uri                 = URI.parse(path)
       pinfo, query        = uri.path, uri.query
 
+      sock = nil
       status, headers, body = File.open(File::NULL) do |input|
         Rack::Builder.app{ eval(code) }.call(
           'REQUEST_METHOD' => method, 'PATH_INFO'  => pinfo,
-          'QUERY_STRING'   => query , 'rack.input' => input,
-          'SCRIPT_NAME'    => '')
+          'QUERY_STRING'   => query , 'SCRIPT_NAME'=> ''   ,
+          'rack.input'     => input ,
+          'rack.hijack'    => lambda{
+            sock = StringIO.new
+            # or TypeError: no implicit conversion of StringIO into IO
+            mock(IO).select([sock]){ [[sock], [], []] }
+            sock
+          })
       end
 
       body.extend(Enumerable)
       [status, headers, body.to_a].should.eq eval(expect, binding, __FILE__)
     end