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
|
require 'spec_helper'
require 'puppet/util/autoload'
require 'fileutils'
class AutoloadIntegrator
@things = []
def self.newthing(name)
@things << name
end
def self.thing?(name)
@things.include? name
end
def self.clear
@things.clear
end
end
require 'puppet_spec/files'
describe Puppet::Util::Autoload do
include PuppetSpec::Files
def with_file(name, *path)
path = File.join(*path)
# Now create a file to load
File.open(path, "w") { |f|
f.puts "\nAutoloadIntegrator.newthing(:#{name.to_s})\n"
}
yield
File.delete(path)
end
def with_loader(name, path)
dir = tmpfile(name + path)
$LOAD_PATH << dir
Dir.mkdir(dir)
rbdir = File.join(dir, path.to_s)
Dir.mkdir(rbdir)
loader = Puppet::Util::Autoload.new(name, path)
yield rbdir, loader
Dir.rmdir(rbdir)
Dir.rmdir(dir)
$LOAD_PATH.pop
AutoloadIntegrator.clear
end
it "should not fail when asked to load a missing file" do
expect(Puppet::Util::Autoload.new("foo", "bar").load(:eh)).to be_falsey
end
it "should load and return true when it successfully loads a file" do
with_loader("foo", "bar") { |dir,loader|
with_file(:mything, dir, "mything.rb") {
expect(loader.load(:mything)).to be_truthy
expect(loader.class).to be_loaded("bar/mything")
expect(AutoloadIntegrator).to be_thing(:mything)
}
}
end
it "should consider a file loaded when asked for the name without an extension" do
with_loader("foo", "bar") { |dir,loader|
with_file(:noext, dir, "noext.rb") {
loader.load(:noext)
expect(loader.class).to be_loaded("bar/noext")
}
}
end
it "should consider a file loaded when asked for the name with an extension" do
with_loader("foo", "bar") { |dir,loader|
with_file(:noext, dir, "withext.rb") {
loader.load(:withext)
expect(loader.class).to be_loaded("bar/withext.rb")
}
}
end
it "should be able to load files directly from modules" do
## modulepath can't be used until after app settings are initialized, so we need to simulate that:
expect(Puppet.settings).to receive(:app_defaults_initialized?).and_return(true).at_least(:once)
modulepath = tmpfile("autoload_module_testing")
libdir = File.join(modulepath, "mymod", "lib", "foo")
FileUtils.mkdir_p(libdir)
file = File.join(libdir, "plugin.rb")
Puppet.override(:environments => Puppet::Environments::Static.new(Puppet::Node::Environment.create(:production, [modulepath]))) do
with_loader("foo", "foo") do |dir, loader|
with_file(:plugin, file.split("/")) do
loader.load(:plugin)
expect(loader.class).to be_loaded("foo/plugin.rb")
end
end
end
end
end
|