summaryrefslogtreecommitdiffstats
path: root/spec/unit/transaction/report_spec.rb
blob: 1ce68b3c571a258fd98c8c77351aeb7444cd4a82 (plain)
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
#!/usr/bin/env ruby

require File.dirname(__FILE__) + '/../../spec_helper'

require 'puppet/transaction/report'

describe Puppet::Transaction::Report do
    before do
        Puppet::Util::Storage.stubs(:store)
    end

    it "should set its host name to the certname" do
        Puppet.settings.expects(:value).with(:certname).returns "myhost"
        Puppet::Transaction::Report.new.host.should == "myhost"
    end

    it "should return its host name as its name" do
        r = Puppet::Transaction::Report.new
        r.name.should == r.host
    end

    it "should create an initialization timestamp" do
        Time.expects(:now).returns "mytime"
        Puppet::Transaction::Report.new.time.should == "mytime"
    end

    describe "when accepting logs" do
        before do
            @report = Puppet::Transaction::Report.new
        end

        it "should add new logs to the log list" do
            @report << "log"
            @report.logs[-1].should == "log"
        end

        it "should return self" do
            r = @report << "log"
            r.should equal(@report)
        end
    end

    describe "when accepting resource statuses" do
        before do
            @report = Puppet::Transaction::Report.new
        end

        it "should add each status to its status list" do
            status = stub 'status', :resource => "foo"
            @report.add_resource_status status
            @report.resource_statuses["foo"].should equal(status)
        end
    end

    describe "when using the indirector" do
        it "should redirect :find to the indirection" do
            @indirection = stub 'indirection', :name => :report
            Puppet::Transaction::Report.stubs(:indirection).returns(@indirection)
            @indirection.expects(:find)
            Puppet::Transaction::Report.find(:report)
        end

        it "should redirect :save to the indirection" do
            Facter.stubs(:value).returns("eh")
            @indirection = stub 'indirection', :name => :report
            Puppet::Transaction::Report.stubs(:indirection).returns(@indirection)
            report = Puppet::Transaction::Report.new
            @indirection.expects(:save)
            report.save
        end

        it "should default to the 'processor' terminus" do
            Puppet::Transaction::Report.indirection.terminus_class.should == :processor
        end

        it "should delegate its name attribute to its host method" do
            report = Puppet::Transaction::Report.new
            report.expects(:host).returns "me"
            report.name.should == "me"
        end

        after do
            Puppet::Util::Cacher.expire
        end
    end

    describe "when computing exit status" do
        it "should produce 2 if changes are present" do
            report = Puppet::Transaction::Report.new
            report.add_metric("changes", {:total => 1})
            report.add_metric("resources", {:failed => 0})
            report.exit_status.should == 2
        end

        it "should produce 4 if failures are present" do
            report = Puppet::Transaction::Report.new
            report.add_metric("changes", {:total => 0})
            report.add_metric("resources", {:failed => 1})
            report.exit_status.should == 4
        end

        it "should produce 6 if both changes and failures are present" do
            report = Puppet::Transaction::Report.new
            report.add_metric("changes", {:total => 1})
            report.add_metric("resources", {:failed => 1})
            report.exit_status.should == 6
        end
    end

    describe "when calculating metrics" do
        before do
            @report = Puppet::Transaction::Report.new
        end

        def metric(name, value)
            if metric = @report.metrics[name.to_s]
                metric[value]
            else
                nil
            end
        end

        def add_statuses(count, type = :file)
            3.times do |i|
                status = Puppet::Resource::Status.new(Puppet::Type.type(type).new(:title => "/my/path#{i}"))
                yield status if block_given?
                @report.add_resource_status status
            end
        end


        [:time, :resources, :changes, :events].each do |type|
            it "should add #{type} metrics" do
                @report.calculate_metrics
                @report.metrics[type.to_s].should be_instance_of(Puppet::Transaction::Metric)
            end
        end

        describe "for resources" do
            it "should provide the total number of resources" do
                add_statuses(3)

                @report.calculate_metrics
                metric(:resources, :total).should == 3
            end

            Puppet::Resource::Status::STATES.each do |state|
                it "should provide the number of #{state} resources as determined by the status objects" do
                    add_statuses(3) { |status| status.send(state.to_s + "=", true) }

                    @report.calculate_metrics
                    metric(:resources, state).should == 3
                end
            end
        end

        describe "for changes" do
            it "should provide the number of changes from the resource statuses" do
                add_statuses(3) { |status| status.change_count = 3 }
                @report.calculate_metrics
                metric(:changes, :total).should == 9
            end
        end

        describe "for times" do
            it "should provide the total amount of time for each resource type" do
                add_statuses(3, :file) do |status|
                    status.evaluation_time = 1
                end
                add_statuses(3, :exec) do |status|
                    status.evaluation_time = 2
                end
                add_statuses(3, :mount) do |status|
                    status.evaluation_time = 3
                end

                @report.calculate_metrics

                metric(:time, "file").should == 3
                metric(:time, "exec").should == 6
                metric(:time, "mount").should == 9
            end

            it "should add any provided times from external sources" do
                @report.add_times :foobar, 50
                @report.calculate_metrics
                metric(:time, "foobar").should == 50
            end
        end

        describe "for events" do
            it "should provide the total number of events" do
                add_statuses(3) do |status|
                    3.times { |i| status.add_event(Puppet::Transaction::Event.new) }
                end
                @report.calculate_metrics
                metric(:events, :total).should == 9
            end

            Puppet::Transaction::Event::EVENT_STATUSES.each do |status_name|
                it "should provide the number of #{status_name} events" do
                    add_statuses(3) do |status|
                        3.times do |i|
                            event = Puppet::Transaction::Event.new
                            event.status = status_name
                            status.add_event(event)
                        end
                    end

                    @report.calculate_metrics
                    metric(:events, status_name).should == 9
                end
            end
        end
    end

    describe "when producing a summary" do
        before do
            resource = Puppet::Type.type(:notify).new(:name => "testing")
            catalog = Puppet::Resource::Catalog.new
            catalog.add_resource resource
            trans = catalog.apply

            @report = trans.report
            @report.calculate_metrics
        end

        %w{Changes Total Resources}.each do |main|
            it "should include information on #{main} in the summary" do
                @report.summary.should be_include(main)
            end
        end
    end
end