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
|
require 'spec_helper'
require 'puppet_spec/files'
describe "the 'file' function" do
include PuppetSpec::Files
before :all do
Puppet::Parser::Functions.autoloader.loadall
end
let :node do Puppet::Node.new('localhost') end
let :compiler do Puppet::Parser::Compiler.new(node) end
let :scope do Puppet::Parser::Scope.new(compiler) end
def with_file_content(content)
path = tmpfile('file-function')
file = File.new(path, 'wb')
file.sync = true
file.print content
yield path
end
it "should read a file" do
with_file_content('file content') do |name|
expect(scope.function_file([name])).to eq("file content")
end
end
it "should read a file keeping line endings intact" do
with_file_content("file content\r\n") do |name|
expect(scope.function_file([name])).to eq("file content\r\n")
end
end
it "should read a file from a module path" do
with_file_content('file content') do |name|
mod = double('module')
allow(mod).to receive(:file).with('myfile').and_return(name)
allow(compiler.environment).to receive(:module).with('mymod').and_return(mod)
expect(scope.function_file(['mymod/myfile'])).to eq('file content')
end
end
it "should return the first file if given two files with absolute paths" do
with_file_content('one') do |one|
with_file_content('two') do |two|
expect(scope.function_file([one, two])).to eq("one")
end
end
end
it "should return the first file if given two files with module paths" do
with_file_content('one') do |one|
with_file_content('two') do |two|
mod = double('module')
expect(compiler.environment).to receive(:module).with('mymod').and_return(mod)
expect(mod).to receive(:file).with('one').and_return(one)
allow(mod).to receive(:file).with('two').and_return(two)
expect(scope.function_file(['mymod/one','mymod/two'])).to eq('one')
end
end
end
it "should return the first file if given two files with mixed paths, absolute first" do
with_file_content('one') do |one|
with_file_content('two') do |two|
mod = double('module')
allow(compiler.environment).to receive(:module).with('mymod').and_return(mod)
allow(mod).to receive(:file).with('two').and_return(two)
expect(scope.function_file([one,'mymod/two'])).to eq('one')
end
end
end
it "should return the first file if given two files with mixed paths, module first" do
with_file_content('one') do |one|
with_file_content('two') do |two|
mod = double('module')
expect(compiler.environment).to receive(:module).with('mymod').and_return(mod)
allow(mod).to receive(:file).with('two').and_return(two)
expect(scope.function_file(['mymod/two',one])).to eq('two')
end
end
end
it "should not fail when some files are absent" do
expect {
with_file_content('one') do |one|
expect(scope.function_file([make_absolute("/should-not-exist"), one])).to eq('one')
end
}.to_not raise_error
end
it "should fail when all files are absent" do
expect {
scope.function_file([File.expand_path('one')])
}.to raise_error(Puppet::ParseError, /Could not find any files/)
end
end
|
