pry--pry/test/test_code.rb

require 'helper'

describe Pry::Code do
  describe '.from_file' do
    should 'read lines from a file on disk' do
      Pry::Code.from_file('lib/pry.rb').length.should > 0
    end

    should 'read lines from Pry\'s line buffer' do
      mock_pry(':hay_guys')
      Pry::Code.from_file('(pry)').grep(/:hay_guys/).length.should == 1
    end

    should 'default to Ruby' do
      temp_file('') do |f|
        Pry::Code.from_file(f.path).code_type.should == :ruby
      end
    end

    should 'check the extension' do
      temp_file('.c') do |f|
        Pry::Code.from_file(f.path).code_type.should == :c
      end
    end

    should 'use the provided extension' do
      temp_file('.c') do |f|
        Pry::Code.from_file(f.path, :ruby).code_type.should == :ruby
      end
    end

    should 'raise an error if the file doesn\'t exist' do
      proc do
        Pry::Code.from_file('/knalkjsdnalsd/alkjdlkq')
      end.should.raise(MethodSource::SourceNotFoundError)
    end
  end

  describe '.from_method' do
    should 'read lines from a method\'s definition' do
      m = Pry::Method.from_obj(Pry, :load_history)
      Pry::Code.from_method(m).length.should > 0
    end
  end

  describe '#initialize' do
    before do
      @str = Pry::Helpers::CommandHelpers.unindent <<-CODE
        def hay
          :guys
        end
      CODE

      @array = ['def hay', '  :guys', 'end']
    end

    should 'break a string into lines' do
      Pry::Code.new(@str).length.should == 3
    end

    should 'accept an array' do
      Pry::Code.new(@array).length.should == 3
    end

    it 'an array or string should produce an equivalent object' do
      Pry::Code.new(@str).should == Pry::Code.new(@array)
    end
  end

  describe 'filters and formatters' do
    before do
      @code = Pry::Code(Pry::Helpers::CommandHelpers.unindent <<-STR)
        class MyProgram
          def self.main
            puts 'Hello, world!'
          end
        end
      STR
    end

    describe 'filters' do
      describe '#between' do
        should 'work with an inclusive range' do
          @code = @code.between(1..3)
          @code.length.should == 3
          @code.should =~ /\Aclass MyProgram/
          @code.should =~ /world!'\Z/
        end

        should 'default to an inclusive range' do
          @code = @code.between(3, 5)
          @code.length.should == 3
        end

        should 'work with an exclusive range' do
          @code = @code.between(2...4)
          @code.length.should == 2
          @code.should =~ /\A  def self/
          @code.should =~ /world!'\Z/
        end

        should 'use real line numbers for positive indices' do
          @code = @code.after(3, 3)
          @code = @code.between(4, 4)
          @code.length.should == 1
          @code.should =~ /\A  end\Z/
        end
      end

      describe '#before' do
        should 'work' do
          @code = @code.before(3, 1)
          @code.should =~ /\A  def self\.main\Z/
        end
      end

      describe '#around' do
        should 'work' do
          @code = @code.around(3, 1)
          @code.length.should == 3
          @code.should =~ /\A  def self/
          @code.should =~ /  end\Z/
        end
      end

      describe '#after' do
        should 'work' do
          @code = @code.after(3, 1)
          @code.should =~ /\A  end\Z/
        end
      end

      describe '#grep' do
        should 'work' do
          @code = @code.grep(/end/)
          @code.length.should == 2
        end
      end
    end

    describe 'formatters' do
      describe '#with_line_numbers' do
        should 'show line numbers' do
          @code = @code.with_line_numbers
          @code.should =~ /1:/
        end

        should 'disable line numbers when falsy' do
          @code = @code.with_line_numbers
          @code = @code.with_line_numbers(false)
          @code.should.not =~ /1:/
        end
      end

      describe '#with_marker' do
        should 'show a marker in the right place' do
          @code = @code.with_marker(2)
          @code.should =~ /^ =>   def self/
        end

        should 'disable the marker when falsy' do
          @code = @code.with_marker(2)
          @code = @code.with_marker(false)
          @code.should =~ /^  def self/
        end
      end

      describe '#with_indentation' do
        should 'indent the text' do
          @code = @code.with_indentation(2)
          @code.should =~ /^    def self/
        end

        should 'disable the indentation when falsy' do
          @code = @code.with_indentation(2)
          @code = @code.with_indentation(false)
          @code.should =~ /^  def self/
        end
      end
    end

    describe 'composition' do
      describe 'grep and with_line_numbers' do
        should 'work' do
          @code = @code.grep(/end/).with_line_numbers
          @code.should =~ /\A4:   end/
          @code.should =~ /5: end\Z/
        end
      end

      describe 'grep and before and with_line_numbers' do
        should 'work' do
          @code = @code.grep(/e/).before(5, 5).with_line_numbers
          @code.should =~ /\A2:   def self.main\n3:/
          @code.should =~ /4:   end\Z/
        end
      end

      describe 'before and after' do
        should 'work' do
          @code = @code.before(4, 2).after(2)
          @code.should == "    puts 'Hello, world!'"
        end
      end
    end
  end
end