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
|
require 'puppet-lint'
#class PuppetLint::Warning < Exception; end
#class PuppetLint::Error < Exception; end
#PuppetLint::CheckPlugin.any_instance.stub(:warn) do |arg|
# raise PuppetLint::Warning
#end
#PuppetLint::CheckPlugin.any_instance.stub(:error) do |arg|
# raise PuppetLint::Error
#end
# filter_array_of_hashes(array, filter) -> an_array
#
# Filters out hashes by applying filter_hash to each hash
# in the array. All set value/key pairs in filter_hash must
# match before a hash is allowed.
# Returns all hashes that matched in an array.
#
# filter_array_of_hashes(
# [
# {:filter => 1, :name => 'one'},
# {:filter => 2, :name => 'two'},
# {:filter => 3, :name => 'three'},
# ],
# { :filter => 2 }
# )
# => [{:filter=>2, :name=>"two"}]
#
# filter_array_of_hashes([{:f => 1}, {:f => 2}], {})
# => [{:f=>1}, {:f=>2}]
#
def filter_array_of_hashes(array_of_hashes, filter_hash)
array_of_hashes.select { |hash_to_check|
val = true
filter_hash.each do |k,v|
if ! hash_to_check.key?(k)
val = false
break
elsif hash_to_check[k].to_s != v.to_s
val = false
break
end
end
val
}
end
RSpec::Matchers.define :have_problem do |filter|
match do |problems|
filter_array_of_hashes(problems, filter).length > 0
end
failure_message_for_should do |problems|
message = "could not find any problems matching the filter."
message << "
* filter = #{filter.inspect}
* problems = [
"
problems.each { |prob| message << " #{prob.inspect}," }
message << "
]"
message
end
failure_message_for_should_not do |problems|
message = "some problems matched the filter."
message << "
* filter = #{filter.inspect}
* matched = [
"
filter_array_of_hashes(problems, filter).each { |prob| message << " #{prob.inspect}," }
message << "
]"
message
end
end
RSpec::Matchers.define :only_have_problem do |filter|
match do |actual|
res = filter_array_of_hashes(actual, filter)
res.length == actual.length
end
failure_message_for_should do |problems|
left = problems - filter_array_of_hashes(actual, filter)
message = "There were problems not matching filter."
message << "
* filter = #{filter.inspect}
* unmatched = [
"
left.each { |prob| message << " #{prob.inspect}," }
message << "
]"
message
end
failure_message_for_should_not do |problems|
message = "There were no problems found besides the ones matched with filter."
message << "
* filter = #{filter.inspect}
"
message
end
end
|
