在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更合適，因為它的時機更具確定性。