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.

162 lines
3.8 KiB

'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 { once } = require('events')
const pump = util.promisify(stream.pipeline)
const filePath = path.join(__dirname, '../README.md')
test('should be able to attach all parsed fields and files and make it accessible through "req.body"', async function (t) {
t.plan(6)
const fastify = Fastify()
t.teardown(fastify.close.bind(fastify))
fastify.register(multipart, { attachFieldsToBody: true })
const original = fs.readFileSync(filePath, 'utf8')
fastify.post('/', 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()
})
await fastify.listen(0)
// 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)
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')
}
const [res] = await once(req, 'response')
t.equal(res.statusCode, 200)
res.resume()
await once(res, 'end')
t.pass('res ended successfully')
})
test('should be able to define a custom "onFile" handler', async function (t) {
t.plan(7)
const fastify = Fastify()
t.teardown(fastify.close.bind(fastify))
async function onFile (part) {
t.pass('custom onFile handler')
await part.toBuffer()
}
fastify.register(multipart, { attachFieldsToBody: true, onFile })
const original = fs.readFileSync(filePath, 'utf8')
fastify.post('/', 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()
})
await fastify.listen(0)
// 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)
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')
}
const [res] = await once(req, 'response')
t.equal(res.statusCode, 200)
res.resume()
await once(res, 'end')
t.pass('res ended successfully')
})
test('should not process requests with content-type other than multipart', function (t) {
t.plan(3)
const fastify = Fastify()
t.teardown(fastify.close.bind(fastify))
fastify.register(multipart, { attachFieldsToBody: true })
fastify.post('/', async function (req) {
return { hello: req.body.name }
})
fastify.listen(0, function () {
const opts = {
protocol: 'http:',
hostname: 'localhost',
port: fastify.server.address().port,
path: '/',
headers: { 'content-type': 'application/json' },
method: 'POST'
}
const req = http.request(opts, (res) => {
t.equal(res.statusCode, 200)
res.on('data', function (data) {
t.equal(JSON.parse(data).hello, 'world')
})
res.resume()
res.on('end', () => {
t.pass('res ended successfully')
})
})
req.end(JSON.stringify({ name: 'world' }))
})
})