From 453890a97296d9e342384832bffad5af031d0bca Mon Sep 17 00:00:00 2001
From: akr <akr@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Sun, 15 Nov 2009 17:11:54 +0000
Subject: tests refined.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@25784 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 test/ruby/envutil.rb | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

(limited to 'test/ruby/envutil.rb')

diff --git a/test/ruby/envutil.rb b/test/ruby/envutil.rb
index dfc75bb0e1..293e3f2a50 100644
--- a/test/ruby/envutil.rb
+++ b/test/ruby/envutil.rb
@@ -127,7 +127,8 @@ module Test
       end
 
       LANG_ENVS = %w"LANG LC_ALL LC_CTYPE"
-      def assert_in_out_err(args, test_stdin = "", test_stdout = [], test_stderr = [], message = nil, opt={})
+
+      def invoke_ruby_assertion(args, test_stdin="", test_stdout=nil, test_stderr=nil, test_status=true, message = nil, opt={})
         in_c, in_p = IO.pipe
         out_p, out_c = IO.pipe if test_stdout
         err_p, err_c = IO.pipe if test_stderr
@@ -155,6 +156,7 @@ module Test
         out_p.close if test_stdout
         err_p.close if test_stderr
         Process.wait pid
+        status = $?
         if block_given?
           yield(test_stdout ? stdout.lines.map {|l| l.chomp } : nil, test_stderr ? stderr.lines.map {|l| l.chomp } : nil)
         else
@@ -173,6 +175,10 @@ module Test
             end
           end
         end
+        if test_status
+          assert(status.success?, "ruby exit stauts is not success: #{status.inspect}")
+        end
+        status
       ensure
         env.each_pair {|lc, v|
           if v
@@ -191,9 +197,14 @@ module Test
         (th_stderr.kill; th_stderr.join) if th_stderr
       end
 
-      def assert_in_out(args, test_stdin = "", test_stdout = [], message = nil, opt={})
-        assert_in_out_err(args, test_stdin, test_stdout, nil, message, opt)
+      def assert_in_out_err(args, test_stdin = "", test_stdout = [], test_stderr = [], message = nil, opt={})
+        invoke_ruby_assertion(args, test_stdin, test_stdout, test_stderr, false, message, opt)
+      end
+
+      def assert_ruby_status(args, test_stdin = "", message = nil, opt={})
+        invoke_ruby_assertion(args, test_stdin, nil, nil, true, message, opt)
       end
+
     end
   end
 end
-- 
cgit v1.2.3