sqlpad/drivers/vertica/test.js

Web app. Write SQL and visualize the results. Supports Postgres, MySQL, SQL Server, Crate, Vertica and SAP HANA.

const assert = require('assert');
const vertica = require('./index.js');

const connection = {
  name: 'test vertica',
  driver: 'vertica',
  host: 'localhost',
  username: 'dbadmin',
  maxRows: 50000
};

const initSql = `
  DROP TABLE IF EXISTS test;

  CREATE TABLE test (id int);
  COMMIT;

  -- In vertica you must commit after insert
  INSERT INTO test (id) VALUES (1); 
  INSERT INTO test (id) VALUES (2);
  INSERT INTO test (id) VALUES (3);
  COMMIT;
`;

describe('drivers/vertica', function() {
  before(function() {
    this.timeout(10000);
    return vertica.runQuery(initSql, connection);
  });

  it('tests connection', function() {
    return vertica.testConnection(connection);
  });

  it('getSchema()', function() {
    return vertica.getSchema(connection).then(schemaInfo => {
      assert(schemaInfo.public, 'public');
      assert(schemaInfo.public.test, 'public.test');
      const columns = schemaInfo.public.test;
      assert.equal(columns.length, 1, 'columns.length');
      assert.equal(columns[0].table_schema, 'public', 'table_schema');
      assert.equal(columns[0].table_name, 'test', 'table_name');
      assert.equal(columns[0].column_name, 'id', 'column_name');
      assert(columns[0].hasOwnProperty('data_type'), 'data_type');
    });
  });

  it('runQuery under limit', function() {
    return vertica
      .runQuery('SELECT id FROM test WHERE id = 1;', connection)
      .then(results => {
        assert(!results.incomplete, 'not incomplete');
        assert.equal(results.rows.length, 1, 'rows length');
      });
  });

  it('runQuery over limit', function() {
    const limitedConnection = Object.assign({}, connection, { maxRows: 2 });
    return vertica
      .runQuery('SELECT * FROM test;', limitedConnection)
      .then(results => {
        assert(results.incomplete, 'incomplete');
        assert.equal(results.rows.length, 2, 'row length');
      });
  });

  it('returns descriptive error message', function() {
    let error;
    return vertica
      .runQuery('SELECT * FROM missing_table;', connection)
      .catch(e => {
        error = e;
      })
      .then(() => {
        assert(error);
        assert(error.toString().indexOf('missing_table') > -1);
      });
  });
});