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
|
#! /usr/bin/env ruby
require 'spec_helper'
require 'tempfile'
describe "the extlookup function" do
before :all do
Puppet::Parser::Functions.autoloader.loadall
end
before :each do
@scope = Puppet::Parser::Scope.new
@scope.stubs(:environment).returns(Puppet::Node::Environment.new('production'))
end
it "should exist" do
Puppet::Parser::Functions.function("extlookup").should == "function_extlookup"
end
it "should raise a ParseError if there is less than 1 arguments" do
lambda { @scope.function_extlookup([]) }.should( raise_error(Puppet::ParseError))
end
it "should raise a ParseError if there is more than 3 arguments" do
lambda { @scope.function_extlookup(["foo", "bar", "baz", "gazonk"]) }.should( raise_error(Puppet::ParseError))
end
it "should return the default" do
result = @scope.function_extlookup([ "key", "default"])
result.should == "default"
end
it "should lookup the key in a supplied datafile" do
t = Tempfile.new('extlookup.csv') do
t.puts 'key,value'
t.puts 'nonkey,nonvalue'
t.close
result = @scope.function_extlookup([ "key", "default", t.path])
result.should == "value"
end
end
it "should return an array if the datafile contains more than two columns" do
t = Tempfile.new('extlookup.csv') do
t.puts 'key,value1,value2'
t.puts 'nonkey,nonvalue,nonvalue'
t.close
result = @scope.function_extlookup([ "key", "default", t.path])
result.should == ["value1", "value2"]
end
end
it "should raise an error if there's no matching key and no default" do
t = Tempfile.new('extlookup.csv') do
t.puts 'key,value'
t.puts 'nonkey,nonvalue'
t.close
result = @scope.function_extlookup([ "key", nil, t.path])
result.should == "value"
end
end
describe "should look in $extlookup_datadir for data files listed by $extlookup_precedence" do
before do
@scope.stubs(:lookupvar).with('extlookup_datadir').returns("/tmp")
File.open("/tmp/one.csv","w"){|one| one.puts "key,value1" }
File.open("/tmp/two.csv","w") do |two|
two.puts "key,value2"
two.puts "key2,value_two"
end
end
it "when the key is in the first file" do
@scope.stubs(:lookupvar).with('extlookup_precedence').returns(["one","two"])
result = @scope.function_extlookup([ "key" ])
result.should == "value1"
end
it "when the key is in the second file" do
@scope.stubs(:lookupvar).with('extlookup_precedence').returns(["one","two"])
result = @scope.function_extlookup([ "key2" ])
result.should == "value_two"
end
it "should not modify extlookup_precedence data" do
variable = '%{fqdn}'
@scope.stubs(:lookupvar).with('extlookup_precedence').returns([variable,"one"])
@scope.stubs(:lookupvar).with('fqdn').returns('myfqdn')
result = @scope.function_extlookup([ "key" ])
variable.should == '%{fqdn}'
end
end
end
|
