Remove temporary directories

Remove temporary directories for GEM_HOME and GEM_PATH at exit. Entries under these directories are removed usually, but top directories are left.
rubygems · Mar 5, 2015 · 82be822 · 82be822
1 parent 800f2e6
commit 82be822
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lib/rubygems/test_case.rb b/lib/rubygems/test_case.rb
@@ -1442,6 +1442,9 @@ def self.key_path key_name
 end
 
 require 'rubygems/test_utilities'
-ENV['GEM_HOME'] = Dir.mktmpdir "home"
-ENV['GEM_PATH'] = Dir.mktmpdir "path"
+tmpdirs = []
+tmpdirs << (ENV['GEM_HOME'] = Dir.mktmpdir("home"))
+tmpdirs << (ENV['GEM_PATH'] = Dir.mktmpdir("path"))
+pid = $$
+END {tmpdirs.each {|dir| Dir.rmdir(dir)} if $$ == pid}
 Gem.clear_paths