在Module / Class的末尾执行代码，比如Ruby test / unit

Question

Ruby的单元测试框架执行单元测试，即使没有人创建单元测试对象。 例如，

在MyUnitTest.rb中

require 'test/unit'

class MyUnitTest < Test::Unit::TestCase
    def test_true
        assert true
    end
end

当我调用该脚本时

ruby MyUnitTest.rb

test_true方法自动执行。 这是怎么做到的？

我试图提出一个可以做类似的框架。 我不想在使用我的框架的每个模块的末尾“if __ FILE __ == $ 0”。

谢谢。

Answer 1

Test::Unit使用at_exit ，在应用程序退出之前立即运行代码：

at_exit do
  puts "printed before quit"
end

.. other stuff

在关闭类或模块定义之后，我认为没有任何方法可以专门运行代码。

Answer 2

与Daniel Lucraft的答案类似，您可以定义一个终结器防护，以便在垃圾收集器运行时运行一些代码：

ObjectSpace.define_finalizer(MyClass, lambda {
  puts "printed before MyClass is removed from the Object space"
})

但我个人认为at_exit更合适，因为它的时机更具确定性。