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
|
# frozen_string_literal: true
require 'rspec/core'
require 'rspec/core/formatters/base_text_formatter'
require 'ruby-progressbar'
require 'fuubar/output'
::RSpec.configuration.add_setting :fuubar_progress_bar_options, :default => {}
::RSpec.configuration.add_setting :fuubar_auto_refresh, :default => false
::RSpec.configuration.add_setting :fuubar_output_pending_results, :default => true
class Fuubar < ::RSpec::Core::Formatters::BaseTextFormatter
DEFAULT_PROGRESS_BAR_OPTIONS = { :format => ' %c/%C |%w>%i| %e ' }.freeze
::RSpec::Core::Formatters.register self,
:close,
:dump_failures,
:dump_pending,
:example_failed,
:example_passed,
:example_pending,
:message,
:start
attr_accessor :example_tick_lock,
:progress,
:passed_count,
:pending_count,
:failed_count
def initialize(*args)
super
self.example_tick_lock = ::Mutex.new
self.progress = ::ProgressBar.create(
DEFAULT_PROGRESS_BAR_OPTIONS.
merge(:throttle_rate => continuous_integration? ? 1.0 : nil).
merge(:total => 0,
:output => output,
:autostart => false)
)
end
def start(notification)
progress_bar_options = DEFAULT_PROGRESS_BAR_OPTIONS.
merge(:throttle_rate => continuous_integration? ? 1.0 : nil).
merge(configuration.fuubar_progress_bar_options).
merge(:total => notification.count,
:output => output,
:autostart => false)
self.progress = ::ProgressBar.create(progress_bar_options)
self.passed_count = 0
self.pending_count = 0
self.failed_count = 0
super
with_current_color { progress.start }
end
def close(_notification)
example_tick_thread.kill
end
def example_passed(_notification)
self.passed_count += 1
increment
end
def example_pending(_notification)
self.pending_count += 1
increment
end
def example_failed(notification)
self.failed_count += 1
progress.clear
output.puts notification.fully_formatted(failed_count)
output.puts
increment
end
def example_tick(_notification)
example_tick_lock.synchronize do
refresh
end
end
def example_tick_thread
::Thread.new do
loop do
sleep(1)
if configuration.fuubar_auto_refresh
example_tick(notification)
end
end
end
end
def message(notification)
if progress.respond_to? :log
progress.log(notification.message)
else
super
end
end
def dump_failures(_notification)
#
# We output each failure as it happens so we don't need to output them en
# masse at the end of the run.
#
end
def dump_pending(notification)
return unless configuration.fuubar_output_pending_results
super
end
def output
@fuubar_output ||= ::Fuubar::Output.new(super, configuration.tty?) # rubocop:disable Naming/MemoizedInstanceVariableName
end
private
def increment
with_current_color { progress.increment }
end
def refresh
with_current_color { progress.refresh }
end
def with_current_color
output.print "\e[#{color_code_for(current_color)}m" if color_enabled?
yield
output.print "\e[0m" if color_enabled?
end
def color_enabled?
configuration.color_enabled? && !continuous_integration?
end
def current_color
if failed_count > 0
configuration.failure_color
elsif pending_count > 0
configuration.pending_color
else
configuration.success_color
end
end
def color_code_for(*args)
::RSpec::Core::Formatters::ConsoleCodes.console_code_for(*args)
end
def configuration
::RSpec.configuration
end
def continuous_integration?
@continuous_integration ||= \
![nil, '', 'false'].include?(ENV['CONTINUOUS_INTEGRATION'])
end
end
|