postgresql.discover.test.js

// Copyright IBM Corp. 2013,2020. All Rights Reserved.
// Node module: loopback-connector-postgresql
// This file is licensed under the Artistic License 2.0.
// License text available at https://opensource.org/licenses/Artistic-2.0

'use strict';
process.env.NODE_ENV = 'test';
require('should');

const assert = require('assert');
const should = require('should');
const _ = require('lodash');

const DataSource = require('loopback-datasource-juggler').DataSource;
let db, City;

before(function() {
  const config = global.getDBConfig();
  config.database = 'strongloop';
  db = new DataSource(require('../'), config);
});

describe('discoverModels', function() {
  describe('Discover database schemas', function() {
    it('should return an array of db schemas', function(done) {
      db.connector.discoverDatabaseSchemas(function(err, schemas) {
        if (err) return done(err);
        schemas.should.be.an.array;
        schemas.length.should.be.above(0);
        done();
      });
    });
  });

  describe('Discover models including views', function() {
    it('should return an array of tables and views', function(done) {
      db.discoverModelDefinitions({
        views: true,
        limit: 3,
      }, function(err, models) {
        if (err) {
          console.error(err);
          done(err);
        } else {
          let views = false;
          models.forEach(function(m) {
            // console.dir(m);
            if (m.type === 'view') {
              views = true;
            }
          });
          assert(views, 'Should have views');
          done(null, models);
        }
      });
    });
  });

  describe('Discover models excluding views', function() {
    it('should return an array of only tables', function(done) {
      db.discoverModelDefinitions({
        views: false,
        limit: 3,
      }, function(err, models) {
        if (err) {
          console.error(err);
          done(err);
        } else {
          let views = false;
          models.forEach(function(m) {
            // console.dir(m);
            if (m.type === 'view') {
              views = true;
            }
          });
          models.should.have.length(3);
          assert(!views, 'Should not have views');
          done(null, models);
        }
      });
    });
  });
});

describe('Discover models including other users', function() {
  it('should return an array of all tables and views', function(done) {
    db.discoverModelDefinitions({
      all: true,
      limit: 3,
    }, function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        let others = false;
        models.forEach(function(m) {
          // console.dir(m);
          if (m.owner !== 'strongloop') {
            others = true;
          }
        });
        assert(others, 'Should have tables/views owned by others');
        done(err, models);
      }
    });
  });
});

describe('Discover model properties', function() {
  before(function(done) {
    City = db.define('City', {
      name: {type: String},
      lng: {type: Number, postgresql: {
        dataType: 'double precision',
      }},
      lat: {type: Number, postgresql: {
        dataType: 'real',
      }},
      code: {type: Number},
      population: {type: Number, postgresql: {dataType: 'bigint'}},
      currency: {type: Number, postgresql: {dataType: 'smallint'}},
    });
    db.automigrate(done);
  });

  after(function(done) {
    City.destroyAll(done);
  });

  it('coerces `real` and `data precision` types to float on model prop ' +
    'discovery', function(done) {
    db.discoverModelProperties('city', function(err, properties) {
      assert(!err);
      assert(properties);
      const dataTypes = _.map(properties, function(prop) {
        return prop.dataType;
      });
      assert(dataTypes);
      assert.equal(dataTypes[1], 'float');
      assert.equal(dataTypes[2], 'float');
      done();
    });
  });

  it('drops precision from a field which is `integer`', function(done) {
    db.discoverModelProperties('city', function(err, properties) {
      assert(!err);
      assert(properties);
      const prop = _.find(properties, function(prop) {
        return prop.columnName === 'code';
      });
      assert(prop);
      assert.equal(prop.dataType, 'integer');
      assert(!prop.dataPrecision);
      done();
    });
  });

  it('drops precision from a field which is `bigint`', function(done) {
    db.discoverModelProperties('city', function(err, properties) {
      assert(!err);
      assert(properties);
      const prop = _.find(properties, function(prop) {
        return prop.columnName === 'population';
      });
      assert(prop);
      assert.equal(prop.dataType, 'bigint');
      assert(!prop.dataPrecision);
      done();
    });
  });

  it('drops precision from a field which is `smallint`', function(done) {
    db.discoverModelProperties('city', function(err, properties) {
      assert(!err);
      assert(properties);
      const prop = _.find(properties, function(prop) {
        return prop.columnName === 'currency';
      });
      assert(prop);
      assert.equal(prop.dataType, 'smallint');
      assert(!prop.dataPrecision);
      done();
    });
  });

  it('discover model definition and autoupdate', function(done) {
    db.discoverSchemas('city', function(err, schema) {
      assert(!err);
      assert(schema);
      schema = schema['public.city'];
      schema.properties.country = {
        type: String,
      };
      db.createModel(schema.name, schema.properties, schema.options);
      db.autoupdate(function(err) {
        assert(!err);
        const sql = db.connector.buildQueryColumns('public', 'city');
        db.connector.execute(sql, function(err, columns) {
          assert(!err);
          assert(columns);
          const cols = _.filter(columns, function(col) {
            return col.dataType === 'real' ||
            col.dataType === 'double precision' || col.dataType === 'integer'
            || col.dataType === 'bigint' || col.dataType === 'smallint';
          });
          assert(cols);
          assert.equal(cols[0].columnName, 'lng');
          assert.equal(cols[0].dataType, 'double precision');
          assert.equal(cols[1].columnName, 'lat');
          assert.equal(cols[1].dataType, 'real');
          assert.equal(cols[2].columnName, 'code');
          assert.equal(cols[2].dataType, 'integer');
          assert.equal(cols[3].columnName, 'population');
          assert.equal(cols[3].dataType, 'bigint');
          assert.equal(cols[4].columnName, 'currency');
          assert.equal(cols[4].dataType, 'smallint');
          done();
        });
      });
    });
  });

  describe('Discover a named model', function() {
    it('should return an array of columns for product', function(done) {
      db.discoverModelProperties('product', function(err, models) {
        if (err) {
          console.error(err);
          done(err);
        } else {
          models.forEach(function(m) {
            // console.dir(m);
            assert(m.tableName === 'product');
          });
          done(null, models);
        }
      });
    });
  });
});

