require 'runit/testcase'
require 'runit/cui/testrunner'

require 'rrb/parser.rb'



class TestParser < RUNIT::TestCase

  TEST_SCRIPT_NAME = 'samples/parser_sample.rb'
  
  def test_parse
    parser = RRB::Parser.new
    parsed_info = parser.run File.open( TEST_SCRIPT_NAME, "r" )

    class_info = parsed_info.class_info("TestClassA")

    # test method_calls
    method_1_info = class_info.method_info("method_1")
    assert_equals( %w(each) ,
		  method_1_info.method_calls.map{|info| info.name} )
    assert_equals( %w(call print),
		  method_1_info.fcalls.map{|info| info.name} )
    
    method_2_info = class_info.method_info("method_2")
    assert_equals( [],
		  method_2_info.method_calls.map{|info| info.name} )
    assert_equals( %w(call! call p),
		  method_2_info.fcalls.map{|info| info.name} )

    method_3_info = class_info.method_info("method_3")
    assert_equals( [],
		  method_3_info.method_calls.map{|info| info.name} )
    assert_equals( %w(ConstNameCall),
		  method_3_info.fcalls.map{|info| info.name} )

    # test local_vars
    assert_equals( %w(a b c a x x b c),
		  method_1_info.local_vars.map{|info| info.name} )
    assert_equals( %w(c rest block call call),
		  method_2_info.local_vars.map{|info| info.name} )

    # test singleton method
    method_5_info = class_info.class_method_defs.find{|x| x.name == "method_5"}
    assert_equals( %w(x y x y),
		  method_5_info.local_vars.map{|info| info.name} )

    method_6_info = class_info.singleton_method_defs.find{|x|
      x.name == "method_6"
    }
    assert_not_nil( method_6_info )

    # test singleton class
    method_7_info = class_info.singleton_class_defs[0].method_info("method_7")
    assert_not_nil( method_7_info )

    # test global and instance variable
    method_8_info = class_info.method_info("method_8")
    assert_equals( [], method_8_info.local_vars )
    assert_equals( %w($x $x $xx),
		  method_8_info.global_vars.map{|info| info.name} )
    assert_equals( %w(@y @y),
		  method_8_info.instance_vars.map{|info| info.name} )
    assert_equals( %w(@@z),
		  method_8_info.class_vars.map{|info| info.name} )

    # test constant
    method_9_info = class_info.method_info("method_9")
    assert_equals( [ "Const1",
		    "Const2", "Const2::Const1",
		    "Const2", "Const2::Const1",
		    "Const4", "Const4::Const5",
		    "Const6", "Const6::Const7",
		    "Const6::Const7::Const8", "Const6::Const7::Const8::Const9",
		    "Const10", "Const10::Const11",
		    "Const12"
                   ],
                   method_9_info.consts.map{|info| info.name} )
    
    assert_equals( %w(TestClassA TestClassB TestClassB Const6), class_info.consts.map{|info| info.name } )

    
    # test class method
    assert_equals( [ "method_5", "method_10" ],
                   class_info.class_method_defs.map{|cm| cm.name} )
                   
    # test head_keyword and tail_keyword
    assert_equals( 5, method_1_info.head_keyword.pointer )
    assert_equals( 7, method_1_info.head_keyword.lineno )
    assert_equals( "def", method_1_info.head_keyword.name )
    assert_equals( 12 , method_1_info.tail_keyword.lineno )
    assert_equals( "end", method_1_info.tail_keyword.name )
    assert_equals( "class", class_info.head_keyword.name )

    # test assigned
    assert_equals( [], method_3_info.assigned )
    assert_equals( [ 'c', 'x' ], method_1_info.assigned.map{|info| info.name})

    # test superclass
    assert_equals( 'TestClassA::TestClassB',
		  parsed_info.class_info("TestClassD").superclass.name )
    assert_nil( class_info.superclass )

    # test attr_*
    assert_equals( [], class_info.attr_readers )
    assert_equals( %w(y z), class_info.attr_writers.map{|i| i.name} )
    assert_equals( %w(a b c d), class_info.attr_accessors.map{|i| i.name} )
  end
  
end

if $0 == __FILE__
  if ARGV.size == 0
    suite = TestParser.suite
  else
    suite = RUNIT::TestSuite.new
    ARGV.each do |testmethod|
      suite.add_test(TestParser.new(testmethod))
    end
  end
  RUNIT::CUI::TestRunner.run(suite)
end