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
165
166
167
168
169
170
171
172
173
174
|
test_name "the pluginsync functionality should sync feature and function definitions" do
tag 'audit:high',
'audit:integration'
#
# This test is intended to ensure that pluginsync syncs feature definitions to
# the agents. It checks the feature twice; once to make sure that it gets
# loaded successfully during the run in which it was synced, and once to ensure
# that it still gets loaded successfully during the subsequent run (in which it
# should not be synced because the files haven't changed.)
#
require 'puppet/acceptance/environment_utils'
extend Puppet::Acceptance::EnvironmentUtils
require 'puppet/acceptance/temp_file_utils'
extend Puppet::Acceptance::TempFileUtils
teardown do
agents.each do |agent|
on(agent, puppet('config print lastrunfile')) do |command_result|
agent.rm_rf(command_result.stdout)
end
end
end
module_name = 'superbogus'
tmp_environment = mk_tmp_environment_with_teardown(master, 'sync')
master_module_dir = "#{environmentpath}/#{tmp_environment}/modules"
master_module_type_path = "#{master_module_dir}/#{module_name}/lib/puppet/type/"
master_module_feature_path = "#{master_module_dir}/#{module_name}/lib/puppet/feature"
master_module_function_path = "#{master_module_dir}/#{module_name}/lib/puppet/functions"
master_module_function_path_namespaced = "#{master_module_dir}/#{module_name}/lib/puppet/functions/superbogus"
on(master, "mkdir -p '#{master_module_dir}'")
on(master, "mkdir -p '#{master_module_type_path}' '#{master_module_feature_path}' '#{master_module_function_path}' #{master_module_function_path_namespaced}")
master_module_type_file = "#{master_module_type_path}/#{module_name}.rb"
master_module_type_content = <<-HERE
module Puppet
Type.newtype(:#{module_name}) do
newparam(:name) do
isnamevar
end
newproperty(:testfeature) do
def sync
Puppet.info("The value of the #{module_name} feature is: \#{Puppet.features.#{module_name}?}")
end
def retrieve
:absent
end
def insync?(is)
false
end
end
end
end
HERE
create_remote_file(master, master_module_type_file, master_module_type_content)
master_module_feature_file = "#{master_module_feature_path}/#{module_name}.rb"
master_module_feature_content = <<-HERE
Puppet.features.add(:#{module_name}) do
Puppet.info("#{module_name} feature being queried")
true
end
HERE
create_remote_file(master, master_module_feature_file, master_module_feature_content)
on(master, "chmod 755 '#{master_module_type_file}' '#{master_module_feature_file}'")
master_module_function_file = "#{master_module_function_path}/bogus_function.rb"
master_module_function_content = <<-HERE
Puppet::Functions.create_function(:bogus_function) do
dispatch :bogus_function do
end
def bogus_function()
three = call_function('round', 3.14)
hostname = `facter hostname`
"Three is \#{three}. bogus_function reporting hostname is \#{hostname}"
end
end
HERE
create_remote_file(master, master_module_function_file, master_module_function_content)
on(master, "chmod 755 '#{master_module_function_file}' '#{master_module_function_file}'")
master_module_namespaced_function_file = "#{master_module_function_path_namespaced}/bogus_function2.rb"
master_module_namespaced_function_content = <<-HERE
Puppet::Functions.create_function(:'superbogus::bogus_function2') do
dispatch :bogus_function2 do
end
def bogus_function2()
four = call_function('round', 4.14)
hostname = `facter hostname`
"Four is \#{four}. bogus_function reporting hostname is \#{hostname}"
end
end
HERE
create_remote_file(master, master_module_namespaced_function_file, master_module_namespaced_function_content)
on(master, "chmod 755 '#{master_module_namespaced_function_file}'")
site_pp = <<-HERE
#{module_name} { "This is the title of the #{module_name} type instance in site.pp":
testfeature => "Hi. I'm setting the testfeature property of #{module_name} here in site.pp",
}
notify { module_function:
message => Deferred('bogus_function', [])
}
notify { module_function2:
message => Deferred('superbogus::bogus_function2', [])
}
HERE
create_sitepp(master, tmp_environment, site_pp)
# These master opts should not be necessary whence content negotation for
# Puppet 6.0.0 is completed, and this should just be removed.
master_opts = {
'master' => {
'rich_data' => 'true'
}
}
step 'start the master' do
with_puppet_running_on(master, master_opts) do
agents.each do |agent|
agent_lib_dir = agent.tmpdir('libdir')
agent_module_type_file = "#{agent_lib_dir}/puppet/type/#{module_name}.rb"
agent_module_feature_file = "#{agent_lib_dir}/puppet/feature/#{module_name}.rb"
agent_module_function_file = "#{agent_lib_dir}/puppet/functions/bogus_function.rb"
agent_module_namespaced_function_file = "#{agent_lib_dir}/puppet/functions/superbogus/bogus_function2.rb"
facter_hostname = fact_on(agent, 'hostname')
step "verify that the module files don't exist on the agent path" do
[agent_module_type_file, agent_module_feature_file, agent_module_function_file].each do |file_path|
if file_exists?(agent, file_path)
fail_test("file should not exist on the agent yet: '#{file_path}'")
end
end
end
step 'run the agent and verify that it loaded the feature' do
on(agent, puppet("agent -t --libdir='#{agent_lib_dir}' --rich_data --environment '#{tmp_environment}'"),
:acceptable_exit_codes => [2]) do |result|
assert_match(/The value of the #{module_name} feature is: true/, result.stdout,
"Expected agent stdout to include confirmation that the feature was 'true'")
assert_match(/Three is 3. bogus_function reporting hostname is #{facter_hostname}/, result.stdout,
"Expect the agent stdout to run bogus_function and report hostname")
assert_match(/Four is 4. bogus_function reporting hostname is #{facter_hostname}/, result.stdout,
"Expect the agent stdout to run bogus_function and report hostname")
end
end
step 'verify that the module files were synced down to the agent' do
[agent_module_type_file, agent_module_feature_file, agent_module_function_file, agent_module_namespaced_function_file].each do |file_path|
unless file_exists?(agent, file_path)
fail_test("Expected file to exist on the agent now: '#{file_path}'")
end
end
end
step 'run the agent again with a cached catalog' do
on(agent, puppet("agent -t --libdir='#{agent_lib_dir}' --use_cached_catalog --rich_data --environment '#{tmp_environment}'"), :acceptable_exit_codes => [2]) do |result|
assert_match(/The value of the #{module_name} feature is: true/, result.stdout,
"Expected agent stdout to include confirmation that the feature was 'true'")
assert_match(/Three is 3. bogus_function reporting hostname is #{facter_hostname}/, result.stdout,
"Expect the agent stdout to run bogus_function and report hostname")
assert_match(/Four is 4. bogus_function reporting hostname is #{facter_hostname}/, result.stdout,
"Expect the agent stdout to run bogus_function and report hostname")
end
end
end
end
end
end
|
