在Ruby测试/单元拆解方法中检查测试结果

Question

是否可以在拆解方法中检查Ruby测试/单元的结果？

我正在使用Ruby with Test / Unit，WATIR和Webdriver来测试Web应用程序，如果测试失败，我想在拆解方法中获取截图。

Answer 1

如何改变assert_equal（或者你正在使用的任何断言）呢？

require 'test/unit'

class Test::Unit::TestCase
  def assert_equal(expected, got, msg)
    begin
      super(expected, got, msg)
    rescue
      p "caught ya!" # make screenshot here
      raise
    end
  end
end

class DemoTest < Test::Unit::TestCase

  def test_fail
    assert_equal(1, 0, 'ups')
  end
end