require 'runit/testcase'
require 'file-win32'
require 'excel'

class XlsJoinTest < RUNIT::TestCase
  def setup
    @xls_file = "xlsjoin-test.xls"
    @excel = Excel.new(false)
  end
  
  def teardown
    File.delete(@xls_file) if File.exist?(@xls_file)
    @excel.quit
  end

  def open_xls(filename)
    path = FileSystemObject.instance.getAbsolutePathName(filename)
    @excel.workbooks.open({'Filename'=>path, 'ReadOnly'=>true})
  end

  def test_ab
    system <<XXX
ruby xlsjoin.rb -o #{@xls_file} xlsjoin-test-a.xls xlsjoin-test-b.xls
XXX
    worksheets = open_xls(@xls_file).worksheets
    assert_equals(3, worksheets.count)
    assert_equals('a-1', worksheets.item(1).name)
    assert_equals('a-1', worksheets.item(1).cells(1,1).value)
    assert_equals('a-2', worksheets.item(2).name)
    assert_equals('a-2', worksheets.item(2).cells(1,1).value)
    assert_equals('b-1', worksheets.item(3).name)
    assert_equals('b-1', worksheets.item(3).cells(1,1).value)
  end

  def test_a
    system <<XXX
ruby xlsjoin.rb -o #{@xls_file} xlsjoin-test-a.xls
XXX
    worksheets = open_xls(@xls_file).worksheets
    assert_equals(2, worksheets.count)
    assert_equals('a-1', worksheets.item(1).name)
    assert_equals('a-1', worksheets.item(1).cells(1,1).value)
    assert_equals('a-2', worksheets.item(2).name)
    assert_equals('a-2', worksheets.item(2).cells(1,1).value)
  end

  def test_empty
    system <<XXX
ruby xlsjoin.rb -o #{@xls_file}
XXX
    worksheets = open_xls(@xls_file).worksheets
    assert_equals(1, worksheets.count)
    assert_equals('Sheet1', worksheets.item(1).name)
    assert_nil(worksheets.item(1).cells(1,1).value)
  end

end

if __FILE__ == $0
  require 'runit/cui/testrunner'
  testrunner = RUNIT::CUI::TestRunner.new
  if ARGV.size == 0
    suite = XlsJoinTest.suite
  else
    suite = RUNIT::TestSuite.new
    ARGV.each do |testmethod|
      suite.add_test(XlsJoinTest.new(testmethod))
    end
  end
  testrunner.run(suite)
end