connection-config-provider.js

/*global expect*/

'use strict';

const _ = require('lodash');
const makeKnex = require('../../knex');

module.exports = function(config) {
  describe('Connection configuration provider', function() {
    let configWorkingCopy;
    let providerInvocationCount;
    let connectionConfigWorkingCopy;

    this.beforeEach(() => {
      configWorkingCopy = _.cloneDeep(config);
      configWorkingCopy.pool.min = 1;
      configWorkingCopy.pool.max = 2;
      providerInvocationCount = 0;
      connectionConfigWorkingCopy = configWorkingCopy.connection;
    });

    it('is not used when configuration is static', async function() {
      return runTwoConcurrentTransactions(0);
    });

    it('can return a promise for a config object, which is reused when not given given an expiry checker', async () => {
      configWorkingCopy.connection = () => {
        ++providerInvocationCount;
        return Promise.resolve(connectionConfigWorkingCopy);
      };
      return runTwoConcurrentTransactions(1);
    });

    it('can return a config object, which is reused when not given given an expiry checker', async () => {
      configWorkingCopy.connection = () => {
        ++providerInvocationCount;
        return connectionConfigWorkingCopy;
      };
      return runTwoConcurrentTransactions(1);
    });

    it('reuses the same resolved config when not yet expired', async () => {
      configWorkingCopy.connection = () => {
        ++providerInvocationCount;
        return Object.assign(connectionConfigWorkingCopy, {
          expirationChecker: () => false,
        });
      };
      return runTwoConcurrentTransactions(1);
    });

    it('replaces the resolved config when expired', async () => {
      configWorkingCopy.connection = () => {
        ++providerInvocationCount;
        return Object.assign(connectionConfigWorkingCopy, {
          expirationChecker: () => true,
        });
      };
      return runTwoConcurrentTransactions(2);
    });

    async function runTwoConcurrentTransactions(expectedInvocationCount) {
      const knex = makeKnex(configWorkingCopy);
      await knex.transaction(async (trx) => {
        await trx.select(1);
        await knex.transaction(async (trx2) => {
          await trx2.select(2);
        });
      });
      await knex.destroy();
      expect(providerInvocationCount).equals(expectedInvocationCount);
    }
  });
};