1 files changed, 36 insertions, 0 deletions
diff --git a/tool/lib/gc_checker.rb b/tool/lib/gc_checker.rb
new file mode 100644
index 0000000000..719da8cac0
--- /dev/null
+++ b/tool/lib/gc_checker.rb
@@ -0,0 +1,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']