require File.join(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__)))), 'spec_helper')
require File.join(File.dirname(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__))))), 'lib', 'reek', 'smells', 'simulated_polymorphism')
require File.join(File.dirname(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__))))), 'lib', 'reek', 'core', 'code_context')
require File.join(File.dirname(File.expand_path(__FILE__)), 'smell_detector_shared')

include Reek::Core
include Reek::Smells

describe SimulatedPolymorphism do
  before :each do
    @source_name = 'howdy-doody'
    @detector = SimulatedPolymorphism.new(@source_name)
  end

  it_should_behave_like 'SmellDetector'

  context 'with no conditionals' do
    it 'gathers an empty hash' do
      ast = 'module Stable; end'.to_reek_source.syntax_tree
      ctx = CodeContext.new(nil, ast)
      @detector.conditional_counts(ctx).length.should == 0
    end
  end

  context 'with a test of block_given?' do
    it 'does not record the condition' do
      ast = 'def fred() yield(3) if block_given?; end'.to_reek_source.syntax_tree
      ctx = CodeContext.new(nil, ast)
      @detector.conditional_counts(ctx).length.should == 0
    end
  end

  context 'with three identical conditionals' do
    before :each do
      @cond = '@field == :sym'
      @cond_expr = @cond.to_reek_source.syntax_tree
      src = <<EOS
class Scrunch
  def first
    puts "hello" if @debug
    return #{@cond} ? 0 : 3;
  end
  def second
    if #{@cond}
      @other += " quarts"
    end
  end
  def third
    raise 'flu!' unless #{@cond}
  end
end
EOS

      ast = src.to_reek_source.syntax_tree
      @ctx = CodeContext.new(nil, ast)
      @conds = @detector.conditional_counts(@ctx)
    end
    it 'finds both conditionals' do
      @conds.length.should == 2
    end
    it 'knows there are three copies' do
      @conds[@cond_expr].length.should == 3
    end

    context 'looking at the YAML' do
      before :each do
        @detector.examine(@ctx)
        warning = @detector.smells_found.to_a[0]   # SMELL: too cumbersome!
        @yaml = warning.to_yaml
      end
      it 'reports the source' do
        @yaml.should match(/source:\s*#{@source_name}/)
      end
      it 'reports the class' do
        @yaml.should match(/class:\s*SimulatedPolymorphism/)
      end
      it 'reports the subclass' do
        @yaml.should match(/subclass:\s*RepeatedConditional/)
      end
      it 'reports the expression' do
        @yaml.should match(/expression:\s*\(#{@cond}\)/)
      end
      it 'reports the number of occurrences' do
        @yaml.should match(/occurrences:\s*3/)
      end
      it 'reports the referring lines' do
        @yaml.should match(/lines:\s*- 4\s*- 7\s*- 12/)
      end
    end
  end

  context 'with a matching if and case' do
    before :each do
      cond = '@field == :sym'
      @cond_expr = cond.to_reek_source.syntax_tree
      src = <<EOS
class Scrunch
  def alpha
    return #{cond} ? 0 : 2;
  end
  def beta
    case #{cond}
    when :symbol
      @tother += " pints"
    end
  end
end
EOS

      ast = src.to_reek_source.syntax_tree
      ctx = CodeContext.new(nil, ast)
      @conds = @detector.conditional_counts(ctx)
    end
    it 'finds exactly one conditional' do
      @conds.length.should == 1
    end
    it 'returns the condition expr' do
      @conds.keys[0].should == @cond_expr
    end
    it 'knows there are three copies' do
      @conds[@cond_expr].length.should == 2
    end
  end

  # And count code in superclasses, if we have it
end