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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
require 'fileutils'
require 'tempfile'
require 'tmpdir'
require 'pathname'
# A support module for testing files.
module PuppetSpec::Files
@global_tempfiles = []
def self.cleanup
until @global_tempfiles.empty?
path = @global_tempfiles.pop
begin
allow(Dir).to receive(:entries).and_call_original
FileUtils.rm_rf path, secure: true
rescue Errno::ENOENT
# nothing to do
end
end
end
def make_absolute(path)
PuppetSpec::Files.make_absolute(path)
end
def self.make_absolute(path)
path = File.expand_path(path)
path[0] = 'c' if Puppet.features.microsoft_windows?
path
end
def tmpfile(name, dir = nil)
PuppetSpec::Files.tmpfile(name, dir)
end
def self.tmpfile(name, dir = nil)
# Generate a temporary file, just for the name...
source = dir ? Tempfile.new(name, dir) : Tempfile.new(name)
path = Puppet::FileSystem.expand_path(source.path.encode(Encoding::UTF_8))
source.close!
record_tmp(File.expand_path(path))
path
end
def file_containing(name, contents)
PuppetSpec::Files.file_containing(name, contents)
end
def self.file_containing(name, contents)
file = tmpfile(name)
File.open(file, 'wb') { |f| f.write(contents) }
file
end
def script_containing(name, contents)
PuppetSpec::Files.script_containing(name, contents)
end
def self.script_containing(name, contents)
file = tmpfile(name)
if Puppet.features.microsoft_windows?
file += '.bat'
text = contents[:windows]
else
text = contents[:posix]
end
File.open(file, 'wb') { |f| f.write(text) }
Puppet::FileSystem.chmod(0o755, file)
file
end
def tmpdir(name)
PuppetSpec::Files.tmpdir(name)
end
def self.tmpdir(name)
dir = Puppet::FileSystem.expand_path(Dir.mktmpdir(name).encode!(Encoding::UTF_8))
record_tmp(dir)
dir
end
def dir_containing(name, contents_hash)
PuppetSpec::Files.dir_containing(name, contents_hash)
end
def self.dir_containing(name, contents_hash)
dir_contained_in(tmpdir(name), contents_hash)
end
def dir_contained_in(dir, contents_hash)
PuppetSpec::Files.dir_contained_in(dir, contents_hash)
end
def self.dir_contained_in(dir, contents_hash)
contents_hash.each do |k, v|
if v.is_a?(Hash)
Dir.mkdir(tmp = File.join(dir, k))
dir_contained_in(tmp, v)
else
file = File.join(dir, k)
File.open(file, 'wb') { |f| f.write(v) }
end
end
dir
end
def self.record_tmp(tmp)
# ...record it for cleanup,
@global_tempfiles ||= []
@global_tempfiles << tmp
end
def expect_file_mode(file, mode)
actual_mode = '%o' % Puppet::FileSystem.stat(file).mode
target_mode = if Puppet.features.microsoft_windows?
mode
else
'10' + '%04i' % mode.to_i
end
expect(actual_mode).to eq(target_mode)
end
end
|
