blob: f6f35cc4763b403365196c074031f47001a5d767 (
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
|
require 'mspec/expectations/expectations'
require 'mspec/runner/formatters/dotted'
class SpinnerFormatter < DottedFormatter
attr_reader :length
Spins = %w!| / - \\!
HOUR = 3600
MIN = 60
def initialize(out=nil)
super(nil)
@which = 0
@loaded = 0
self.length = 40
@percent = 0
@start = Time.now
term = ENV['TERM']
@color = (term != "dumb")
@fail_color = "32"
@error_color = "32"
end
def register
super
MSpec.register :start, self
MSpec.register :unload, self
MSpec.unregister :before, self
end
def length=(length)
@length = length
@ratio = 100.0 / length
@position = length / 2 - 2
end
def compute_etr
return @etr = "00:00:00" if @percent == 0
elapsed = Time.now - @start
remain = (100 * elapsed / @percent) - elapsed
hour = remain >= HOUR ? (remain / HOUR).to_i : 0
remain -= hour * HOUR
min = remain >= MIN ? (remain / MIN).to_i : 0
sec = remain - min * MIN
@etr = "%02d:%02d:%02d" % [hour, min, sec]
end
def compute_percentage
@percent = @loaded * 100 / @total
bar = ("=" * (@percent / @ratio)).ljust @length
label = "%d%%" % @percent
bar[@position, label.size] = label
@bar = bar
end
def compute_progress
compute_percentage
compute_etr
end
def progress_line
@which = (@which + 1) % Spins.size
data = [Spins[@which], @bar, @etr, @counter.failures, @counter.errors]
if @color
"\r[%s | %s | %s] \e[0;#{@fail_color}m%6dF \e[0;#{@error_color}m%6dE\e[0m " % data
else
"\r[%s | %s | %s] %6dF %6dE " % data
end
end
def clear_progress_line
print "\r#{' '*progress_line.length}"
end
# Callback for the MSpec :start event. Stores the total
# number of files that will be processed.
def start
@total = MSpec.retrieve(:files).size
compute_progress
print progress_line
end
# Callback for the MSpec :unload event. Increments the number
# of files that have been run.
def unload
@loaded += 1
compute_progress
print progress_line
end
# Callback for the MSpec :exception event. Changes the color
# used to display the tally of errors and failures
def exception(exception)
super
@fail_color = "31" if exception.failure?
@error_color = "33" unless exception.failure?
clear_progress_line
print_exception(exception, @count)
end
# Callback for the MSpec :after event. Updates the spinner.
def after(state)
print progress_line
end
def finish(printed_exceptions = true)
# We already printed the exceptions
@exceptions = [] if printed_exceptions
super()
end
end
|
