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.
		
		
		
		
		
			
		
			
				
					138 lines
				
				3.2 KiB
			
		
		
			
		
	
	
					138 lines
				
				3.2 KiB
			| 
								 
											3 years ago
										 
									 | 
							
								'use strict'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const util = require('util')
							 | 
						||
| 
								 | 
							
								const test = require('tap').test
							 | 
						||
| 
								 | 
							
								const FormData = require('form-data')
							 | 
						||
| 
								 | 
							
								const Fastify = require('fastify')
							 | 
						||
| 
								 | 
							
								const multipart = require('..')
							 | 
						||
| 
								 | 
							
								const http = require('http')
							 | 
						||
| 
								 | 
							
								const path = require('path')
							 | 
						||
| 
								 | 
							
								const fs = require('fs')
							 | 
						||
| 
								 | 
							
								const stream = require('stream')
							 | 
						||
| 
								 | 
							
								const pump = util.promisify(stream.pipeline)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								const filePath = path.join(__dirname, '../README.md')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								test('should be able to use JSON schema to validate request', function (t) {
							 | 
						||
| 
								 | 
							
								  t.plan(6)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const fastify = Fastify()
							 | 
						||
| 
								 | 
							
								  t.teardown(fastify.close.bind(fastify))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.register(multipart, { attachFieldsToBody: true, sharedSchemaId: '#mySharedSchema' })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const original = fs.readFileSync(filePath, 'utf8')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.post('/', {
							 | 
						||
| 
								 | 
							
								    schema: {
							 | 
						||
| 
								 | 
							
								      body: {
							 | 
						||
| 
								 | 
							
								        type: 'object',
							 | 
						||
| 
								 | 
							
								        required: ['hello', 'upload'],
							 | 
						||
| 
								 | 
							
								        properties: {
							 | 
						||
| 
								 | 
							
								          hello: { $ref: '#mySharedSchema' },
							 | 
						||
| 
								 | 
							
								          upload: { $ref: '#mySharedSchema' }
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }, async function (req, reply) {
							 | 
						||
| 
								 | 
							
								    t.ok(req.isMultipart())
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    t.same(Object.keys(req.body), ['upload', 'hello'])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    const content = await req.body.upload.toBuffer()
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    t.equal(content.toString(), original)
							 | 
						||
| 
								 | 
							
								    t.equal(req.body.hello.value, 'world')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    reply.code(200).send()
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.listen(0, async function () {
							 | 
						||
| 
								 | 
							
								    // request
							 | 
						||
| 
								 | 
							
								    const form = new FormData()
							 | 
						||
| 
								 | 
							
								    const opts = {
							 | 
						||
| 
								 | 
							
								      protocol: 'http:',
							 | 
						||
| 
								 | 
							
								      hostname: 'localhost',
							 | 
						||
| 
								 | 
							
								      port: fastify.server.address().port,
							 | 
						||
| 
								 | 
							
								      path: '/',
							 | 
						||
| 
								 | 
							
								      headers: form.getHeaders(),
							 | 
						||
| 
								 | 
							
								      method: 'POST'
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    const req = http.request(opts, (res) => {
							 | 
						||
| 
								 | 
							
								      t.equal(res.statusCode, 200)
							 | 
						||
| 
								 | 
							
								      res.resume()
							 | 
						||
| 
								 | 
							
								      res.on('end', () => {
							 | 
						||
| 
								 | 
							
								        t.pass('res ended successfully')
							 | 
						||
| 
								 | 
							
								      })
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								    form.append('upload', fs.createReadStream(filePath))
							 | 
						||
| 
								 | 
							
								    form.append('hello', 'world')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    try {
							 | 
						||
| 
								 | 
							
								      await pump(form, req)
							 | 
						||
| 
								 | 
							
								    } catch (error) {
							 | 
						||
| 
								 | 
							
								      t.error(error, 'formData request pump: no err')
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								})
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								test('should throw because JSON schema is invalid', function (t) {
							 | 
						||
| 
								 | 
							
								  t.plan(2)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  const fastify = Fastify()
							 | 
						||
| 
								 | 
							
								  t.teardown(fastify.close.bind(fastify))
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.register(multipart, { attachFieldsToBody: true, sharedSchemaId: '#mySharedSchema' })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.post('/', {
							 | 
						||
| 
								 | 
							
								    schema: {
							 | 
						||
| 
								 | 
							
								      body: {
							 | 
						||
| 
								 | 
							
								        type: 'object',
							 | 
						||
| 
								 | 
							
								        required: ['hello'],
							 | 
						||
| 
								 | 
							
								        properties: {
							 | 
						||
| 
								 | 
							
								          hello: {
							 | 
						||
| 
								 | 
							
								            properties: {
							 | 
						||
| 
								 | 
							
								              value: {
							 | 
						||
| 
								 | 
							
								                type: 'string',
							 | 
						||
| 
								 | 
							
								                enum: ['red']
							 | 
						||
| 
								 | 
							
								              }
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								          }
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								      }
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  }, async function (req, reply) {
							 | 
						||
| 
								 | 
							
								    console.log(req.body)
							 | 
						||
| 
								 | 
							
								    reply.code(200).send()
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								  fastify.listen(0, async function () {
							 | 
						||
| 
								 | 
							
								    // request
							 | 
						||
| 
								 | 
							
								    const form = new FormData()
							 | 
						||
| 
								 | 
							
								    const opts = {
							 | 
						||
| 
								 | 
							
								      protocol: 'http:',
							 | 
						||
| 
								 | 
							
								      hostname: 'localhost',
							 | 
						||
| 
								 | 
							
								      port: fastify.server.address().port,
							 | 
						||
| 
								 | 
							
								      path: '/',
							 | 
						||
| 
								 | 
							
								      headers: form.getHeaders(),
							 | 
						||
| 
								 | 
							
								      method: 'POST'
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    const req = http.request(opts, (res) => {
							 | 
						||
| 
								 | 
							
								      t.equal(res.statusCode, 400)
							 | 
						||
| 
								 | 
							
								      res.resume()
							 | 
						||
| 
								 | 
							
								      res.on('end', () => {
							 | 
						||
| 
								 | 
							
								        t.pass('res ended successfully')
							 | 
						||
| 
								 | 
							
								      })
							 | 
						||
| 
								 | 
							
								    })
							 | 
						||
| 
								 | 
							
								    form.append('hello', 'world')
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    try {
							 | 
						||
| 
								 | 
							
								      await pump(form, req)
							 | 
						||
| 
								 | 
							
								    } catch (error) {
							 | 
						||
| 
								 | 
							
								      t.error(error, 'formData request pump: no err')
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								  })
							 | 
						||
| 
								 | 
							
								})
							 |