/
error_generator.rb
119 lines (98 loc) · 3.82 KB
/
error_generator.rb
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
module RSpec
module Mocks
# @private
class ErrorGenerator
attr_writer :opts
def initialize(target, name, options={})
@declared_as = options[:__declared_as] || 'Mock'
@target = target
@name = name
end
# @private
def opts
@opts ||= {}
end
# @private
def raise_unexpected_message_error(message, *args)
__raise "#{intro} received unexpected message :#{message}#{arg_message(*args)}"
end
# @private
def raise_unexpected_message_args_error(expectation, *args)
expected_args = format_args(*expectation.expected_args)
actual_args = format_args(*args)
__raise "#{intro} received #{expectation.message.inspect} with unexpected arguments\n expected: #{expected_args}\n got: #{actual_args}"
end
# @private
def raise_missing_default_stub_error(expectation, *args)
expected_args = format_args(*expectation.expected_args)
actual_args = format_args(*args)
__raise "#{intro} received #{expectation.message.inspect} with unexpected arguments\n expected: #{expected_args}\n got: #{actual_args}\n Please stub a default value first if message might be received with other args as well. \n"
end
# @private
def raise_similar_message_args_error(expectation, *args_for_multiple_calls)
expected_args = format_args(*expectation.expected_args)
actual_args = args_for_multiple_calls.collect {|a| format_args(*a)}.join(", ")
__raise "#{intro} received #{expectation.message.inspect} with unexpected arguments\n expected: #{expected_args}\n got: #{actual_args}"
end
# @private
def raise_expectation_error(message, expected_received_count, actual_received_count, *args)
__raise "(#{intro}).#{message}#{format_args(*args)}\n expected: #{count_message(expected_received_count)}\n received: #{count_message(actual_received_count)}"
end
# @private
def raise_out_of_order_error(message)
__raise "#{intro} received :#{message} out of order"
end
# @private
def raise_block_failed_error(message, detail)
__raise "#{intro} received :#{message} but passed block failed with: #{detail}"
end
# @private
def raise_missing_block_error(args_to_yield)
__raise "#{intro} asked to yield |#{arg_list(*args_to_yield)}| but no block was passed"
end
# @private
def raise_wrong_arity_error(args_to_yield, arity)
__raise "#{intro} yielded |#{arg_list(*args_to_yield)}| to block with arity of #{arity}"
end
# @private
def raise_only_valid_on_a_partial_mock(method)
__raise "#{intro} is a pure mock object. `#{method}` is only " +
"available on a partial mock object."
end
private
def intro
if @name
"#{@declared_as} #{@name.inspect}"
elsif TestDouble === @target
@declared_as
elsif Class === @target
"<#{@target.inspect} (class)>"
elsif @target
@target
else
"nil"
end
end
def __raise(message)
message = opts[:message] unless opts[:message].nil?
Kernel::raise(RSpec::Mocks::MockExpectationError, message)
end
def arg_message(*args)
" with " + format_args(*args)
end
def format_args(*args)
args.empty? ? "(no args)" : "(" + arg_list(*args) + ")"
end
def arg_list(*args)
args.collect {|arg| arg.respond_to?(:description) ? arg.description : arg.inspect}.join(", ")
end
def count_message(count)
return "at least #{pretty_print(count.abs)}" if count < 0
return pretty_print(count)
end
def pretty_print(count)
"#{count} time#{count == 1 ? '' : 's'}"
end
end
end
end