You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							1065 lines
						
					
					
						
							26 KiB
						
					
					
				
			
		
		
	
	
							1065 lines
						
					
					
						
							26 KiB
						
					
					
				| 'use strict'
 | |
| 
 | |
| /* eslint no-prototype-builtins: 0 */
 | |
| 
 | |
| const t = require('tap')
 | |
| const test = t.test
 | |
| const Fastify = require('..')
 | |
| const fp = require('fastify-plugin')
 | |
| const sget = require('simple-get').concat
 | |
| const symbols = require('../lib/symbols.js')
 | |
| const proxyquire = require('proxyquire')
 | |
| 
 | |
| test('server methods should exist', t => {
 | |
|   t.plan(2)
 | |
|   const fastify = Fastify()
 | |
|   t.ok(fastify.decorate)
 | |
|   t.ok(fastify.hasDecorator)
 | |
| })
 | |
| 
 | |
| test('should check if the given decoration already exist when null', t => {
 | |
|   t.plan(1)
 | |
|   const fastify = Fastify()
 | |
|   fastify.decorate('null', null)
 | |
|   fastify.ready(() => {
 | |
|     t.ok(fastify.hasDecorator('null'))
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('server methods should be encapsulated via .register', t => {
 | |
|   t.plan(2)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorate('test', () => {})
 | |
|     t.ok(instance.test)
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     t.notOk(fastify.test)
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('hasServerMethod should check if the given method already exist', t => {
 | |
|   t.plan(2)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorate('test', () => {})
 | |
|     t.ok(instance.hasDecorator('test'))
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     t.notOk(fastify.hasDecorator('test'))
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorate should throw if a declared dependency is not present', t => {
 | |
|   t.plan(3)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     try {
 | |
|       instance.decorate('test', () => {}, ['dependency'])
 | |
|       t.fail()
 | |
|     } catch (e) {
 | |
|       t.same(e.code, 'FST_ERR_DEC_MISSING_DEPENDENCY')
 | |
|       t.same(e.message, 'The decorator is missing dependency \'dependency\'.')
 | |
|     }
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => t.pass())
 | |
| })
 | |
| 
 | |
| test('decorate should throw if declared dependency is not array', t => {
 | |
|   t.plan(3)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     try {
 | |
|       instance.decorate('test', () => {}, {})
 | |
|       t.fail()
 | |
|     } catch (e) {
 | |
|       t.same(e.code, 'FST_ERR_DEC_DEPENDENCY_INVALID_TYPE')
 | |
|       t.same(e.message, 'The dependencies of decorator \'test\' must be of type Array.')
 | |
|     }
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => t.pass())
 | |
| })
 | |
| 
 | |
| // issue #777
 | |
| test('should pass error for missing request decorator', t => {
 | |
|   t.plan(2)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   const plugin = fp(function (instance, opts, done) {
 | |
|     done()
 | |
|   }, {
 | |
|     decorators: {
 | |
|       request: ['foo']
 | |
|     }
 | |
|   })
 | |
|   fastify
 | |
|     .register(plugin)
 | |
|     .ready((err) => {
 | |
|       t.type(err, Error)
 | |
|       t.match(err, /The decorator 'foo'/)
 | |
|     })
 | |
| })
 | |
| 
 | |
| test('decorateReply inside register', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorateReply('test', 'test')
 | |
| 
 | |
|     instance.get('/yes', (req, reply) => {
 | |
|       t.ok(reply.test, 'test exists')
 | |
|       reply.send({ hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(reply.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateReply as plugin (inside .after)', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.register(fp((i, o, n) => {
 | |
|       instance.decorateReply('test', 'test')
 | |
|       n()
 | |
|     })).after(() => {
 | |
|       instance.get('/yes', (req, reply) => {
 | |
|         t.ok(reply.test)
 | |
|         reply.send({ hello: 'world' })
 | |
|       })
 | |
|     })
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(reply.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateReply as plugin (outside .after)', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.register(fp((i, o, n) => {
 | |
|       instance.decorateReply('test', 'test')
 | |
|       n()
 | |
|     }))
 | |
| 
 | |
|     instance.get('/yes', (req, reply) => {
 | |
|       t.ok(reply.test)
 | |
|       reply.send({ hello: 'world' })
 | |
|     })
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(reply.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest inside register', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorateRequest('test', 'test')
 | |
| 
 | |
|     instance.get('/yes', (req, reply) => {
 | |
|       t.ok(req.test, 'test exists')
 | |
|       reply.send({ hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(req.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest as plugin (inside .after)', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.register(fp((i, o, n) => {
 | |
|       instance.decorateRequest('test', 'test')
 | |
|       n()
 | |
|     })).after(() => {
 | |
|       instance.get('/yes', (req, reply) => {
 | |
|         t.ok(req.test)
 | |
|         reply.send({ hello: 'world' })
 | |
|       })
 | |
|     })
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(req.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest as plugin (outside .after)', t => {
 | |
|   t.plan(11)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.register(fp((i, o, n) => {
 | |
|       instance.decorateRequest('test', 'test')
 | |
|       n()
 | |
|     }))
 | |
| 
 | |
|     instance.get('/yes', (req, reply) => {
 | |
|       t.ok(req.test)
 | |
|       reply.send({ hello: 'world' })
 | |
|     })
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/no', (req, reply) => {
 | |
|     t.notOk(req.test)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/no'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorators should be instance separated', t => {
 | |
|   t.plan(1)
 | |
| 
 | |
|   const fastify1 = Fastify()
 | |
|   const fastify2 = Fastify()
 | |
| 
 | |
|   fastify1.decorate('test', 'foo')
 | |
|   fastify2.decorate('test', 'foo')
 | |
| 
 | |
|   fastify1.decorateRequest('test', 'foo')
 | |
|   fastify2.decorateRequest('test', 'foo')
 | |
| 
 | |
|   fastify1.decorateReply('test', 'foo')
 | |
|   fastify2.decorateReply('test', 'foo')
 | |
| 
 | |
|   t.pass()
 | |
| })
 | |
| 
 | |
| test('hasRequestDecorator', t => {
 | |
|   const requestDecoratorName = 'my-decorator-name'
 | |
| 
 | |
|   t.test('is a function', t => {
 | |
|     t.plan(1)
 | |
|     const fastify = Fastify()
 | |
|     t.ok(fastify.hasRequestDecorator)
 | |
|   })
 | |
| 
 | |
|   t.test('should check if the given request decoration already exist', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|     fastify.decorateRequest(requestDecoratorName, 42)
 | |
|     t.ok(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|   })
 | |
| 
 | |
|   t.test('should check if the given request decoration already exist when null', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|     fastify.decorateRequest(requestDecoratorName, null)
 | |
|     t.ok(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|   })
 | |
| 
 | |
|   t.test('should be plugin encapsulable', t => {
 | |
|     t.plan(4)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasRequestDecorator(requestDecoratorName))
 | |
| 
 | |
|     fastify.register(function (fastify2, opts, done) {
 | |
|       fastify2.decorateRequest(requestDecoratorName, 42)
 | |
|       t.ok(fastify2.hasRequestDecorator(requestDecoratorName))
 | |
|       done()
 | |
|     })
 | |
| 
 | |
|     t.notOk(fastify.hasRequestDecorator(requestDecoratorName))
 | |
| 
 | |
|     fastify.ready(function () {
 | |
|       t.notOk(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   t.test('should be inherited', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     fastify.decorateRequest(requestDecoratorName, 42)
 | |
| 
 | |
|     fastify.register(function (fastify2, opts, done) {
 | |
|       t.ok(fastify2.hasRequestDecorator(requestDecoratorName))
 | |
|       done()
 | |
|     })
 | |
| 
 | |
|     fastify.ready(function () {
 | |
|       t.ok(fastify.hasRequestDecorator(requestDecoratorName))
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   t.end()
 | |
| })
 | |
| 
 | |
| test('hasReplyDecorator', t => {
 | |
|   const replyDecoratorName = 'my-decorator-name'
 | |
| 
 | |
|   t.test('is a function', t => {
 | |
|     t.plan(1)
 | |
|     const fastify = Fastify()
 | |
|     t.ok(fastify.hasReplyDecorator)
 | |
|   })
 | |
| 
 | |
|   t.test('should check if the given reply decoration already exist', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|     fastify.decorateReply(replyDecoratorName, 42)
 | |
|     t.ok(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|   })
 | |
| 
 | |
|   t.test('should check if the given reply decoration already exist when null', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|     fastify.decorateReply(replyDecoratorName, null)
 | |
|     t.ok(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|   })
 | |
| 
 | |
|   t.test('should be plugin encapsulable', t => {
 | |
|     t.plan(4)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     t.notOk(fastify.hasReplyDecorator(replyDecoratorName))
 | |
| 
 | |
|     fastify.register(function (fastify2, opts, done) {
 | |
|       fastify2.decorateReply(replyDecoratorName, 42)
 | |
|       t.ok(fastify2.hasReplyDecorator(replyDecoratorName))
 | |
|       done()
 | |
|     })
 | |
| 
 | |
|     t.notOk(fastify.hasReplyDecorator(replyDecoratorName))
 | |
| 
 | |
|     fastify.ready(function () {
 | |
|       t.notOk(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   t.test('should be inherited', t => {
 | |
|     t.plan(2)
 | |
|     const fastify = Fastify()
 | |
| 
 | |
|     fastify.decorateReply(replyDecoratorName, 42)
 | |
| 
 | |
|     fastify.register(function (fastify2, opts, done) {
 | |
|       t.ok(fastify2.hasReplyDecorator(replyDecoratorName))
 | |
|       done()
 | |
|     })
 | |
| 
 | |
|     fastify.ready(function () {
 | |
|       t.ok(fastify.hasReplyDecorator(replyDecoratorName))
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   t.end()
 | |
| })
 | |
| 
 | |
| test('should register properties via getter/setter objects', t => {
 | |
|   t.plan(3)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorate('test', {
 | |
|       getter () {
 | |
|         return 'a getter'
 | |
|       }
 | |
|     })
 | |
|     t.ok(instance.test)
 | |
|     t.ok(instance.test, 'a getter')
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     t.notOk(fastify.test)
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest should work with getter/setter', t => {
 | |
|   t.plan(5)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorateRequest('test', {
 | |
|       getter () {
 | |
|         return 'a getter'
 | |
|       }
 | |
|     })
 | |
| 
 | |
|     instance.get('/req-decorated-get-set', (req, res) => {
 | |
|       res.send({ test: req.test })
 | |
|     })
 | |
| 
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/not-decorated', (req, res) => {
 | |
|     t.notOk(req.test)
 | |
|     res.send()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     fastify.inject({ url: '/req-decorated-get-set' }, (err, res) => {
 | |
|       t.error(err)
 | |
|       t.same(JSON.parse(res.payload), { test: 'a getter' })
 | |
|     })
 | |
| 
 | |
|     fastify.inject({ url: '/not-decorated' }, (err, res) => {
 | |
|       t.error(err)
 | |
|       t.pass()
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateReply should work with getter/setter', t => {
 | |
|   t.plan(5)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorateReply('test', {
 | |
|       getter () {
 | |
|         return 'a getter'
 | |
|       }
 | |
|     })
 | |
| 
 | |
|     instance.get('/res-decorated-get-set', (req, res) => {
 | |
|       res.send({ test: res.test })
 | |
|     })
 | |
| 
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/not-decorated', (req, res) => {
 | |
|     t.notOk(res.test)
 | |
|     res.send()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     fastify.inject({ url: '/res-decorated-get-set' }, (err, res) => {
 | |
|       t.error(err)
 | |
|       t.same(JSON.parse(res.payload), { test: 'a getter' })
 | |
|     })
 | |
| 
 | |
|     fastify.inject({ url: '/not-decorated' }, (err, res) => {
 | |
|       t.error(err)
 | |
|       t.pass()
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('should register empty values', t => {
 | |
|   t.plan(2)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     instance.decorate('test', null)
 | |
|     t.ok(instance.hasOwnProperty('test'))
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     t.notOk(fastify.test)
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('nested plugins can override things', t => {
 | |
|   t.plan(6)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   const rootFunc = () => {}
 | |
|   fastify.decorate('test', rootFunc)
 | |
|   fastify.decorateRequest('test', rootFunc)
 | |
|   fastify.decorateReply('test', rootFunc)
 | |
| 
 | |
|   fastify.register((instance, opts, done) => {
 | |
|     const func = () => {}
 | |
|     instance.decorate('test', func)
 | |
|     instance.decorateRequest('test', func)
 | |
|     instance.decorateReply('test', func)
 | |
| 
 | |
|     t.equal(instance.test, func)
 | |
|     t.equal(instance[symbols.kRequest].prototype.test, func)
 | |
|     t.equal(instance[symbols.kReply].prototype.test, func)
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(() => {
 | |
|     t.equal(fastify.test, rootFunc)
 | |
|     t.equal(fastify[symbols.kRequest].prototype.test, rootFunc)
 | |
|     t.equal(fastify[symbols.kReply].prototype.test, rootFunc)
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('a decorator should addSchema to all the encapsulated tree', t => {
 | |
|   t.plan(1)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   const decorator = function (instance, opts, done) {
 | |
|     instance.decorate('decoratorAddSchema', function (whereAddTheSchema) {
 | |
|       instance.addSchema({
 | |
|         $id: 'schema',
 | |
|         type: 'string'
 | |
|       })
 | |
|     })
 | |
|     done()
 | |
|   }
 | |
| 
 | |
|   fastify.register(fp(decorator))
 | |
| 
 | |
|   fastify.register(function (instance, opts, done) {
 | |
|     instance.register((subInstance, opts, done) => {
 | |
|       subInstance.decoratorAddSchema()
 | |
|       done()
 | |
|     })
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.ready(t.error)
 | |
| })
 | |
| 
 | |
| test('after can access to a decorated instance and previous plugin decoration', t => {
 | |
|   t.plan(11)
 | |
|   const TEST_VALUE = {}
 | |
|   const OTHER_TEST_VALUE = {}
 | |
|   const NEW_TEST_VALUE = {}
 | |
| 
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.register(fp(function (instance, options, done) {
 | |
|     instance.decorate('test', TEST_VALUE)
 | |
| 
 | |
|     done()
 | |
|   })).after(function (err, instance, done) {
 | |
|     t.error(err)
 | |
|     t.equal(instance.test, TEST_VALUE)
 | |
| 
 | |
|     instance.decorate('test2', OTHER_TEST_VALUE)
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.register(fp(function (instance, options, done) {
 | |
|     t.equal(instance.test, TEST_VALUE)
 | |
|     t.equal(instance.test2, OTHER_TEST_VALUE)
 | |
| 
 | |
|     instance.decorate('test3', NEW_TEST_VALUE)
 | |
| 
 | |
|     done()
 | |
|   })).after(function (err, instance, done) {
 | |
|     t.error(err)
 | |
|     t.equal(instance.test, TEST_VALUE)
 | |
|     t.equal(instance.test2, OTHER_TEST_VALUE)
 | |
|     t.equal(instance.test3, NEW_TEST_VALUE)
 | |
| 
 | |
|     done()
 | |
|   })
 | |
| 
 | |
|   fastify.get('/', function (req, res) {
 | |
|     t.equal(this.test, TEST_VALUE)
 | |
|     t.equal(this.test2, OTHER_TEST_VALUE)
 | |
|     res.send({})
 | |
|   })
 | |
| 
 | |
|   fastify.inject('/')
 | |
|     .then(response => {
 | |
|       t.equal(response.statusCode, 200)
 | |
|     })
 | |
| })
 | |
| 
 | |
| test('decorate* should throw if called after ready', async t => {
 | |
|   t.plan(6)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.get('/', (request, reply) => {
 | |
|     reply.send({
 | |
|       hello: 'world'
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   await fastify.listen(0)
 | |
|   try {
 | |
|     fastify.decorate('test', true)
 | |
|     t.fail('should not decorate')
 | |
|   } catch (err) {
 | |
|     t.same(err.code, 'FST_ERR_DEC_AFTER_START')
 | |
|     t.same(err.message, "The decorator 'test' has been added after start!")
 | |
|   }
 | |
|   try {
 | |
|     fastify.decorateRequest('test', true)
 | |
|     t.fail('should not decorate')
 | |
|   } catch (e) {
 | |
|     t.same(e.code, 'FST_ERR_DEC_AFTER_START')
 | |
|     t.same(e.message, "The decorator 'test' has been added after start!")
 | |
|   }
 | |
|   try {
 | |
|     fastify.decorateReply('test', true)
 | |
|     t.fail('should not decorate')
 | |
|   } catch (e) {
 | |
|     t.same(e.code, 'FST_ERR_DEC_AFTER_START')
 | |
|     t.same(e.message, "The decorator 'test' has been added after start!")
 | |
|   }
 | |
|   await fastify.close()
 | |
| })
 | |
| 
 | |
| test('decorate* should emit warning if an array is passed', t => {
 | |
|   t.plan(2)
 | |
|   function onWarning (code, name) {
 | |
|     t.equal(name, 'test_array')
 | |
|     t.equal(code, 'FSTDEP006')
 | |
|   }
 | |
|   const warning = {
 | |
|     emit: onWarning
 | |
|   }
 | |
| 
 | |
|   const decorate = proxyquire('../lib/decorate', { './warnings': warning })
 | |
|   const fastify = proxyquire('..', { './lib/decorate.js': decorate })()
 | |
|   fastify.decorateRequest('test_array', [])
 | |
| })
 | |
| 
 | |
| test('decorate* should emit warning if object type is passed', t => {
 | |
|   t.plan(2)
 | |
|   function onWarning (code, name) {
 | |
|     t.equal(name, 'test_object')
 | |
|     t.equal(code, 'FSTDEP006')
 | |
|   }
 | |
|   const warning = {
 | |
|     emit: onWarning
 | |
|   }
 | |
| 
 | |
|   const decorate = proxyquire('../lib/decorate', { './warnings': warning })
 | |
|   const fastify = proxyquire('..', { './lib/decorate.js': decorate })()
 | |
|   fastify.decorateRequest('test_object', { foo: 'bar' })
 | |
| })
 | |
| 
 | |
| test('decorate* should not emit warning if object with getter/setter is passed', t => {
 | |
|   function onWarning (warning) {
 | |
|     t.fail('Should not call a warn')
 | |
|   }
 | |
|   const warning = {
 | |
|     emit: onWarning
 | |
|   }
 | |
|   const decorate = proxyquire('../lib/decorate', { './warnings': warning })
 | |
|   const fastify = proxyquire('..', { './lib/decorate.js': decorate })()
 | |
| 
 | |
|   fastify.decorateRequest('test_getter_setter', {
 | |
|     setter (val) {
 | |
|       this._ = val
 | |
|     },
 | |
|     getter () {
 | |
|       return 'a getter'
 | |
|     }
 | |
|   })
 | |
|   t.end('Done')
 | |
| })
 | |
| 
 | |
| test('decorate* should not emit warning if string,bool,numbers are passed', t => {
 | |
|   function onWarning (warning) {
 | |
|     t.fail('Should not call a warn')
 | |
|   }
 | |
|   const warning = {
 | |
|     emit: onWarning
 | |
|   }
 | |
| 
 | |
|   const decorate = proxyquire('../lib/decorate', { './warnings': warning })
 | |
|   const fastify = proxyquire('..', { './lib/decorate.js': decorate })()
 | |
| 
 | |
|   fastify.decorateRequest('test_str', 'foo')
 | |
|   fastify.decorateRequest('test_bool', true)
 | |
|   fastify.decorateRequest('test_number', 42)
 | |
|   fastify.decorateRequest('test_null', null)
 | |
|   fastify.decorateRequest('test_undefined', undefined)
 | |
|   fastify.decorateReply('test_str', 'foo')
 | |
|   fastify.decorateReply('test_bool', true)
 | |
|   fastify.decorateReply('test_number', 42)
 | |
|   fastify.decorateReply('test_null', null)
 | |
|   fastify.decorateReply('test_undefined', undefined)
 | |
|   t.end('Done')
 | |
| })
 | |
| 
 | |
| test('Request/reply decorators should be able to access the server instance', async t => {
 | |
|   t.plan(6)
 | |
| 
 | |
|   const server = require('..')({ logger: false })
 | |
|   server.decorateRequest('assert', rootAssert)
 | |
|   server.decorateReply('assert', rootAssert)
 | |
| 
 | |
|   server.get('/root-assert', async (req, rep) => {
 | |
|     req.assert()
 | |
|     rep.assert()
 | |
|     return 'done'
 | |
|   })
 | |
| 
 | |
|   server.register(async instance => {
 | |
|     instance.decorateRequest('assert', nestedAssert)
 | |
|     instance.decorateReply('assert', nestedAssert)
 | |
|     instance.decorate('foo', 'bar')
 | |
| 
 | |
|     instance.get('/nested-assert', async (req, rep) => {
 | |
|       req.assert()
 | |
|       rep.assert()
 | |
|       return 'done'
 | |
|     })
 | |
|   })
 | |
| 
 | |
|   await server.inject({ method: 'GET', url: '/root-assert' })
 | |
|   await server.inject({ method: 'GET', url: '/nested-assert' })
 | |
| 
 | |
|   // ----
 | |
|   function rootAssert () {
 | |
|     t.equal(this.server, server)
 | |
|   }
 | |
| 
 | |
|   function nestedAssert () {
 | |
|     t.not(this.server, server)
 | |
|     t.equal(this.server.foo, 'bar')
 | |
|   }
 | |
| })
 | |
| 
 | |
| test('plugin required decorators', async t => {
 | |
|   const plugin1 = fp(
 | |
|     async (instance) => {
 | |
|       instance.decorateRequest('someThing', null)
 | |
| 
 | |
|       instance.addHook('onRequest', async (request, reply) => {
 | |
|         request.someThing = 'hello'
 | |
|       })
 | |
|     },
 | |
|     {
 | |
|       name: 'custom-plugin-one',
 | |
|       fastify: '3.x'
 | |
|     }
 | |
|   )
 | |
| 
 | |
|   const plugin2 = fp(
 | |
|     async () => {
 | |
|       // nothing
 | |
|     },
 | |
|     {
 | |
|       name: 'custom-plugin-two',
 | |
|       fastify: '3.x',
 | |
|       dependencies: ['custom-plugin-one'],
 | |
|       decorators: {
 | |
|         request: ['someThing']
 | |
|       }
 | |
|     }
 | |
|   )
 | |
| 
 | |
|   const app = Fastify()
 | |
|   app.register(plugin1)
 | |
|   app.register(plugin2)
 | |
|   await app.ready()
 | |
| })
 | |
| 
 | |
| test('decorateRequest/decorateReply empty string', t => {
 | |
|   t.plan(7)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.decorateRequest('test', '')
 | |
|   fastify.decorateReply('test2', '')
 | |
|   fastify.get('/yes', (req, reply) => {
 | |
|     t.equal(req.test, '')
 | |
|     t.equal(reply.test2, '')
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
|   t.teardown(fastify.close.bind(fastify))
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest/decorateReply is undefined', t => {
 | |
|   t.plan(7)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.decorateRequest('test', undefined)
 | |
|   fastify.decorateReply('test2', undefined)
 | |
|   fastify.get('/yes', (req, reply) => {
 | |
|     t.equal(req.test, undefined)
 | |
|     t.equal(reply.test2, undefined)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
|   t.teardown(fastify.close.bind(fastify))
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest/decorateReply is not set to a value', t => {
 | |
|   t.plan(7)
 | |
|   const fastify = Fastify()
 | |
| 
 | |
|   fastify.decorateRequest('test')
 | |
|   fastify.decorateReply('test2')
 | |
|   fastify.get('/yes', (req, reply) => {
 | |
|     t.equal(req.test, undefined)
 | |
|     t.equal(reply.test2, undefined)
 | |
|     reply.send({ hello: 'world' })
 | |
|   })
 | |
|   t.teardown(fastify.close.bind(fastify))
 | |
| 
 | |
|   fastify.listen(0, err => {
 | |
|     t.error(err)
 | |
|     fastify.server.unref()
 | |
| 
 | |
|     sget({
 | |
|       method: 'GET',
 | |
|       url: 'http://localhost:' + fastify.server.address().port + '/yes'
 | |
|     }, (err, response, body) => {
 | |
|       t.error(err)
 | |
|       t.equal(response.statusCode, 200)
 | |
|       t.equal(response.headers['content-length'], '' + body.length)
 | |
|       t.same(JSON.parse(body), { hello: 'world' })
 | |
|     })
 | |
|   })
 | |
| })
 | |
| 
 | |
| test('decorateRequest with dependencies', (t) => {
 | |
|   t.plan(2)
 | |
|   const app = Fastify()
 | |
| 
 | |
|   const decorator1 = 'bar'
 | |
|   const decorator2 = 'foo'
 | |
| 
 | |
|   app.decorate('decorator1', decorator1)
 | |
|   app.decorateRequest('decorator1', decorator1)
 | |
| 
 | |
|   if (
 | |
|     app.hasDecorator('decorator1') &&
 | |
|     app.hasRequestDecorator('decorator1')
 | |
|   ) {
 | |
|     t.doesNotThrow(() => app.decorateRequest('decorator2', decorator2, ['decorator1']))
 | |
|     t.ok(app.hasRequestDecorator('decorator2'))
 | |
|   }
 | |
| })
 | |
| 
 | |
| test('decorateRequest with dependencies (functions)', (t) => {
 | |
|   t.plan(2)
 | |
|   const app = Fastify()
 | |
| 
 | |
|   const decorator1 = () => 'bar'
 | |
|   const decorator2 = () => 'foo'
 | |
| 
 | |
|   app.decorate('decorator1', decorator1)
 | |
|   app.decorateRequest('decorator1', decorator1)
 | |
| 
 | |
|   if (
 | |
|     app.hasDecorator('decorator1') &&
 | |
|     app.hasRequestDecorator('decorator1')
 | |
|   ) {
 | |
|     t.doesNotThrow(() => app.decorateRequest('decorator2', decorator2, ['decorator1']))
 | |
|     t.ok(app.hasRequestDecorator('decorator2'))
 | |
|   }
 | |
| })
 |