Add "message" to assert_screen

lib/yamatanooroti/vterm.rb

@@ -105,13 +105,13 @@ def result
     @result
   end
 
-  def assert_screen(expected_lines)
+  def assert_screen(expected_lines, message = nil)
     actual_lines = result
     case expected_lines
     when Array
-      assert_equal(expected_lines, actual_lines)
+      assert_equal(expected_lines, actual_lines, message)
     when String
-      assert_equal(expected_lines, actual_lines.join("\n").sub(/\n*\z/, "\n"))
+      assert_equal(expected_lines, actual_lines.join("\n").sub(/\n*\z/, "\n"), message)
     end
   end
 end

lib/yamatanooroti/windows.rb

@@ -519,12 +519,12 @@ def result
     @result
   end
 
-  def assert_screen(expected_lines)
+  def assert_screen(expected_lines, message = nil)
     case expected_lines
     when Array
-      assert_equal(expected_lines, @result)
+      assert_equal(expected_lines, @result, message)
     when String
-      assert_equal(expected_lines, @result.join("\n").sub(/\n*\z/, "\n"))
+      assert_equal(expected_lines, @result.join("\n").sub(/\n*\z/, "\n"), message)
     end
   end
 

test/yamatanooroti/test_run_ruby.rb

@@ -44,6 +44,16 @@ def test_meta_key
     get_out_from_tmpdir
   end
 
+  def test_assert_screen_takes_a_message_when_failed
+    start_terminal(5, 30, ['ruby', '-e', 'puts "aaa"'])
+    close
+    assert_raise_with_message Test::Unit::AssertionFailedError, /\Amessage when failed/ do
+      assert_screen(<<~EOC, 'message when failed')
+        bbb
+      EOC
+    end
+  end
+
   private
 
   def get_into_tmpdir