tool/lib/gc_checker.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

# frozen_string_literal: true

module GCDisabledChecker
  def before_setup
    if @__gc_disabled__ = GC.enable # return true if GC is disabled
      GC.disable
    end

    super
  end

  def after_teardown
    super

    disabled = GC.enable
    GC.disable if @__gc_disabled__

    if @__gc_disabled__ != disabled
      label = {
        true => 'disabled',
        false => 'enabled',
      }
      raise "GC was #{label[@__gc_disabled__]}, but is #{label[disabled]} after the test."
    end
  end
end

module GCCompactChecker
  def after_teardown
    super
    GC.compact
  end
end

Test::Unit::TestCase.include GCDisabledChecker
Test::Unit::TestCase.include GCCompactChecker if ENV['RUBY_TEST_GC_COMPACT']