# encoding: utf-8

require 'spec_helper'

describe Rubocop::Cop::Style::MultilineIfThen do
  subject(:cop) { described_class.new }

  # if

  it 'registers an offense for then in multiline if' do
    inspect_source(cop, ['if cond then',
                         'end',
                         "if cond then\t",
                         'end',
                         'if cond then  ',
                         'end',
                         'if cond',
                         'then',
                         'end',
                         'if cond then # bad',
                         'end'])
    expect(cop.offenses.map(&:line)).to eq([1, 3, 5, 7, 10])
  end

  it 'accepts multiline if without then' do
    inspect_source(cop, ['if cond',
                         'end'])
    expect(cop.offenses).to be_empty
  end

  it 'accepts table style if/then/elsif/ends' do
    inspect_source(cop,
                   ['if    @io == $stdout then str << "$stdout"',
                    'elsif @io == $stdin  then str << "$stdin"',
                    'elsif @io == $stderr then str << "$stderr"',
                    'else                      str << @io.class.to_s',
                    'end'])
    expect(cop.offenses).to be_empty
  end

  it 'does not get confused by a then in a when' do
    inspect_source(cop,
                   ['if a',
                    '  case b',
                    '  when c then',
                    '  end',
                    'end'])
    expect(cop.offenses).to be_empty
  end

  it 'does not get confused by a commented-out then' do
    inspect_source(cop,
                   ['if a # then',
                    '  b',
                    'end',
                    'if c # then',
                    'end'])
    expect(cop.offenses).to be_empty
  end

  it 'does not raise an error for an implicit match if' do
    expect do
      inspect_source(cop,
                     ['if //',
                      'end'])
    end.not_to raise_error
  end

  # unless

  it 'registers an offense for then in multiline unless' do
    inspect_source(cop, ['unless cond then',
                         'end'])
    expect(cop.messages).to eq(
      ['Never use then for multi-line unless.'])
  end

  it 'accepts multiline unless without then' do
    inspect_source(cop, ['unless cond',
                         'end'])
    expect(cop.offenses).to be_empty
  end

  it 'does not get confused by a postfix unless' do
    inspect_source(cop,
                   ['two unless one'
                   ])
    expect(cop.offenses).to be_empty
  end

  it 'does not get confused by a nested postfix unless' do
    inspect_source(cop,
                   ['if two',
                    '  puts 1',
                    'end unless two'
                   ])
    expect(cop.offenses).to be_empty
  end

  it 'does not raise an error for an implicit match unless' do
    expect do
      inspect_source(cop,
                     ['unless //',
                      'end'])
    end.not_to raise_error
  end
end