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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
|
#!/usr/bin/env ruby
require 'spec_helper'
require 'puppet/configurer'
require 'puppet/configurer/fact_handler'
class FactHandlerTester
include Puppet::Configurer::FactHandler
end
describe Puppet::Configurer::FactHandler do
before do
@facthandler = FactHandlerTester.new
end
it "should have a method for downloading fact plugins" do
@facthandler.should respond_to(:download_fact_plugins)
end
it "should have a boolean method for determining whether fact plugins should be downloaded" do
@facthandler.should respond_to(:download_fact_plugins?)
end
it "should download fact plugins when :factsync is true" do
Puppet.settings.expects(:value).with(:factsync).returns true
@facthandler.should be_download_fact_plugins
end
it "should not download fact plugins when :factsync is false" do
Puppet.settings.expects(:value).with(:factsync).returns false
@facthandler.should_not be_download_fact_plugins
end
it "should not download fact plugins when downloading is disabled" do
Puppet::Configurer::Downloader.expects(:new).never
@facthandler.expects(:download_fact_plugins?).returns false
@facthandler.download_fact_plugins
end
it "should use an Agent Downloader, with the name, source, destination, and ignore set correctly, to download fact plugins when downloading is enabled" do
downloader = mock 'downloader'
Puppet.settings.expects(:value).with(:factsource).returns "fsource"
Puppet.settings.expects(:value).with(:factdest).returns "fdest"
Puppet.settings.expects(:value).with(:factsignore).returns "fignore"
Puppet::Configurer::Downloader.expects(:new).with("fact", "fdest", "fsource", "fignore").returns downloader
downloader.expects(:evaluate)
@facthandler.expects(:download_fact_plugins?).returns true
@facthandler.download_fact_plugins
end
it "should warn about factsync deprecation when factsync is enabled" do
Puppet::Configurer::Downloader.stubs(:new).returns mock("downloader", :evaluate => nil)
@facthandler.expects(:download_fact_plugins?).returns true
Puppet.expects(:warning)
@facthandler.download_fact_plugins
end
it "should have a method for retrieving facts" do
@facthandler.should respond_to(:find_facts)
end
it "should use the Facts class with the :certname to find the facts" do
Puppet.settings.expects(:value).with(:certname).returns "foo"
Puppet::Node::Facts.indirection.expects(:find).with("foo").returns "myfacts"
@facthandler.stubs(:reload_facter)
@facthandler.find_facts.should == "myfacts"
end
it "should reload Facter and find local facts when asked to find facts" do
@facthandler.expects(:reload_facter)
Puppet.settings.expects(:value).with(:certname).returns "myhost"
Puppet::Node::Facts.indirection.expects(:find).with("myhost")
@facthandler.find_facts
end
it "should fail if finding facts fails" do
@facthandler.stubs(:reload_facter)
Puppet.settings.stubs(:value).with(:trace).returns false
Puppet.settings.stubs(:value).with(:certname).returns "myhost"
Puppet::Node::Facts.indirection.expects(:find).raises RuntimeError
lambda { @facthandler.find_facts }.should raise_error(Puppet::Error)
end
it "should have a method to prepare the facts for uploading" do
@facthandler.should respond_to(:facts_for_uploading)
end
# I couldn't get marshal to work for this, only yaml, so we hard-code yaml.
it "should serialize and CGI escape the fact values for uploading" do
facts = stub 'facts'
facts.expects(:support_format?).with(:b64_zlib_yaml).returns true
facts.expects(:render).returns "my text"
text = CGI.escape("my text")
@facthandler.expects(:find_facts).returns facts
@facthandler.facts_for_uploading.should == {:facts_format => :b64_zlib_yaml, :facts => text}
end
it "should properly accept facts containing a '+'" do
facts = stub 'facts'
facts.expects(:support_format?).with(:b64_zlib_yaml).returns true
facts.expects(:render).returns "my+text"
text = "my%2Btext"
@facthandler.expects(:find_facts).returns facts
@facthandler.facts_for_uploading.should == {:facts_format => :b64_zlib_yaml, :facts => text}
end
it "use compressed yaml as the serialization if zlib is supported" do
facts = stub 'facts'
facts.expects(:support_format?).with(:b64_zlib_yaml).returns true
facts.expects(:render).with(:b64_zlib_yaml).returns "my text"
text = CGI.escape("my text")
@facthandler.expects(:find_facts).returns facts
@facthandler.facts_for_uploading
end
it "should use yaml as the serialization if zlib is not supported" do
facts = stub 'facts'
facts.expects(:support_format?).with(:b64_zlib_yaml).returns false
facts.expects(:render).with(:yaml).returns "my text"
text = CGI.escape("my text")
@facthandler.expects(:find_facts).returns facts
@facthandler.facts_for_uploading
end
describe "when reloading Facter" do
before do
Facter.stubs(:clear)
Facter.stubs(:load)
Facter.stubs(:loadfacts)
end
it "should clear Facter" do
Facter.expects(:clear)
@facthandler.reload_facter
end
it "should load all Facter facts" do
Facter.expects(:loadfacts)
@facthandler.reload_facter
end
it "should use the Facter terminus load all Puppet Fact plugins" do
Puppet::Node::Facts::Facter.expects(:load_fact_plugins)
@facthandler.reload_facter
end
end
end
|
