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
|
#!/usr/bin/env ruby
require File.dirname(__FILE__) + '/../lib/puppettest'
require 'puppet'
require 'puppettest'
class TestStorage < Test::Unit::TestCase
include PuppetTest
def mkfile
path = tempfile()
File.open(path, "w") { |f| f.puts :yayness }
f = Puppet.type(:file).create(
:name => path,
:check => %w{checksum type}
)
return f
end
def test_storeandretrieve
path = tempfile()
f = mkfile()
# Load first, since that's what we do in the code base; this creates
# all of the necessary directories.
assert_nothing_raised {
Puppet::Util::Storage.load
}
hash = {:a => :b, :c => :d}
state = nil
assert_nothing_raised {
state = Puppet::Util::Storage.cache(f)
}
assert(!state.include?("name"))
assert_nothing_raised {
state["name"] = hash
}
assert_nothing_raised {
Puppet::Util::Storage.store
}
assert_nothing_raised {
Puppet::Util::Storage.clear
}
assert_nothing_raised {
Puppet::Util::Storage.load
}
# Reset it
state = nil
assert_nothing_raised {
state = Puppet::Util::Storage.cache(f)
}
assert_equal(state["name"], hash)
end
# we're getting corrupt files, probably because multiple processes
# are reading or writing the file at once
# so we need to test that
def test_multiwrite
f = mkfile()
value = {:a => :b}
threads = []
9.times { |a|
threads << Thread.new {
9.times { |b|
assert_nothing_raised {
Puppet::Util::Storage.load
state = Puppet::Util::Storage.cache(f)
value.each { |k,v| state[k] = v }
state[:e] = rand(100)
Puppet::Util::Storage.store
}
}
}
}
threads.each { |th| th.join }
end
def test_emptyrestore
Puppet::Util::Storage.load
Puppet::Util::Storage.store
Puppet::Util::Storage.clear
Puppet::Util::Storage.load
f = mkfile()
state = Puppet::Util::Storage.cache(f)
assert_same Hash, state.class
assert_equal 0, state.size
end
def test_caching
hash = nil
one = Puppet::Type.type(:exec).create :title => "/bin/echo one"
[one, :yayness].each do |object|
assert_nothing_raised do
hash = Puppet::Util::Storage.cache(object)
end
assert_equal({}, hash, "Did not get empty hash back for %s" % object)
hash[:testing] = true
assert_nothing_raised do
hash = Puppet::Util::Storage.cache(object)
end
assert_equal({:testing => true}, hash, "Did not get hash back for %s" % object)
end
assert_raise(ArgumentError, "was able to cache from string") do
Puppet::Util::Storage.cache("somethingelse")
end
end
end
|
