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
|
#!/usr/bin/ruby
if __FILE__ == $0
$:.unshift '../../lib'
$:.unshift '..'
$puppetbase = "../.."
end
require 'puppet'
require 'puppet/rails'
require 'puppet/parser/interpreter'
require 'puppet/parser/parser'
require 'puppet/client'
require 'test/unit'
require 'puppettest'
class TestRails < Test::Unit::TestCase
include ParserTesting
def test_includerails
assert_nothing_raised {
require 'puppet/rails'
}
end
# Don't do any tests w/out this class
if defined? ActiveRecord::Base
def test_hostcache
# First make some objects
bucket = mk_transtree do |object, depth, width|
# and mark some of them collectable
if width % 2 == 1
object.collectable = true
end
end
# Now collect our facts
facts = {}
Facter.each do |fact, value| facts[fact] = value end
assert_nothing_raised {
Puppet::Rails.init
}
# Now try storing our crap
host = nil
assert_nothing_raised {
host = Puppet::Rails::Host.store(
:objects => bucket,
:facts => facts,
:host => facts["hostname"]
)
}
assert(host, "Did not create host")
host = nil
assert_nothing_raised {
host = Puppet::Rails::Host.find_by_name(facts["hostname"])
}
assert(host, "Could not find host object")
assert(host.rails_objects, "No objects on host")
assert_equal(facts["hostname"], host.facts["hostname"],
"Did not retrieve facts")
inline_test_objectcollection(host)
end
# This is called from another test, it just makes sense to split it out some
def inline_test_objectcollection(host)
# XXX For some reason, find_all doesn't work here at all.
collectable = []
host.rails_objects.each do |obj|
if obj.collectable?
collectable << obj
end
end
assert(collectable.length > 0, "Found no collectable objects")
collectable.each do |obj|
trans = nil
assert_nothing_raised {
trans = obj.to_trans
}
# Make sure that the objects do not retain their collectable
# nature.
assert(!trans.collectable, "Object from db was collectable")
end
# Now find all collectable objects directly through database APIs
list = Puppet::Rails::RailsObject.find_all_by_collectable(true)
assert_equal(collectable.length, list.length,
"Did not get the right number of objects")
end
def test_railsinit
assert_nothing_raised {
Puppet::Rails.init
}
assert(FileTest.exists?(Puppet[:dblocation]), "Database does not exist")
end
else
$stderr.puts "Install Rails for Rails and Caching tests"
end
end
# $Id$
|
