iotagent-node-lib/test/unit/mongodb/mongoDBUtils.js

IoT Agent library to interface with NGSI Context Broker

/*
 * Copyright 2014 Telefonica Investigación y Desarrollo, S.A.U
 *
 * This file is part of fiware-iotagent-lib
 *
 * fiware-iotagent-lib is free software: you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the License,
 * or (at your option) any later version.
 *
 * fiware-iotagent-lib is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public
 * License along with fiware-iotagent-lib.
 * If not, see http://www.gnu.org/licenses/.
 *
 * For those usages not covered by the GNU Affero General Public License
 * please contact with::[contacto@tid.es]
 */

/* eslint-disable no-unused-vars */

const MongoClient = require('mongodb').MongoClient;
const async = require('async');

function cleanDb(host, name, callback) {
    const url = 'mongodb://' + host + ':27017/' + name;

    MongoClient.connect(url, function (err, db) {
        if (db && db.db()) {
            db.db().dropDatabase(function (err, result) {
                db.close();
                callback();
            });
        }
    });
}

function cleanDbs(callback) {
    async.series([async.apply(cleanDb, 'localhost', 'iotagent')], callback);
}

function populate(host, dbName, entityList, collectionName, callback) {
    const url = 'mongodb://' + host + ':27017/' + dbName;

    MongoClient.connect(url, function (err, db) {
        if (db) {
            db.db()
                .collection(collectionName)
                .insertMany(entityList, function (err, r) {
                    db.close();
                    callback(err);
                });
        } else {
            callback();
        }
    });
}

exports.cleanDb = cleanDb;
exports.cleanDbs = cleanDbs;
exports.populate = populate;