From b6cc34e2e884d7ad0e62258398540fd0b51720f9 Mon Sep 17 00:00:00 2001
From: Matt Broadstone <mbroadst@gmail.com>
Date: Fri, 25 Aug 2017 08:20:22 -0400
Subject: [PATCH] feat(shared): add helper utilities for assertion and suite
 setup

NODE-1095
---
 test/functional/shared.js | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/test/functional/shared.js b/test/functional/shared.js
index 1499936f19f..db05c51b650 100644
--- a/test/functional/shared.js
+++ b/test/functional/shared.js
@@ -1,4 +1,5 @@
-var MongoClient = require('../../').MongoClient;
+var MongoClient = require('../../').MongoClient,
+  expect = require('chai').expect;
 
 function connectToDb(url, db, options, callback) {
   if (typeof options === 'function') {
@@ -12,6 +13,30 @@ function connectToDb(url, db, options, callback) {
   });
 }
 
+function setupDatabase(configuration) {
+  var dbName = configuration.db;
+  var client = configuration.newClient(configuration.writeConcernMax(), {
+    poolSize: 1
+  });
+
+  return client.connect().then(function() {
+    var db = client.db(dbName);
+    return db.dropDatabase();
+  });
+}
+
+var assert = {
+  equal: function(a, b) {
+    expect(a).to.equal(b);
+  },
+
+  ok: function(a) {
+    expect(a).to.be.ok;
+  }
+};
+
 module.exports = {
-  connectToDb: connectToDb
+  connectToDb: connectToDb,
+  setupDatabase: setupDatabase,
+  assert: assert
 };