Update mongodb node module in test (#1108)

Author: lijinxia

test/restTestCasesKarma/scripts/authorization/init/initdb.js

@@ -15,6 +15,11 @@ var currentVersion = '1.0';
 var fs = require('fs');
 var path = require('path');
 
+var MongoClient = require('mongodb').MongoClient;
+var connectOption = {
+  useUnifiedTopology: true,
+};
+var client;
 var db;
 var cipher = require('./cipher');
 
@@ -23,6 +28,9 @@ var configFile = path.join(dirName, 'management_api.toml');
 var sampleServiceFile = path.resolve(dirName, '../environment.js');
 
 function prepareDB(next) {
+  if (dbURL.indexOf('mongodb://') !== 0) {
+    dbURL = 'mongodb://' + dbURL;
+  }
   if (fs.existsSync(cipher.astore)) {
     cipher.unlock(cipher.k, cipher.astore, function cb (err, authConfig) {
       if (!err) {
@@ -34,20 +42,34 @@ function prepareDB(next) {
       } else {
         log.error('Failed to get mongodb auth:', err);
       }
-      db = require('mongojs')(dbURL, ['services', 'infos', 'rooms']);
-      next();
+      MongoClient.connect(dbURL, connectOption, function(err, cli) {
+        if (!err) {
+          client = cli;
+          db = client.db();
+          next();
+        } else {
+          console.error('Failed to connect mongodb:', err);
+        }
+      });
     });
   } else {
-    db = require('mongojs')(dbURL, ['services', 'infos', 'rooms']);
-    next();
+    MongoClient.connect(dbURL, connectOption, function(err, cli) {
+        if (!err) {
+          client = cli;
+          db = client.db();
+          next();
+        } else {
+          console.error('Failed to connect mongodb:', err);
+        }
+      });
   }
 }
 
 function upgradeH264(next) {
-  db.rooms.find({}).toArray(function (err, rooms) {
+  db.collection('rooms').find({}).toArray(function (err, rooms) {
     if (err) {
       console.log('Error in getting rooms:', err.message);
-      db.close();
+      client.close();
       return;
     }
     if (!rooms || rooms.length === 0) {
@@ -83,7 +105,7 @@ function upgradeH264(next) {
         }
       });
 
-      db.rooms.save(room, function (e, saved) {
+      db.collection('rooms').updateOne({_id: room._id}, room, function (e, saved) {
         if (e) {
           console.log('Error in upgrading room:', room._id, e.message);
         }
@@ -97,28 +119,28 @@ function upgradeH264(next) {
 }
 
 function checkVersion (next) {
-  db.infos.findOne({_id: 1}, function cb (err, info) {
+  db.collection('infos').findOne({_id: 1}, function cb (err, info) {
     if (err) {
       console.log('mongodb: error in query info');
-      return db.close();
+      return client.close();
     }
     if (info) {
       if (info.version === '1.0') {
         next();
       }
     } else {
-      db.services.findOne({}, function cb (err, service) {
+      db.collection('services').findOne({}, function cb (err, service) {
         if (err) {
           console.log('mongodb: error in query service');
-          return db.close();
+          return client.close();
         }
         var upgradeNext = function(next) {
           upgradeH264(function() {
             info = {_id: 1, version: currentVersion};
-            db.infos.save(info, function cb (e, s) {
+            db.collection('infos').save(info, function cb (e, s) {
               if (e) {
                 console.log('mongodb: error in updating version');
-                return db.close();
+                return client.close();
               }
               next();
             });
@@ -156,15 +178,15 @@ function checkVersion (next) {
 }
 
 function prepareService (serviceName, next) {
-  db.services.findOne({name: serviceName}, function cb (err, service) {
+  db.collection('services').findOne({name: serviceName}, function cb (err, service) {
     if (err || !service) {
       var crypto = require('crypto');
       var key = crypto.pbkdf2Sync(crypto.randomBytes(64).toString('hex'), crypto.randomBytes(32).toString('hex'), 4000, 128, 'sha256').toString('base64');
       service = {name: serviceName, key: cipher.encrypt(cipher.k, key), encrypted: true, rooms: [], __v: 0};
-      db.services.save(service, function cb (err, saved) {
+      db.collection('services').save(service, function cb (err, saved) {
         if (err) {
           console.log('mongodb: error in adding', serviceName);
-          return db.close();
+          return client.close();
         }
         saved.key = key;
         next(saved);
@@ -222,7 +244,7 @@ prepareDB(function() {
         var sampleServiceKey = service.key;
         console.log('sampleServiceId:', sampleServiceId);
         console.log('sampleServiceKey:', sampleServiceKey);
-        db.close();
+        client.close();
 
         writeSampleFile(sampleServiceId, sampleServiceKey);
       });

test/restTestCasesKarma/scripts/authorization/init/package.json

@@ -8,7 +8,7 @@
     "express": "*",
     "fraction.js": "*",
     "log4js": "^6.3.0",
-    "mongojs": "",
+    "mongodb": "^3.6.4",
     "mongoose": "^5.3.9",
     "toml": "*"
   },