require 'test_helper'

class ConnectionTest < Test::Unit::TestCase
  
  def teardown
    @connection.close if @connection
  end

  def test_new_connection
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH)

    assert !@connection.parameters.empty?
    assert @connection.backend_pid
    assert @connection.backend_key
    assert @connection.transaction_status
    assert @connection.opened?
    assert !@connection.closed?
    
    # parameters
    assert @connection.parameters.kind_of?(Hash)
    assert @connection.parameters.include?('server_version')
  end
  
  def test_close_connection
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH)
    @connection.close

    assert !@connection.opened?
    assert @connection.closed?
    assert_equal({}, @connection.parameters)
    assert_nil @connection.backend_pid
    assert_nil @connection.backend_key
    assert_nil @connection.transaction_status
  end
  
  def test_connection_with_ssl
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH.merge(:ssl => true))

    assert @connection.ssl?
    assert !@connection.parameters.empty?
    assert @connection.backend_pid
    assert @connection.backend_key
    assert @connection.transaction_status

    @connection.close
    
    assert_equal({}, @connection.parameters)
    assert_nil @connection.backend_pid
    assert_nil @connection.backend_key
    assert_nil @connection.transaction_status
  end

  def test_reset_connection
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH)
    original_backend_pid = @connection.backend_pid
    original_backend_key = @connection.backend_key

    @connection.reset_connection

    assert_not_equal original_backend_pid, @connection.backend_pid
    assert_not_equal original_backend_key, @connection.backend_key
    assert_equal :no_transaction, @connection.transaction_status
  end
  
  def test_interrupt_connection
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH.merge(:interruptable => true))
    assert @connection.interruptable?, "The connection should be interruptable!"
  end

  def test_new_with_error_response
    assert_raises Vertica::Error::ConnectionError do
      Vertica::Connection.new(TEST_CONNECTION_HASH.merge('database' => 'nonexistant_db'))
    end
  end
  
  def test_connection_inspect_should_not_print_password
    @connection = Vertica::Connection.new(TEST_CONNECTION_HASH)
    inspected_string = @connection.inspect
    assert_no_match /:password=>#{TEST_CONNECTION_HASH[:password]}/, inspected_string
  end
end