1 files changed, 23 insertions, 18 deletions
diff --git a/test/ruby/test_jit.rb b/test/ruby/test_jit.rb
index 92a7bd56d8..039a2ff35f 100644
--- a/test/ruby/test_jit.rb
+++ b/test/ruby/test_jit.rb
@@ -530,26 +530,31 @@ class TestJIT < Test::Unit::TestCase
   end
 
   def test_unload_units
-    # MIN_CACHE_SIZE is 10
-    out, err = eval_with_jit("#{<<~"begin;"}\n#{<<~'end;'}", verbose: 1, min_calls: 1, max_cache: 10)
-    begin;
-      10.times do |i|
-        eval(<<-EOS)
-          def mjit#{i}
-            print #{i}
-          end
-          mjit#{i}
-        EOS
+    Dir.mktmpdir("jit_test_clean_so_") do |dir|
+      # MIN_CACHE_SIZE is 10
+      out, err = eval_with_jit({"TMPDIR"=>dir}, "#{<<~"begin;"}\n#{<<~'end;'}", verbose: 1, min_calls: 1, max_cache: 10)
+      begin;
+        10.times do |i|
+          eval(<<-EOS)
+            def mjit#{i}
+              print #{i}
+            end
+            mjit#{i}
+          EOS
+        end
+      end;
+      assert_equal('0123456789', out)
+      errs = err.lines
+      assert_match(/\A#{JIT_SUCCESS_PREFIX}: block in <main>@-e:/, errs[0])
+      9.times do |i|
+        assert_match(/\A#{JIT_SUCCESS_PREFIX}: mjit#{i}@\(eval\):/, errs[i + 1])
       end
-    end;
-    assert_equal('0123456789', out)
-    errs = err.lines
-    assert_match(/\A#{JIT_SUCCESS_PREFIX}: block in <main>@-e:/, errs[0])
-    9.times do |i|
-      assert_match(/\A#{JIT_SUCCESS_PREFIX}: mjit#{i}@\(eval\):/, errs[i + 1])
+      assert_equal("Too many JIT code -- 1 units unloaded\n", errs[10])
+      assert_match(/\A#{JIT_SUCCESS_PREFIX}: mjit9@\(eval\):/, errs[11])
+
+      # verify .o files are deleted on unload_units
+      assert_send([Dir, :empty?, dir])
     end
-    assert_equal("Too many JIT code -- 1 units unloaded\n", errs[10])
-    assert_match(/\A#{JIT_SUCCESS_PREFIX}: mjit9@\(eval\):/, errs[11])
   end
 
   def test_local_stack_on_exception