hnhzServer/node_modules/memdb-server/test/app/document.js

// Copyright 2015 The MemDB Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
// implied. See the License for the specific language governing
// permissions and limitations under the License. See the AUTHORS file
// for names of contributors.

'use strict';

var P = require('bluebird');
var _ = require('lodash');
var should = require('should');
var assert = require('assert');
var Document = require('../../app/document'); // jshint ignore:line
var utils = require('../../app/utils');
var logger = require('memdb-logger').getLogger('test', __filename);

describe('document test', function(){
    it('find', function(){
        var value = {k1 : 1, k2 : 1};
        var doc = new Document({_id : '1', doc : value});
        // Get all fields
        doc.find('c1').should.eql(value);
        // Get specified fields
        doc.find('c1', 'k1').should.eql({_id : '1', k1 : 1});
        doc.find('c1', {'k1' : true}).should.eql({_id : '1', k1 : 1});
        doc.find('c1', {'k1' : false}).should.eql({_id : '1', k2 : 1});
    });

    it('update', function(cb){
        var doc = new Document({_id : '1', doc : null, watchedFields : ['k1']});

        doc.on('updateUncommited', function(connId, field, oldValue, newValue){
            logger.debug(field, oldValue, newValue);
        });

        return P.try(function(){
            // Lock for write
            return doc.lock('c1');
        })
        .then(function(){
            should(doc.update.bind(doc, 'c1', {k1 : 1})).throw();

            doc.insert('c1', {k1 : 1});
            doc.find('c1').should.eql({_id : '1', k1 : 1});

            //replace doc
            doc.update('c1', {k1 : 2, k2 : 2});
            doc.find('c1').should.eql({_id : '1', k1 : 2, k2 : 2});

            //$set $unset
            doc.update('c1', {k1 : 1});
            doc.update('c1', {$set : {k1 : 2, k2 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : 2, k2 : 2});
            doc.update('c1', {$unset : {k2 : true}});
            doc.find('c1').should.eql({_id : '1', k1 : 2});
            doc.update('c1', {$set : {'k2.k3' : 3}});
            doc.find('c1').should.eql({_id : '1', k1 : 2, k2 : {k3 : 3}});

            //$inc
            doc.update('c1', {k1 : 1});
            doc.update('c1', {$inc : {k1 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : 3});

            //$push $pushAll $pop $addToSet $pull
            doc.update('c1', {});
            doc.update('c1', {$push : {k1 : 1}});
            doc.update('c1', {$push : {k1 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : [1, 2]});
            doc.update('c1', {$pop : {k1 : 1}});
            doc.find('c1').should.eql({_id : '1', k1 : [1]});
            doc.update('c1', {$addToSet: {k1 : 1}});
            doc.update('c1', {$addToSet: {k1 : 1}});
            doc.update('c1', {$addToSet: {k1 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : [1, 2]});
            doc.update('c1', {$pull : {k1 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : [1]});
            doc.update('c1', {$pushAll : {k1 : [2, 3]}});
            doc.find('c1').should.eql({_id : '1', k1 : [1, 2, 3]});

            //multiple modifiers
            doc.update('c1', {k1 : 0, k2 : 1, k3 : [1]});
            doc.update('c1', {$set : {k1 : 1}, $inc : {k2 : 1}, $push : {k3 : 2}});
            doc.find('c1').should.eql({_id : '1', k1 : 1, k2 : 2, k3 : [1, 2]});
        })
        .nodeify(cb);
    });

    it('insert/remove', function(cb){
        var value = {k1 : 1};
        // Init with non-exist doc
        var doc = new Document({_id : '1', exist : false});

        return P.try(function(){
            assert(doc.find('c1') === null);
            // Lock for write
            return doc.lock('c1');
        })
        .then(function(){
            // insert doc
            doc.insert('c1', value);
            doc.find('c1').should.eql(value);
            // should throw exception when insert existing doc
            should(doc.insert.bind(doc, 'c1', value)).throw();
            // remove doc
            doc.remove('c1');
            assert(doc.find('c1') === null);
            // should throw exception when remove non-exist doc
            should(doc.remove.bind(doc, 'c1')).throw();
        })
        .nodeify(cb);
    });

    it('commit/rollback/lock/unlock', function(cb){
        var doc = new Document({_id : '1', doc : {k1 : 1}, watchedFields : ['k1']});

        doc.on('updateUncommited', function(connId, field, oldValue, newValue){
            logger.debug(field, oldValue, newValue);
        });

        return P.try(function(){
            // should throw when write without lock
            should(doc.remove.bind(doc, 'c1')).throw();

            return doc.lock('c1')
            .then(function(){
                // lock twice should be ok
                return doc.lock('c1');
            });
        })
        .then(function(){
            doc.update('c1', {k1 : 2});
            doc.find('c1', {_id : false}).should.eql({k1 : 2});
            doc.rollback('c1');
            // should rolled back
            doc.find('c1', {_id : false}).should.eql({k1 : 1});
            // should unlocked
            should(doc.remove.bind(doc, 'c1')).throw();

            return doc.lock('c1');
        })
        .then(function(){
            doc.update('c1', {k1 : 2});
            doc.commit('c1');
            // should commited
            doc.find('c1', {_id : false}).should.eql({k1 : 2});
            // should unlocked
            should(doc.remove.bind(doc, 'c1')).throw();

            // remove and rollback
            return doc.lock('c1');
        })
        .then(function(){
            doc.remove('c1');
            doc.rollback('c1');
            assert(doc.find('c1') !== null);

            // remove and commit
            return doc.lock('c1');
        })
        .then(function(){
            doc.remove('c1');
            doc.commit('c1');
            assert(doc.find('c1') === null);

            // insert and rollback
            return doc.lock('c1');
        })
        .then(function(){
            doc.insert('c1', {k1 : 1});
            doc.rollback('c1');
            assert(doc.find('c1') === null);

            // insert and commit
            return doc.lock('c1');
        })
        .then(function(){
            doc.insert('c1', {k1 : 1});
            doc.commit('c1');
            doc.find('c1', {_id : false}).should.eql({k1 : 1});
        })
        .nodeify(cb);
    });

    it('read from other connections', function(cb){
        var value = {k1 : 1, k2 : 1};
        var doc = new Document({_id : '1', doc : utils.clone(value), watchedFields : ['k1', 'k2']});

        doc.on('updateUncommited', function(connId, field, oldValue, newValue){
            logger.debug(field, oldValue, newValue);
        });

        return P.try(function(){
            return doc.lock('c1');
        })
        .then(function(){
            // update field
            doc.update('c1', {k1 : 2});
            // read from c2
            doc.find('c2', 'k1').should.eql({'_id' : '1', k1 : 1});
            doc.find('c2', {_id : false}).should.eql(value);
            // add field
            doc.update('c1', {k3 : 1});
            doc.find('c2', {_id : false}).should.eql(value);
            // remove field
            doc.update('c1', {k2 : undefined});
            doc.find('c2', {_id : false}).should.eql(value);
            // replace doc
            doc.update('c1', {k1 : 1}, {replace : true});
            doc.find('c2', {_id : false}).should.eql(value);
            // commit
            doc.commit('c1');
            // c2 should see the newest value now
            doc.find('c2', {_id : false}).should.eql({k1 : 1});

            // remove and commit
            return doc.lock('c1');
        })
        .then(function(){
            doc.remove('c1');
            assert(doc.find('c2') !== null);
            doc.commit('c1');
            assert(doc.find('c2') === null);

            // insert and commit
            return doc.lock('c1');
        })
        .then(function(){
            doc.insert('c1', utils.clone(value));
            assert(doc.find('c2') === null);
            doc.commit('c1');
            doc.find('c2', {_id : false}).should.eql(value);
        })
        .nodeify(cb);
    });

    it('concurrency', function(cb){
        var doc = new Document({_id : '1', doc : {k : 0}, watchedFields : ['k']});

        doc.on('updateUncommited', function(connId, field, oldValue, newValue){
            logger.debug(field, oldValue, newValue);
        });

        var concurrency = 4;
        // Simulate non-atomic check and update
        return P.map(_.range(concurrency), function(connId){
            var value = null;
            return P.delay(_.random(10))
            .then(function(){
                logger.trace('%s start lock', connId);
                return doc.lock(connId);
            })
            .then(function(){
                logger.trace('%s got lock', connId);
                value = doc.find(connId, {_id : false});
            })
            .delay(_.random(20))
            .then(function(){
                value.k++;
                doc.update(connId, value);
                doc.commit(connId);
                logger.trace('%s commited', connId);
            });
        })
        .then(function(){
            //Result should equal to concurrency
            doc.find(null, {_id : false}).should.eql({k : concurrency});
        })
        .nodeify(cb);
    });

    it.skip('too large doc', function(cb){
        var doc = new Document({_id : 1});
        doc.lock(1);
        should(function(){
            doc.insert(1, {tooLargeArray : _.range(1000000)});
        }).throw();
        cb();
    });
});