#!/usr/local/bin/ruby
# Ruby Unit Tests

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

require 'DBD/Pg/Pg'

$last_suite = RUNIT::TestSuite.new

######################################################################
# Test the PostgreSql DBD driver.  This test exercises options
# difficult to test through the standard DBI interface.
#
class TestDbdPostgres < RUNIT::TestCase

  def test_connect
    dbd = get_dbd
    assert_not_nil dbd.connection
    assert_equal 'localhost', dbd.connection.host
    assert_equal 'jim', dbd.connection.user
    assert_equal 'rubytest', dbd.connection.db
    assert_equal 5432, dbd.connection.port
  ensure
     dbd.disconnect if dbd
  end

  def test_connect_errors
    dbd = nil
    ex = assert_exception(DBI::OperationalError) {
      dbd = DBI::DBD::Pg::Database.new('rubytest:1234', 'jim', nil, {})
    }
    ex = assert_exception(DBI::OperationalError) {
      dbd = DBI::DBD::Pg::Database.new('bad_db_name', 'jim', nil, {})
    }
  ensure
    dbd.disconnect if dbd
  end

  def test_type_map
    dbd = get_dbd
    def dbd.type_map
      @type_map
    end
    assert dbd.type_map
    assert_equal 21, dbd.type_map[23].call("21")
    assert_equal "21", dbd.type_map[1043].call("21")
    assert_equal 21.5, dbd.type_map[701].call("21.5")
  end

  def test_simple_command
    dbd = get_dbd
    dbd.do("INSERT INTO names (name, age) VALUES('Dan', 16)")
    res = dbd.do("SELECT name FROM names WHERE age=16")
    assert_equal 1, res
  ensure
    dbd.do("DELETE FROM names WHERE age < 20")
    dbd.disconnect if dbd
  end

  def test_bad_command
    dbd = get_dbd
    assert_exception (DBI::ProgrammingError) {
      dbd.do("INSERT INTO bad_table (name, age) VALUES('Dave', 12)")
    }
  ensure
    dbd.disconnect if dbd
  end

  def test_query_single
    dbd = get_dbd
    res = dbd.prepare("SELECT name, age FROM names WHERE age=21;")
    assert res
    res.execute
    fields = res.column_info
    assert_equal 2, fields.length
    assert_equal 'name', fields[0]['name']
    assert_equal 'age', fields[1]['name']

    row = res.fetch

    assert_equal 'Bob', row[0]
    assert_equal 21, row[1]

    row = res.fetch
    assert_nil row

    res.finish
  ensure
    dbd.disconnect if dbd
  end

  def test_query_multi
    dbd = get_dbd
    res = dbd.prepare("SELECT name, age FROM names WHERE age > 20;")

    expected_list = ['Bob', 'Charlie']
    res.execute
    while row=res.fetch
      expected = expected_list.shift
      assert_equal expected, row[0]
    end

    res.finish
  ensure
    dbd.disconnect if dbd
  end

  private # ----------------------------------------------------------

  def get_dbd
    result = DBI::DBD::Pg::Database.new('rubytest', 'jim', nil, {})
    result['AutoCommit'] = true
    result
  end
end

$last_suite.add_test(TestDbdPostgres.suite)


# ====================================================================
class TestDbiPostgresWrapper < RUNIT::TestCase

  def initialize(suite)
    @test_suite = suite
  end

  def setup
    system "testsetup.sh rubytest jim '' names >sql.log 2>&1"
  end

  def teardown
    system "testteardown.sh rubytest jim '' names >>sql.log 2>&1"
  end

  def run(test_result)
    setup
    @test_suite.run(test_result)
    teardown
  end
end

$last_suite = TestDbiPostgresWrapper.new($last_suite)


# --------------------------------------------------------------------

if __FILE__ == $0 then
  RUNIT::CUI::TestRunner.quiet_mode = true
  RUNIT::CUI::TestRunner.run($last_suite)
end