3rdparty/modules/stdlib/spec/lib/puppet_spec/files.rb

   1 #! /usr/bin/env ruby -S rspec
   2 require 'fileutils'
   3 require 'tempfile'
   4 require 'tmpdir'
   5 require 'pathname'
   6
   7 # A support module for testing files.
   8 module PuppetSpec::Files
   9   def self.cleanup
  10     $global_tempfiles ||= []
  11     while path = $global_tempfiles.pop do
  12       begin
  13         Dir.unstub(:entries)
  14         FileUtils.rm_rf path, :secure => true
  15       rescue Errno::ENOENT
  16         # nothing to do
  17       end
  18     end
  19   end
  20
  21   def make_absolute(path) PuppetSpec::Files.make_absolute(path) end
  22   def self.make_absolute(path)
  23     path = File.expand_path(path)
  24     path[0] = 'c' if Puppet.features.microsoft_windows?
  25     path
  26   end
  27
  28   def tmpfile(name, dir = nil) PuppetSpec::Files.tmpfile(name, dir) end
  29   def self.tmpfile(name, dir = nil)
  30     # Generate a temporary file, just for the name...
  31     source = dir ? Tempfile.new(name, dir) : Tempfile.new(name)
  32     path = source.path
  33     source.close!
  34
  35     record_tmp(File.expand_path(path))
  36
  37     path
  38   end
  39
  40   def file_containing(name, contents) PuppetSpec::Files.file_containing(name, contents) end
  41   def self.file_containing(name, contents)
  42     file = tmpfile(name)
  43     File.open(file, 'wb') { |f| f.write(contents) }
  44     file
  45   end
  46
  47   def tmpdir(name) PuppetSpec::Files.tmpdir(name) end
  48   def self.tmpdir(name)
  49     dir = Dir.mktmpdir(name)
  50
  51     record_tmp(dir)
  52
  53     dir
  54   end
  55
  56   def self.record_tmp(tmp)
  57     # ...record it for cleanup,
  58     $global_tempfiles ||= []
  59     $global_tempfiles << tmp
  60   end
  61 end