nobu	2012-02-10 00:47:11 +0900 (Fri, 10 Feb 2012)

  New Revision: 34520

  http://svn.ruby-lang.org/cgi-bin/viewvc.cgi?view=rev&revision=34520

  Log:
    * test/ruby/envutil.rb (assert_no_memory_leak): new assertion to
      check memory leak by invoking child ruby process and watch its
      memory size.

  Modified files:
    trunk/ChangeLog
    trunk/test/-ext-/string/test_modify_expand.rb
    trunk/test/ruby/envutil.rb

Index: ChangeLog
===================================================================
--- ChangeLog	(revision 34519)
+++ ChangeLog	(revision 34520)
@@ -1,3 +1,9 @@
+Fri Feb 10 00:47:07 2012  Nobuyoshi Nakada  <nobu@r...>
+
+	* test/ruby/envutil.rb (assert_no_memory_leak): new assertion to
+	  check memory leak by invoking child ruby process and watch its
+	  memory size.
+
 Thu Feb  9 23:41:44 2012  CHIKANAGA Tomoyuki  <nagachika00@g...>
 
 	* test/pathname/test_pathname.rb (test_binread): add assertion to
Index: test/ruby/envutil.rb
===================================================================
--- test/ruby/envutil.rb	(revision 34519)
+++ test/ruby/envutil.rb	(revision 34520)
@@ -184,6 +184,28 @@
         assert(msg === stderr, "warning message #{stderr.inspect} is expected to match #{msg.inspect}")
       end
 
+      def assert_no_memory_leak(args, prepare, code, message=nil, limit: 1.5)
+        token = "\e[7;1m#{$$.to_s}:#{Time.now.strftime('%s.%L')}:#{rand(0x10000).to_s(16)}:\e[m"
+        token_dump = token.dump
+        token_re = Regexp.quote(token)
+        args = [
+          "--disable=gems",
+          "-r", File.expand_path("../memory_status", __FILE__),
+          *args,
+          "-v", "-",
+        ]
+        cmd = [
+          'END {STDERR.puts '"#{token_dump}"'"FINAL=#{Memory::Status.new.size}"}',
+          prepare,
+          'STDERR.puts('"#{token_dump}"'"START=#{$initial_size = Memory::Status.new.size}")',
+          code,
+        ].join("\n")
+        out, err, status = EnvUtil.invoke_ruby(args, cmd, true, true)
+        before = err.sub!(/^#{token_re}START=(\d+)\n/, '') && $1.to_i
+        after = err.sub!(/^#{token_re}FINAL=(\d+)\n/, '') && $1.to_i
+        assert_equal([true, ""], [status.success?, err], message)
+        assert_operator(after.fdiv(before), :<, limit, message)
+      end
     end
   end
 end
Index: test/-ext-/string/test_modify_expand.rb
===================================================================
--- test/-ext-/string/test_modify_expand.rb	(revision 34519)
+++ test/-ext-/string/test_modify_expand.rb	(revision 34520)
@@ -4,26 +4,12 @@
 
 class Test_StringModifyExpand < Test::Unit::TestCase
   def test_modify_expand_memory_leak
-    before = after = nil
-    args = [
-      "--disable=gems", "-r-test-/string/string",
-      "-I"+File.expand_path("../../..", __FILE__),
-      "-rruby/memory_status",
-      "-e", <<-CMD
+    assert_no_memory_leak(["-r-test-/string/string"], <<-PRE, <<-CMD, "rb_str_modify_expand()")
       s=Bug::String.new
-      size=Memory::Status.new.size
-      puts size
+    PRE
+      size = $initial_size
       10.times{s.modify_expand!(size)}
       s.replace("")
-      puts Memory::Status.new.size
     CMD
-    ]
-    status = EnvUtil.invoke_ruby(args, "", true) do |in_p, out_p, err_p, pid|
-      before, after = out_p.readlines.map(&:to_i)
-      Process.wait(pid)
-      $?
-    end
-    assert_equal(true, status.success?)
-    assert_operator after.fdiv(before), :<, 2
   end
 end

--
ML: ruby-changes@q...
Info: http://www.atdot.net/~ko1/quickml/