Contents

require 'helper'
require 'parser/ruby18'

class TestSourceCommentAssociator < Minitest::Test
  def associate(code)
    parser = Parser::Ruby18.new

    buffer = Parser::Source::Buffer.new('(comments)')
    buffer.source = code

    ast, comments = parser.parse_with_comments(buffer)
    associations  = Parser::Source::Comment.associate(ast, comments)

    [ ast, associations ]
  end

  def test_associate
    ast, associations = associate(<<-END)
#!/usr/bin/env ruby
# coding: utf-8
# Class comment
# another class comment
class Foo
  # attr_accessor comment
  attr_accessor :foo

  # method comment
  def bar
    # expr comment
    1 + # intermediate comment
      2
    # stray comment
  end
end
    END

    klass_node         = ast
    attr_accessor_node = ast.children[2].children[0]
    method_node        = ast.children[2].children[1]
    expr_node          = method_node.children[2]
    intermediate_node  = expr_node.children[2]

    assert_equal 5, associations.size
    assert_equal ['# Class comment', '# another class comment'],
                 associations[klass_node].map(&:text)
    assert_equal ['# attr_accessor comment'],
                 associations[attr_accessor_node].map(&:text)
    assert_equal ['# method comment'],
                 associations[method_node].map(&:text)
    assert_equal ['# expr comment'],
                 associations[expr_node].map(&:text)
    assert_equal ['# intermediate comment'],
                 associations[intermediate_node].map(&:text)
  end

  def test_associate_no_body
    ast, associations = associate(<<-END)
# foo
class Foo
end
    END

    assert_equal 1, associations.size
    assert_equal ['# foo'],
                 associations[ast].map(&:text)
  end

  def test_associate_shebang_only
    ast, associations = associate(<<-END)
#!ruby
class Foo
end
    END

    assert_equal 0, associations.size
  end

  def test_associate_no_comments
    ast, associations = associate(<<-END)
class Foo
end
    END

    assert_equal 0, associations.size
  end
end

Version data entries

9 entries across 9 versions & 1 rubygems

Version	Path
parser-2.1.2	test/test_source_comment_associator.rb
parser-2.1.1	test/test_source_comment_associator.rb
parser-2.1.0	test/test_source_comment_associator.rb
parser-2.1.0.pre2	test/test_source_comment_associator.rb
parser-2.1.0.pre1	test/test_source_comment_associator.rb
parser-2.0.0	test/test_source_comment_associator.rb
parser-2.0.0.pre8	test/test_source_comment_associator.rb
parser-2.0.0.pre7	test/test_source_comment_associator.rb
parser-2.0.0.pre6	test/test_source_comment_associator.rb