# encoding: utf-8
require File.expand_path(File.dirname(__FILE__) + '/../../spec_helper')
require 'gherkin/formatter/pretty_formatter'
require 'gherkin/formatter/argument'
require 'stringio'

module Gherkin
  module Formatter
    describe PrettyFormatter do
      def assert_io(s)
        @io.rewind
        actual = @io.read
        actual.should == s
      end
      
      def assert_pretty(input, output=input)
        [true, false].each do |force_ruby|
          io = StringIO.new
          l = PrettyFormatter.new(io, true)
          parser = Gherkin::Parser::Parser.new(l, true, "root")
          lexer  = Gherkin::I18nLexer.new(parser, force_ruby)
          lexer.scan(input)
          io.rewind
          actual = io.read
          actual.should == output
        end
      end

      before do
        @io = StringIO.new
        @l = PrettyFormatter.new(@io, true)
      end

      it "should print comments when scenario is longer" do
        @l.feature("Feature", "Hello\nWorld", 1)
        @l.steps([
          ['Given ', 'some stuff'],
          ['When ', 'foo']
        ])
        @l.scenario("Scenario", "The scenario", 4, "features/foo.feature:4")
        @l.step("Given ", "some stuff", 5, nil, nil, nil, "features/step_definitions/bar.rb:56")
        @l.step("When ", "foo", 6, nil, nil, nil, "features/step_definitions/bar.rb:96")

        assert_io(%{Feature: Hello
  World

  Scenario: The scenario # features/foo.feature:4
    Given some stuff     # features/step_definitions/bar.rb:56
    When foo             # features/step_definitions/bar.rb:96
})
      end

      it "should print comments when step is longer" do
        @l.feature("Feature", "Hello\nWorld", 1)
        @l.steps([
          ['Given ', 'some stuff that is longer']
        ])
        @l.scenario("Scenario", "The scenario", 4, "features/foo.feature:4")
        @l.step("Given ", "some stuff that is longer", 5, nil, nil, nil, "features/step_definitions/bar.rb:56")

        assert_io(%{Feature: Hello
  World

  Scenario: The scenario            # features/foo.feature:4
    Given some stuff that is longer # features/step_definitions/bar.rb:56
})
      end

      it "should print ANSI coloured steps" do
        @l.feature("Feature", "Hello\nWorld", 1)
        @l.steps([
          ['Given ', 'some stuff that is longer']
        ])
        @l.scenario("Scenario", "The scenario", 4, "features/foo.feature:4")
        @l.step("Given ", "some stuff that is longer", 5, nil, nil, nil, "features/step_definitions/bar.rb:56")

        assert_io(%{Feature: Hello
  World

  Scenario: The scenario            # features/foo.feature:4
    Given some stuff that is longer # features/step_definitions/bar.rb:56
})
      end

      it "should prettify a whole table with padding (typically ANSI)" do
        @l.row(%w(a bb), 1)
        @l.row(%w(ccc d), 2)
        @l.row(%w(ee ffff), 3)
        @l.flush_table

        assert_io(
          "      | a   | bb   |\n" +
          "      | ccc | d    |\n" +
          "      | ee  | ffff |\n"
        )
      end

      it "should highlight arguments for regular steps" do
        passed = defined?(JRUBY_VERSION) ? 'passed' : :passed
        @l.step("Given ", "I have 999 cukes in my belly", 3, passed, nil, [Gherkin::Formatter::Argument.new(7, '999')], nil)
        assert_io("    Given I have 999 cukes in my belly\n")
      end

      it "should prettify scenario" do
        assert_pretty(%{Feature: Feature Description
  Some preamble

  Scenario: Scenario Description
    Given there is a step
      """
      with
        pystrings
      """
    And there is another step
      | æ | o |
      | a | ø |
    Then we will see steps
})
      end


      it "should prettify scenario outline with table" do
        assert_pretty(%{# A feature comment
@foo
Feature: Feature Description
  Some preamble

  # A Scenario Outline comment
  @bar
  Scenario Outline: Scenario Ouline Description
    Given there is a
      """
      string with <foo>
      """
    And a table with
      | <bar> |
      | <baz> |

    @zap @boing
    Examples: Examples Description
      | foo    | bar  | baz         |
      | Banana | I    | am hungry   |
      | Beer   | You  | are thirsty |
      | Bed    | They | are tired   |
})
      end

      it "should preserve tabs" do
        assert_pretty(IO.read(File.dirname(__FILE__) + '/tabs.feature'), IO.read(File.dirname(__FILE__) + '/spaces.feature'))
      end
    end
  end
end