Function prepareDatabaseForSuite

test/tools/spec-runner/index.js:251–324

(suite, context)

// Test runner helpers
function prepareDatabaseForSuite(suite, context) {
  context.dbName = suite.database_name || 'spec_db';
  context.collectionName = suite.collection_name || 'spec_collection';

  const db = context.sharedClient.db(context.dbName);

  if (context.skipPrepareDatabase) return Promise.resolve();

  const setupPromise = db
    .admin()
    .command({ killAllSessions: [] })
    .catch(err => {
      if (
        err.message.match(/no such (cmd|command)/) ||
        err.message.match(/Failed to kill on some hosts/) ||
        err.code === 11601
      ) {
        return;
      }

      throw err;
    });

  if (context.collectionName == null || context.dbName === 'admin') {
    return setupPromise;
  }

  const coll = db.collection(context.collectionName);
  return setupPromise
    .then(() => {
      const options = { writeConcern: { w: 'majority' } };
      if (suite.encrypted_fields) {
        options.encryptedFields = suite.encrypted_fields;
      }
      return coll.drop(options);
    })
    .then(() => {
      if (suite.key_vault_data) {
        const dataKeysCollection = context.sharedClient.db('keyvault').collection('datakeys');
        return dataKeysCollection.drop({ writeConcern: { w: 'majority' } }).then(() => {
          if (suite.key_vault_data.length) {
            return dataKeysCollection.insertMany(suite.key_vault_data, {
              writeConcern: { w: 'majority' }
            });
          }
        });
      }
    })
    .then(() => {
      const options = { writeConcern: { w: 'majority' } };
      if (suite.json_schema) {
        options.validator = { $jsonSchema: suite.json_schema };
      }
      if (suite.encrypted_fields) {
        options.encryptedFields = suite.encrypted_fields;
      }

      return db.createCollection(context.collectionName, options);