# encoding: utf-8

require 'spec_helper'

describe RuboCop::Cop::Style::SingleLineMethods, :config do
  subject(:cop) { described_class.new(config) }
  let(:cop_config) { { 'AllowIfMethodIsEmpty' => true } }

  it 'registers an offense for a single-line method' do
    inspect_source(cop,
                   ['def some_method; body end',
                    'def link_to(name, url); {:name => name}; end',
                    'def @table.columns; super; end'])
    expect(cop.messages).to eq(
      ['Avoid single-line method definitions.'] * 3)
  end

  context 'when AllowIfMethodIsEmpty is disabled' do
    let(:cop_config) { { 'AllowIfMethodIsEmpty' => false } }

    it 'registers an offense for an empty method' do
      inspect_source(cop, ['def no_op; end',
                           'def self.resource_class=(klass); end',
                           'def @table.columns; end'])
      expect(cop.offenses.size).to eq(3)
    end
  end

  context 'when AllowIfMethodIsEmpty is enabled' do
    let(:cop_config) { { 'AllowIfMethodIsEmpty' => true } }

    it 'accepts a single-line empty method' do
      inspect_source(cop, ['def no_op; end',
                           'def self.resource_class=(klass); end',
                           'def @table.columns; end'])
      expect(cop.offenses).to be_empty
    end
  end

  it 'accepts a multi-line method' do
    inspect_source(cop, ['def some_method',
                         '  body',
                         'end'])
    expect(cop.offenses).to be_empty
  end

  it 'does not crash on an method with a capitalized name' do
    inspect_source(cop, ['def NoSnakeCase',
                         'end'])
    expect(cop.offenses).to be_empty
  end

  it 'auto-corrects def with semicolon after method name' do
    corrected = autocorrect_source(cop,
                                   ['  def some_method; body end # Cmnt'])
    expect(corrected).to eq ['  # Cmnt',
                             '  def some_method; ',
                             '    body ',
                             '  end '].join("\n")
  end

  it 'auto-corrects defs with parentheses after method name' do
    corrected = autocorrect_source(cop, ['  def self.some_method() body end'])
    expect(corrected).to eq ['  def self.some_method() ',
                             '    body ',
                             '  end'].join("\n")
  end

  it 'auto-corrects def with argument in parentheses' do
    corrected = autocorrect_source(cop, ['  def some_method(arg) body end'])
    expect(corrected).to eq ['  def some_method(arg) ',
                             '    body ',
                             '  end'].join("\n")
  end

  it 'auto-corrects def with argument and no parentheses' do
    corrected = autocorrect_source(cop, ['  def some_method arg; body end'])
    expect(corrected).to eq ['  def some_method arg; ',
                             '    body ',
                             '  end'].join("\n")
  end

  it 'auto-corrects def with semicolon before end' do
    corrected = autocorrect_source(cop, ['  def some_method; b1; b2; end'])
    expect(corrected).to eq ['  def some_method; ',
                             '    b1; ',
                             '    b2; ',
                             '  end'].join("\n")
  end
end