describe('Discover model primary keys', function() {
  let Demo;
  before(function(done) {
    const demo_schema = {
      'name': 'Demo',
      'options': {
        'idInjection': false,
        'postgresql': {
          'schema': 'public',
          'table': 'demo',
        },
      },
      'properties': {
        'id': {
          'type': 'Number',
          'required': true,
          'id': 1,
          'postgresql': {
            'columnName': 'demoId',
          },
        },
        'name': {
          'type': 'String',
          'required': true,
        },
      },
    };
    Demo = db.createModel(demo_schema.name, demo_schema.properties, demo_schema.options);
    Demo.destroyAll(done);
  });

  it('should return an array of primary keys for product', function(done) {
    db.discoverPrimaryKeys('product', function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        models.forEach(function(m) {
          assert.deepEqual(m, {owner: 'strongloop',
            tableName: 'product',
            columnName: 'id',
            keySeq: 1,
            pkName: 'product_pkey'});
        });
        done(null, models);
      }
    });
  });

  it('should discover primary key and db generates instances properly', function(done) {
    db.discoverPrimaryKeys('demo', function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        models.length.should.be.equal(1);
        assert.deepEqual(models[0], {owner: 'public',
          tableName: 'demo',
          columnName: 'demoId',
          keySeq: 1,
          pkName: 'demo_pkey'});
      }
    });
    Demo.create({id: 1, name: 'checkCase'}, function(err) {
      should.not.exist(err);
      Demo.findOne({}, function(err, d) {
        should.not.exist(err);
        should.exist(d);
        d.should.have.property('id', 1);
        d.should.have.property('name', 'checkCase');
        done();
      });
    });
  });

  it('should return an array of primary keys for strongloop.product', function(done) {
    db.discoverPrimaryKeys('product', {owner: 'strongloop'}, function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        models.forEach(function(m) {
          // console.dir(m);
          assert(m.tableName === 'product');
        });
        done(null, models);
      }
    });
  });
});

describe('Discover model foreign keys', function() {
  it('should return an array of foreign keys for inventory', function(done) {
    db.discoverForeignKeys('inventory', function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        models.forEach(function(m) {
          // console.dir(m);
          assert(m.fkTableName === 'inventory');
        });
        done(null, models);
      }
    });
  });
  it('should return an array of foreign keys for strongloop.inventory', function(done) {
    db.discoverForeignKeys('inventory', {owner: 'strongloop'}, function(err, models) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        models.forEach(function(m) {
          // console.dir(m);
          assert(m.fkTableName === 'inventory');
        });
        done(null, models);
      }
    });
  });
});

describe('Discover LDL schema from a table', function() {
  it('should return an LDL schema for inventory', function(done) {
    db.discoverSchema('inventory', {owner: 'strongloop'}, function(err, schema) {
      console.log('This is our err: ', err);
      console.log('This is our schema: ', schema);
      assert(schema.name === 'Inventory');
      assert(schema.options.postgresql.schema === 'strongloop');
      assert(schema.options.postgresql.table === 'inventory');
      assert(schema.properties.productId);
      assert(schema.properties.productId.type === 'String');
      assert(schema.properties.productId.postgresql.columnName === 'product_id');
      assert(schema.properties.locationId);
      assert(schema.properties.locationId.type === 'String');
      assert(schema.properties.locationId.postgresql.columnName === 'location_id');
      assert(schema.properties.available);
      assert(schema.properties.available.type === 'Number');
      assert(schema.properties.total);
      assert(schema.properties.total.type === 'Number');
      done(null, schema);
    });
  });
});

describe('Discover and map correctly database types', function() {
  it('should handle character varying, date, timestamp with time zone, timestamp without time zone', function(done) {
    db.discoverSchema('customer', {owner: 'strongloop'}, function(err, schema) {
      if (err) {
        console.error(err);
        done(err);
      } else {
        assert(schema.properties.username);
        assert(schema.properties.username.type === 'String');
        assert(schema.properties.dateofbirth);
        assert(schema.properties.dateofbirth.type === 'Date');
        assert(schema.properties.dateofbirth.postgresql.dataType === 'date');
        assert(schema.properties.lastlogin);
        assert(schema.properties.lastlogin.type === 'Date');
        assert(schema.properties.lastlogin.postgresql.dataType === 'timestamp with time zone');
        assert(schema.properties.created);
        assert(schema.properties.created.type === 'Date');
        assert(schema.properties.created.postgresql.dataType === 'timestamp without time zone');
        done();
      }
    });
  });
});

describe('Discover and build models', function() {
  it('should build a model from discovery', function(done) {
    db.discoverAndBuildModels('GeoPoint', {schema: 'strongloop'}, function(err, schema) {
      schema.GeoPoint.find(function(err, data) {
        console.log('This is our err: ', err);
        assert(!err);
        assert(Array.isArray(data));
        assert(data[0].location);
        done();
      });
    });
  });
});