require_relative '../../spec_helper'
require_relative '../../../lib/reek/core/method_context'
require_relative '../../../lib/reek/core/stop_context'

describe Reek::Core::MethodContext, 'matching' do
  before :each do
    exp = double('exp').as_null_object
    expect(exp).to receive(:full_name).at_least(:once).and_return('mod')
    @element = Reek::Core::MethodContext.new(Reek::Core::StopContext.new, exp)
  end

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

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

describe Reek::Core::MethodContext do
  it 'should record ivars as refs to self' do
    sexp = s(:def, :feed, s(:args), nil)
    mctx = Reek::Core::MethodContext.new(Reek::Core::StopContext.new, sexp)
    expect(mctx.envious_receivers).to eq([])
    mctx.record_call_to(s(:send, s(:ivar, :@cow), :feed_to))
    expect(mctx.envious_receivers).to eq([])
  end

  it 'should count calls to self' do
    sexp = s(:def, :equals, s(:args), nil)
    mctx = Reek::Core::MethodContext.new(Reek::Core::StopContext.new, sexp)
    mctx.refs.record_reference_to([:lvar, :other])
    mctx.record_call_to(s(:send, s(:self), :thing))
    expect(mctx.envious_receivers).to be_empty
  end

  it 'should recognise a call on self' do
    sexp = s(:def, :deep, s(:args), nil)
    mc = Reek::Core::MethodContext.new(Reek::Core::StopContext.new, sexp)
    mc.record_call_to(s(:send, s(:lvar, :text), :each, s(:arglist)))
    mc.record_call_to(s(:send, nil, :shelve, s(:arglist)))
    expect(mc.envious_receivers).to be_empty
  end
end

describe Reek::Core::MethodParameters, 'default assignments' do
  def assignments_from(src)
    exp = Reek::Source::SourceCode.from(src).syntax_tree
    ctx = Reek::Core::MethodContext.new(Reek::Core::StopContext.new, exp)
    ctx.parameters.default_assignments
  end

  context 'with no defaults' do
    it 'returns an empty hash' do
      src = 'def meth(arga, argb, &blk) end'
      expect(assignments_from(src)).to 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
      expect(@defaults[0]).to eq [:argb, s(:int, 456)]
    end
    it 'returns the nothing else' do
      expect(@defaults.length).to eq(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
      expect(@defaults[0]).to eq [:arga, s(:int, 123)]
      expect(@defaults[1]).to eq [:argb, s(:int, 456)]
    end
    it 'returns nothing else' do
      expect(@defaults.length).to eq(2)
    end
  end
end