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
|
#!/usr/bin/env ruby
#
# Created by Luke A. Kanies on 2007-06-10.
# Copyright (c) 2007. All rights reserved.
require File.expand_path(File.dirname(__FILE__) + '/../../lib/puppettest')
require 'puppettest'
class TestTypeInstances < Test::Unit::TestCase
include PuppetTest
def setup
super
@type = Puppet::Type.newtype(:instance_test) do
newparam(:name) {}
ensurable
end
cleanup { Puppet::Type.rmtype(:instance_test) }
end
# Make sure the instances class method works as expected.
def test_instances
# First make sure it throws an error when there are no providers
assert_raise(Puppet::DevError, "Did not fail when no providers are present") do
@type.instances
end
# Now add a couple of providers
# The default
@type.provide(:default) do
defaultfor :operatingsystem => Facter.value(:operatingsystem)
mk_resource_methods
class << self
attr_accessor :names
end
def self.instance(name)
new(:name => name, :ensure => :present)
end
def self.instances
@instances ||= names.collect { |name| instance(name) }
end
@names = [:one, :five, :six]
end
# A provider with the same source
@type.provide(:sub, :source => :default, :parent => :default) do
@names = [:two, :seven, :eight]
end
# An unsuitable provider
@type.provide(:nope, :parent => :default) do
confine :exists => "/no/such/file"
@names = [:three, :nine, :ten]
end
# Another suitable, non-default provider
@type.provide(:yep, :parent => :default) do
@names = [:four, :seven, :ten]
end
# Now make a couple of instances, so we know we correctly match instead of always
# trying to create new ones.
one = @type.new(:name => :one, :ensure => :present)
three = @type.new(:name => :three, :ensure => :present, :provider => :sub)
five = @type.new(:name => :five, :ensure => :present, :provider => :yep)
result = nil
assert_nothing_raised("Could not get instance list") do
result = @type.instances
end
result.each do |resource|
assert_instance_of(@type, resource, "Returned non-resource")
end
assert_equal(:one, result[0].name, "Did not get default instances first")
resources = result.inject({}) { |hash, res| hash[res.name] = res; hash }
assert(resources.include?(:four), "Did not get resources from other suitable providers")
assert(! resources.include?(:three), "Got resources from unsuitable providers")
# Now make sure we didn't change the provider type for :five
assert_equal(:yep, five.provider.class.name, "Changed provider type when listing resources")
# Now make sure the resources have an 'ensure' property to go with the value in the provider
assert(resources[:one].send(:instance_variable_get, "@parameters").include?(:ensure), "Did not create ensure property")
end
end
|
