require "../../spec_helper"

module Ameba::Formatter
  class Subject
    include Util
  end

  subject = Subject.new

  describe Util do
    describe "#affected_code" do
      it "returns nil if there is no such a line number" do
        source = Source.new %(
          a = 1
        )
        location = Crystal::Location.new("filename", 2, 1)
        subject.affected_code(source, location).should be_nil
      end

      it "returns correct line if it is found" do
        source = Source.new %(
          a = 1
        )
        location = Crystal::Location.new("filename", 1, 1)
        subject.deansify(subject.affected_code(source, location))
          .should eq "> a = 1\n  ^"
      end

      it "returns correct line if it is found" do
        source = Source.new <<-EOF
          # pre:1
            # pre:2
              # pre:3
                # pre:4
                  # pre:5
          a = 1
                  # post:1
                # post:2
              # post:3
            # post:4
          # post:5
          EOF

        location = Crystal::Location.new("filename", 6, 1)
        subject.deansify(subject.affected_code(source, location, context_lines: 3))
          .should eq <<-STR
            >     # pre:3
            >       # pre:4
            >         # pre:5
            > a = 1
              ^
            >         # post:1
            >       # post:2
            >     # post:3

            STR
      end
    end
  end
end