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
|
#!/usr/bin/env ruby
$:.unshift("../lib").unshift("../../lib") if __FILE__ =~ /\.rb$/
require 'puppet'
require 'puppet/autoload'
require 'puppettest'
class TestAutoload < Test::Unit::TestCase
include PuppetTest
@things = []
def self.newthing(name)
@things << name
end
def self.thing?(name)
@things.include? name
end
def self.clear
@things.clear
end
def mkfile(name, path)
# Now create a file to load
File.open(path, "w") do |f|
f.puts %{
TestAutoload.newthing(:#{name.to_s})
}
end
end
def teardown
super
self.class.clear
end
def test_load
dir = tempfile()
$: << dir
cleanup do
$:.delete(dir)
end
Dir.mkdir(dir)
rbdir = File.join(dir, "yayness")
Dir.mkdir(rbdir)
# An object for specifying autoload
klass = self.class
loader = nil
assert_nothing_raised {
loader = Puppet::Autoload.new(klass, :yayness)
}
assert_equal(loader.object_id, Puppet::Autoload[klass].object_id,
"Did not retrieve loader object by class")
# Make sure we don't fail on missing files
assert_nothing_raised {
assert_equal(false, loader.load(:mything),
"got incorrect return on failed load")
}
# Now create a couple of files for testing
path = File.join(rbdir, "mything.rb")
mkfile(:mything, path)
opath = File.join(rbdir, "othing.rb")
mkfile(:othing, opath)
# Now try to actually load it.
assert_nothing_raised {
assert_equal(true, loader.load(:mything),
"got incorrect return on failed load")
}
assert(loader.loaded?(:mything), "Not considered loaded")
assert(klass.thing?(:mything),
"Did not get loaded thing")
# Now clear everything, and test loadall
assert_nothing_raised {
loader.clear
}
self.class.clear
assert_nothing_raised {
loader.loadall
}
[:mything, :othing].each do |thing|
assert(loader.loaded?(thing), "#{thing.to_s} not considered loaded")
assert(klass.thing?(thing),
"Did not get loaded #{thing.to_s}")
end
end
end
|
