aboutsummaryrefslogtreecommitdiffstats
path: root/activesupport/lib/active_support/core_ext/test/difference.rb
blob: 05eab30603af703466f93c11987c2e96d12fd6da (plain) (blame)
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
module Test #:nodoc:
  module Unit #:nodoc:
    class TestCase #:nodoc:      
      # Test numeric difference between the return value of an expression as a result of what is evaluated
      # in the yielded block.
      #
      #   assert_difference 'Article.count' do
      #     post :create, :article => {...}
      #   end
      #
      # An arbitrary expression is passed in and evaluated.
      #
      #   assert_difference 'assigns(:article).comments(:reload).size' do
      #     post :create, :comment => {...}
      #   end
      #
      # An arbitrary positive or negative difference can be specified. The default is +1.
      #
      #   assert_difference 'Article.count', -1 do
      #     post :delete, :id => ...
      #   end
      #
      # An array of expressions can also be passed in and evaluated.
      #
      #   assert_difference [ 'Article.count', 'Post.count' ], +2 do
      #     post :create, :article => {...}
      #   end
      def assert_difference(expressions, difference = 1, &block)
        expression_evaluations = [expressions].flatten.collect{|expression| lambda { eval(expression, block.binding) } } 
        
        original_values = expression_evaluations.inject([]) { |memo, expression| memo << expression.call }
        yield
        expression_evaluations.each_with_index do |expression, i|
          assert_equal original_values[i] + difference, expression.call
        end
      end

      # Assertion that the numeric result of evaluating an expression is not changed before and after
      # invoking the passed in block.
      #
      #   assert_no_difference 'Article.count' do
      #     post :create, :article => invalid_attributes
      #   end
      def assert_no_difference(expressions, &block)
        assert_difference expressions, 0, &block
      end
    end
  end
end