gistfile1.rb

# Adapted from https://github.com/irrationalfab/PrettyBacon

module PrettyBacon
  def self.color(color, string)
    case color
    when :red
      "\e[31m#{string}\e[0m"
    when :green
      "\e[32m#{string}\e[0m"
    when :yellow
      "\e[33m#{string}\e[0m"
    when :none
      string
    else
      "\e[0m#{string}\e[0m"
    end
  end
end

module Bacon
  class Context
    # Add support for disabled specs
    #
    def xit(description, &block)
      if ENV['run-disabled']
        it(description, &block)
      else
        Counter[:disabled] += 1
        #Bacon.handle_requirement_begin(description, true)
        #Bacon.handle_requirement_end(nil)
        it(description) { Bacon.running_disabled_spec = true; true.should == true }
      end
    end

    alias_method :on_64bit_it, RUBY_ARCH =~ /64/ ? :it : :xit

    alias_method :on_ios_it, defined?(UIView) ? :it : :xit
  end

  # Overrides the SpecDoxzRtput to provide colored output by default
  #
  # Based on https://github.com/zen-cms/Zen-Core and subsequently modified
  # which is available under the MIT License. Thanks YorickPeterse!
  #
  module PrettySpecDoxOutput

    def handle_specification_begin(name)
      if @needs_first_put
        @needs_first_put = false
        puts
      end
      @specs_depth = @specs_depth || 0
      puts spaces + name
      @specs_depth += 1
    end

    def handle_specification_end
      @specs_depth -= 1
      puts if @specs_depth.zero?
    end

    attr_accessor :running_disabled_spec

    #def handle_requirement_begin(description, disabled = false)
    def handle_requirement_begin(description)
      self.running_disabled_spec = false
      @start_time = Time.now.to_f
      @description = description
    end

    def handle_requirement_end(error)
      elapsed_time = ((Time.now.to_f - @start_time) * 1000).round

      if !error.empty?
        puts PrettyBacon.color(:red, "#{spaces}- #{@description} [FAILED]")
      #elsif @disabled
      elsif Bacon.running_disabled_spec
        puts PrettyBacon.color(:yellow, "#{spaces}- #{@description} [DISABLED]")
      else
        time_color = case elapsed_time
          when 0..200
            :none
          when 200..500
            :yellow
          else
            :red
          end

        if elapsed_time <= 1
          elapsed_time_string = ''
        elsif elapsed_time >= 1000
          elapsed_time_string = "(#{'%.1f' % (elapsed_time/1000.0)} s)"
        else
          elapsed_time_string = "(#{elapsed_time} ms)"
        end

        elapsed_time_string = PrettyBacon.color(time_color, " #{elapsed_time_string}") unless elapsed_time_string == ''

        puts PrettyBacon.color(:green, "#{spaces}✓ ") + "#{@description}" + elapsed_time_string
      end
    end

    #:nodoc:
    def handle_summary
      print ErrorLog  if Backtraces
      unless Counter[:disabled].zero?
        puts PrettyBacon.color(:yellow, "#{Counter[:disabled]} disabled specifications\n")
      end
      puts "%d specifications (%d requirements), %d failures, %d errors" %
        Counter.values_at(:specifications, :requirements, :failed, :errors)
    end

    #:nodoc:
    def spaces
      return '  ' * (@specs_depth || 0)
    end
  end

  module TapOutput
    attr_accessor :running_disabled_spec
  end

  Outputs['pretty_spec_dox'] = PrettySpecDoxOutput
end