summaryrefslogtreecommitdiff
path: root/spec/support/matchers/leak.rb
blob: 80f0ae44c7491844900079e4c8f646291df7d82c (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
#
# Author:: Seth Chisamore (<schisamo@opscode.com>)
# Copyright:: Copyright (c) 2011 Opscode, Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

module Matchers
  module LeakBase
    include RSpec::Matchers

    def initialize(opts={}, &block)
      @warmup = opts[:warmup] || 5
      @iterations = opts[:iterations] || 100
      @variance = opts[:variance] || 5000
    end

    def failure_message
      "expected final measure [#{@final_measure}] to be greater than or within +/- #{@variance} delta of initial measure [#{@initial_measure}]"
    end

    def failure_message_when_negated
      "expected final measure [#{@final_measure}] to be less than or within +/- #{@variance} delta of initial measure [#{@initial_measure}]"
    end

    private
    def match(measure, given_proc)
      profiler.start

      @initial_measure = 0
      @final_measure = 0

      @warmup.times do
        given_proc.call
      end

      @initial_measure = profiler.send(measure)

      @iterations.times do
        given_proc.call
      end

      profiler.stop

      @final_measure = profiler.send(measure)
      @final_measure > (@initial_measure + @variance)
    end

    def profiler
      @profiler ||= begin
        if Chef::Platform.windows?
          require File.join(File.dirname(__FILE__), "..", "platforms", "prof", "win32")
          RSpec::Prof::Win32::Profiler.new
        else
          require File.join(File.dirname(__FILE__), "..", "prof", "gc")
          RSpec::Prof::GC::Profiler.new
        end
      end
    end

  end

  class LeakMemory
    include LeakBase

    def matches?(given_proc)
      match(:working_set_size, given_proc)
    end
  end

  class LeakHandles
    include LeakBase

    def matches?(given_proc)
      match(:handle_count, given_proc)
    end
  end

  def leak_memory(opts, &block)
    Matchers::LeakMemory.new(opts, &block)
  end
  def leak_handles(opts, &block)
    Matchers::LeakHandles.new(opts, &block)
  end
end