rtyler
/
rspec-puppet
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

Work around the autoloader if we're testing a standalone module

This commit is contained in:
Tim Sharpe 2011-07-19 08:24:38 +10:00
parent c5d7b8d7db
commit c722b015eb
2 changed files with 23 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
lib/rspec-puppet/example/class_example_group.rb
View File

@ -10,10 +10,20 @@ module RSpec::Puppet
Puppet[:modulepath] = module_path
klass_name = self.class.top_level_description.downcase
if !self.respond_to?(:params) || params == {}
Puppet[:code] = "include #{klass_name}"
# If we're testing a standalone module (i.e. one that's outside of a
# puppet tree), the autoloader won't work, so we need to fudge it a bit.
if File.exists?(File.join(module_path, 'manifests', 'init.pp'))
path_to_manifest = File.join([module_path, 'manifests', define_name.split('::')[1..-1]].flatten)
import_str = "import '#{module_path}/manifests/init.pp'\nimport '#{path_to_manifest}.pp'\n"
else
Puppet[:code] = 'class' + " { " + klass_name + ": " + params.keys.map { |r| "#{r.to_s} => '#{params[r].to_s}'"
import_str = ""
end
if !self.respond_to?(:params) || params == {}
Puppet[:code] = import_str + "include #{klass_name}"
else
Puppet[:code] = import_str + 'class' + " { " + klass_name + ": " + params.keys.map { |r| "#{r.to_s} => '#{params[r].to_s}'"
}.join(', ') + " }"
end

11
lib/rspec-puppet/example/define_example_group.rb
View File

@ -11,6 +11,15 @@ module RSpec::Puppet
Puppet[:modulepath] = module_path
# If we're testing a standalone module (i.e. one that's outside of a
# puppet tree), the autoloader won't work, so we need to fudge it a bit.
if File.exists?(File.join(module_path, 'manifests', 'init.pp'))
path_to_manifest = File.join([module_path, 'manifests', define_name.split('::')[1..-1]].flatten)
import_str = "import '#{module_path}/manifests/init.pp'\nimport '#{path_to_manifest}.pp'\n"
else
import_str = ""
end
if self.respond_to? :params
param_str = params.keys.map { |r|
"#{r.to_s} => \"#{params[r].to_s}\""
@ -19,7 +28,7 @@ module RSpec::Puppet
param_str = ""
end
Puppet[:code] = define_name + " { \"" + title + "\": " + param_str + " }"
Puppet[:code] = import_str + define_name + " { \"" + title + "\": " + param_str + " }"
nodename = self.respond_to?(:node) ? node : Puppet[:certname]
facts_val = self.respond_to?(:facts) ? facts : {}