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', 'core', 'method_context')
require File.join(File.dirname(File.dirname(File.dirname(File.dirname(File.expand_path(__FILE__))))), 'lib', 'reek', 'core', 'stop_context')

include Reek::Core

describe MethodContext, 'matching' do
  before :each do
    @element = MethodContext.new(StopContext.new, s(0, :mod))
  end

  it 'should recognise itself in a collection of names' do
    @element.matches?(['banana', 'mod']).should == true
    @element.matches?(['banana']).should == false
  end

  it 'should recognise itself in a collection of REs' do
    @element.matches?([/banana/, /mod/]).should == true
    @element.matches?([/banana/]).should == false
  end
end

describe MethodContext, 'matching fq names' do
  before :each do
    element = StopContext.new
    element = ModuleContext.new(element, 'mod', s(:module, :mod, nil))
    element = ClassContext.new(element, 'klass', s())
    @element = MethodContext.new(element, s(0, :meth))
  end

  it 'should recognise itself in a collection of names' do
    @element.matches?(['banana', 'meth']).should == true
    @element.matches?(['banana', 'klass#meth']).should == true
    @element.matches?(['banana']).should == false
  end

  it 'should recognise itself in a collection of names' do
    @element.matches?([/banana/, /meth/]).should == true
    @element.matches?([/banana/, /klass#meth/]).should == true
    @element.matches?([/banana/]).should == false
  end
end

describe MethodContext do
  it 'should record ivars as refs to self' do
    mctx = MethodContext.new(StopContext.new, [:defn, :feed])
    mctx.envious_receivers.should == []
    mctx.record_call_to([:call, [:ivar, :@cow], :feed_to])
    mctx.envious_receivers.should == []
  end

  it 'should count calls to self' do
    mctx = MethodContext.new(StopContext.new, [:defn, :equals])
    mctx.refs.record_ref([:lvar, :other])
    mctx.record_call_to([:call, [:self], :thing])
    mctx.envious_receivers.should be_empty
  end

  it 'should recognise a call on self' do
    mc = MethodContext.new(StopContext.new, s(:defn, :deep))
    mc.record_call_to(s(:call, s(:lvar, :text), :each, s(:arglist)))
    mc.record_call_to(s(:call, nil, :shelve, s(:arglist)))
    mc.envious_receivers.should be_empty
  end
end

describe MethodParameters, 'default assignments' do
  def assignments_from(src)
    exp = src.to_reek_source.syntax_tree
    ctx = MethodContext.new(StopContext.new, exp)
    return ctx.parameters.default_assignments
  end

  context 'with no defaults' do
    it 'returns an empty hash' do
      src = 'def meth(arga, argb, &blk) end'
      assignments_from(src).should be_empty
    end
  end

  context 'with 1 default' do
    before :each do
      src = "def meth(arga, argb=456, &blk) end"
      @defaults = assignments_from(src)
    end
    it 'returns the param-value pair' do
      @defaults[:argb].should == s(:lit, 456)
    end
    it 'returns the nothing else' do
      @defaults.length.should == 1
    end
  end

  context 'with 2 defaults' do
    before :each do
      src = "def meth(arga=123, argb=456, &blk) end"
      @defaults = assignments_from(src)
    end
    it 'returns both param-value pairs' do
      @defaults[:arga].should == s(:lit, 123)
      @defaults[:argb].should == s(:lit, 456)
    end
    it 'returns nothing else' do
      @defaults.length.should == 2
    end
  end
end