added helper method of minitest

1 files changed, 14 insertions, 0 deletions

diff --git a/test/utils.rb b/test/utils.rb
index c6e0ccf3..607daea8 100644
--- a/test/utils.rb
+++ b/test/utils.rb
@@ -334,3 +334,17 @@ AQjjxMXhwULlmuR/K+WwlaZPiLIBYalLAZQ7ZbOPeVkJ8ePao0eLAgEC
   end
 
 end if defined?(OpenSSL)
+
+module Test
+  module Unit
+    class TestCase
+      def message msg = nil, ending = ".", &default
+        proc {
+          msg = msg.call.chomp(".") if Proc === msg
+          custom_message = "#{msg}.\n" unless msg.nil? or msg.to_s.empty?
+          "#{custom_message}#{default.call}#{ending}"
+        }
+      end
+    end
+  end
+end