123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467 |
- 'use strict';
- var assert = require('assert');
- var Kareem = require('../');
- const { beforeEach, describe, it } = require('mocha');
- describe('wrap()', function() {
- var hooks;
- beforeEach(function() {
- hooks = new Kareem();
- });
- it('handles pre errors', function(done) {
- hooks.pre('cook', function(done) {
- done('error!');
- });
- hooks.post('cook', function(obj) {
- obj.tofu = 'no';
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [obj];
- args.push(function(error, result) {
- assert.equal('error!', error);
- assert.ok(!result);
- assert.equal(undefined, obj.tofu);
- done();
- });
- hooks.wrap(
- 'cook',
- function(o, callback) {
- // Should never get called
- assert.ok(false);
- callback(null, o);
- },
- obj,
- args);
- });
- it('handles pre errors when no callback defined', function(done) {
- hooks.pre('cook', function(done) {
- done('error!');
- });
- hooks.post('cook', function(obj) {
- obj.tofu = 'no';
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [obj];
- hooks.wrap(
- 'cook',
- function(o, callback) {
- // Should never get called
- assert.ok(false);
- callback(null, o);
- },
- obj,
- args);
- setTimeout(
- function() {
- done();
- },
- 25);
- });
- it('handles errors in wrapped function', function(done) {
- hooks.pre('cook', function(done) {
- done();
- });
- hooks.post('cook', function(obj) {
- obj.tofu = 'no';
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [obj];
- args.push(function(error, result) {
- assert.equal('error!', error);
- assert.ok(!result);
- assert.equal(undefined, obj.tofu);
- done();
- });
- hooks.wrap(
- 'cook',
- function(o, callback) {
- callback('error!');
- },
- obj,
- args);
- });
- it('handles errors in post', function(done) {
- hooks.pre('cook', function(done) {
- done();
- });
- hooks.post('cook', function(obj, callback) {
- obj.tofu = 'no';
- callback('error!');
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [obj];
- args.push(function(error, result) {
- assert.equal('error!', error);
- assert.ok(!result);
- assert.equal('no', obj.tofu);
- done();
- });
- hooks.wrap(
- 'cook',
- function(o, callback) {
- callback(null, o);
- },
- obj,
- args);
- });
- it('defers errors to post hooks if enabled', function(done) {
- hooks.pre('cook', function(done) {
- done(new Error('fail'));
- });
- hooks.post('cook', function(error, res, callback) {
- callback(new Error('another error occurred'));
- });
- var args = [];
- args.push(function(error) {
- assert.equal(error.message, 'another error occurred');
- done();
- });
- hooks.wrap(
- 'cook',
- function(callback) {
- assert.ok(false);
- callback();
- },
- null,
- args,
- { useErrorHandlers: true, numCallbackParams: 1 });
- });
- it('error handlers with no callback', function(done) {
- hooks.pre('cook', function(done) {
- done(new Error('fail'));
- });
- hooks.post('cook', function(error, callback) {
- assert.equal(error.message, 'fail');
- done();
- });
- var args = [];
- hooks.wrap(
- 'cook',
- function(callback) {
- assert.ok(false);
- callback();
- },
- null,
- args,
- { useErrorHandlers: true });
- });
- it('error handlers with no error', function(done) {
- hooks.post('cook', function(error, callback) {
- callback(new Error('another error occurred'));
- });
- var args = [];
- args.push(function(error) {
- assert.ifError(error);
- done();
- });
- hooks.wrap(
- 'cook',
- function(callback) {
- callback();
- },
- null,
- args,
- { useErrorHandlers: true });
- });
- it('works with no args', function(done) {
- hooks.pre('cook', function(done) {
- done();
- });
- hooks.post('cook', function(callback) {
- obj.tofu = 'no';
- callback();
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [];
- hooks.wrap(
- 'cook',
- function(callback) {
- callback(null);
- },
- obj,
- args);
- setTimeout(
- function() {
- assert.equal('no', obj.tofu);
- done();
- },
- 25);
- });
- it('handles pre errors with no args', function(done) {
- hooks.pre('cook', function(done) {
- done('error!');
- });
- hooks.post('cook', function(callback) {
- obj.tofu = 'no';
- callback();
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [];
- hooks.wrap(
- 'cook',
- function(callback) {
- callback(null);
- },
- obj,
- args);
- setTimeout(
- function() {
- assert.equal(undefined, obj.tofu);
- done();
- },
- 25);
- });
- it('handles wrapped function errors with no args', function(done) {
- hooks.pre('cook', function(done) {
- obj.waffles = false;
- done();
- });
- hooks.post('cook', function(callback) {
- obj.tofu = 'no';
- callback();
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [];
- hooks.wrap(
- 'cook',
- function(callback) {
- callback('error!');
- },
- obj,
- args);
- setTimeout(
- function() {
- assert.equal(false, obj.waffles);
- assert.equal(undefined, obj.tofu);
- done();
- },
- 25);
- });
- it('supports overwriteResult', function(done) {
- hooks.post('cook', function(eggs, callback) {
- callback(Kareem.overwriteResult(5));
- });
- const args = [(err, res) => {
- assert.ifError(err);
- assert.equal(res, 5);
- done();
- }];
- hooks.wrap(
- 'cook',
- function(callback) {
- callback(null, 4);
- },
- null,
- args);
- });
- it('supports skipWrappedFunction', function(done) {
- const execed = {};
- hooks.pre('cook', function pre(callback) {
- execed.pre = true;
- callback(Kareem.skipWrappedFunction(3));
- });
- hooks.post('cook', function(res, callback) {
- assert.equal(res, 3);
- execed.post = true;
- callback();
- });
- const args = [(err, res) => {
- assert.ifError(err);
- assert.equal(res, 3);
- assert.ok(execed.pre);
- assert.ok(execed.post);
- assert.ok(!execed.wrapped);
- done();
- }];
- hooks.wrap(
- 'cook',
- function wrapped(callback) {
- execed.wrapped = true;
- callback();
- },
- null,
- args);
- });
- it('supports skipWrappedFunction with arguments', function(done) {
- const execed = {};
- hooks.pre('cook', function pre(callback, arg) {
- execed.pre = true;
- assert.strictEqual(arg, 4);
- callback(Kareem.skipWrappedFunction(3));
- });
- hooks.post('cook', function(res, callback) {
- assert.equal(res, 3);
- execed.post = true;
- callback();
- });
- const args = [4, (err, res) => {
- assert.ifError(err);
- assert.equal(res, 3);
- assert.ok(execed.pre);
- assert.ok(execed.post);
- assert.ok(!execed.wrapped);
- done();
- }];
- hooks.wrap(
- 'cook',
- function wrapped(arg, callback) {
- execed.wrapped = true;
- callback();
- },
- null,
- args);
- });
- it('handles post errors with no args', function(done) {
- hooks.pre('cook', function(done) {
- obj.waffles = false;
- done();
- });
- hooks.post('cook', function(callback) {
- obj.tofu = 'no';
- callback('error!');
- });
- var obj = { bacon: 0, eggs: 0 };
- var args = [];
- hooks.wrap(
- 'cook',
- function(callback) {
- callback();
- },
- obj,
- args);
- setTimeout(
- function() {
- assert.equal(false, obj.waffles);
- assert.equal('no', obj.tofu);
- done();
- },
- 25);
- });
- it('catches sync errors', function(done) {
- hooks.pre('cook', function(done) {
- done();
- });
- hooks.post('cook', function(callback) {
- callback();
- });
- var args = [];
- args.push(function(error) {
- assert.equal(error.message, 'oops!');
- done();
- });
- hooks.wrap(
- 'cook',
- function() {
- throw new Error('oops!');
- },
- null,
- args);
- });
- it('sync wrappers', function() {
- var calledPre = 0;
- var calledFn = 0;
- var calledPost = 0;
- hooks.pre('cook', function() {
- ++calledPre;
- });
- hooks.post('cook', function() {
- ++calledPost;
- });
- var wrapper = hooks.createWrapperSync('cook', function() { ++calledFn; });
- wrapper();
- assert.equal(calledPre, 1);
- assert.equal(calledFn, 1);
- assert.equal(calledPost, 1);
- });
- it('sync wrappers with overwriteResult', function() {
- hooks.pre('cook', function() {
- });
- hooks.post('cook', function() {
- return Kareem.overwriteResult(5);
- });
- const wrapper = hooks.createWrapperSync('cook', function() { return 4; });
- assert.strictEqual(wrapper(), 5);
- });
- });
|