in_pattern_node_spec.rb

# frozen_string_literal: true

RSpec.describe RuboCop::AST::InPatternNode do
  context 'when using Ruby 2.7 or newer', :ruby27 do
    let(:in_pattern_node) { parse_source(source).ast.children[1] }

    describe '.new' do
      let(:source) do
        ['case condition',
         'in [42] then foo',
         'end'].join("\n")
      end

      it { expect(in_pattern_node).to be_a(described_class) }
    end

    describe '#pattern' do
      context 'with a value pattern' do
        let(:source) do
          ['case condition',
           'in 42 then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_int_type }
      end

      context 'with a variable pattern' do
        let(:source) do
          ['case condition',
           'in var then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_match_var_type }
      end

      context 'with an alternative pattern' do
        let(:source) do
          ['case condition',
           'in :foo | :bar | :baz then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_match_alt_type }
      end

      context 'with an as pattern' do
        let(:source) do
          ['case condition',
           'in Integer => var then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_match_as_type }
      end

      context 'with an array pattern' do
        let(:source) do
          ['case condition',
           'in :foo, :bar, :baz then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_array_pattern_type }
      end

      context 'with a hash pattern' do
        let(:source) do
          ['case condition',
           'in foo:, bar:, baz: then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_hash_pattern_type }
      end

      context 'with a pin operator', :ruby31 do
        let(:source) do
          ['case condition',
           'in ^(2 + 2) then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.pattern).to be_pin_type }
      end
    end

    describe '#then?' do
      context 'with a then keyword' do
        let(:source) do
          ['case condition',
           'in [42] then foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node).to be_then }
      end

      context 'without a then keyword' do
        let(:source) do
          ['case condition',
           'in [42]',
           '  foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node).not_to be_then }
      end
    end

    describe '#body' do
      context 'with a then keyword' do
        let(:source) do
          ['case condition',
           'in [42] then :foo',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.body).to be_sym_type }
      end

      context 'without a then keyword' do
        let(:source) do
          ['case condition',
           'in [42]',
           '  [:foo, :bar]',
           'end'].join("\n")
        end

        it { expect(in_pattern_node.body).to be_array_type }
      end
    end

    describe '#branch_index' do
      let(:source) do
        ['case condition',
         'in [42] then 1',
         'in [43] then 2',
         'in [44] then 3',
         'end'].join("\n")
      end

      let(:in_patterns) { parse_source(source).ast.children[1...-1] }

      it { expect(in_patterns[0].branch_index).to eq(0) }
      it { expect(in_patterns[1].branch_index).to eq(1) }
      it { expect(in_patterns[2].branch_index).to eq(2) }
    end
  end
end