blob: 8d3dd01fb084a02194deb632a02926840fa3c804 (
plain)
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
|
#! /usr/bin/env ruby -S rspec
require 'spec_helper'
describe "the generate function" do
include PuppetSpec::Files
before :all do
Puppet::Parser::Functions.autoloader.loadall
end
let(:scope) { Puppet::Parser::Scope.new }
it "should exist" do
Puppet::Parser::Functions.function("generate").should == "function_generate"
end
it "accept a fully-qualified path as a command" do
command = File.expand_path('/command/foo')
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == "yay"
end
it "should not accept a relative path as a command" do
lambda { scope.function_generate(["command"]) }.should raise_error(Puppet::ParseError)
end
it "should not accept a command containing illegal characters" do
lambda { scope.function_generate([File.expand_path('/##/command')]) }.should raise_error(Puppet::ParseError)
end
it "should not accept a command containing spaces" do
lambda { scope.function_generate([File.expand_path('/com mand')]) }.should raise_error(Puppet::ParseError)
end
it "should not accept a command containing '..'" do
command = File.expand_path("/command/../")
lambda { scope.function_generate([command]) }.should raise_error(Puppet::ParseError)
end
it "should execute the generate script with the correct working directory" do
command = File.expand_path("/command")
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
describe "on Windows", :as_platform => :windows do
it "should accept the tilde in the path" do
command = "C:/DOCUME~1/ADMINI~1/foo.bat"
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
it "should accept lower-case drive letters" do
command = 'd:/command/foo'
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
it "should accept upper-case drive letters" do
command = 'D:/command/foo'
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
it "should accept forward and backslashes in the path" do
command = 'D:\command/foo\bar'
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
it "should reject colons when not part of the drive letter" do
lambda { scope.function_generate(['C:/com:mand']) }.should raise_error(Puppet::ParseError)
end
it "should reject root drives" do
lambda { scope.function_generate(['C:/']) }.should raise_error(Puppet::ParseError)
end
end
describe "on non-Windows", :as_platform => :posix do
it "should reject backslashes" do
lambda { scope.function_generate(['/com\\mand']) }.should raise_error(Puppet::ParseError)
end
it "should accept plus and dash" do
command = "/var/folders/9z/9zXImgchH8CZJh6SgiqS2U+++TM/-Tmp-/foo"
Dir.expects(:chdir).with(File.dirname(command)).returns("yay")
scope.function_generate([command]).should == 'yay'
end
end
let :command do
cmd = tmpfile('function_generate')
if Puppet.features.microsoft_windows?
cmd += '.bat'
text = '@echo off' + "\n" + 'echo a-%1 b-%2'
else
text = '#!/bin/sh' + "\n" + 'echo a-$1 b-$2'
end
File.open(cmd, 'w') {|fh| fh.puts text }
File.chmod 0700, cmd
cmd
end
it "should call generator with no arguments" do
scope.function_generate([command]).should == "a- b-\n"
end
it "should call generator with one argument" do
scope.function_generate([command, 'one']).should == "a-one b-\n"
end
it "should call generator with wo arguments" do
scope.function_generate([command, 'one', 'two']).should == "a-one b-two\n"
end
it "should fail if generator is not absolute" do
expect { scope.function_generate(['boo']) }.to raise_error Puppet::ParseError
end
it "should fail if generator fails" do
expect { scope.function_generate(['/boo']) }.to raise_error Puppet::ParseError
end
end
|
