test-form.js

'use strict'

var http = require('http')
var path = require('path')
var mime = require('mime-types')
var request = require('../index')
var fs = require('fs')
var tape = require('tape')

tape('multipart form append', function (t) {
  var remoteFile = path.join(__dirname, 'googledoodle.jpg')
  var localFile = path.join(__dirname, 'unicycle.jpg')
  var totalLength = null
  var FIELDS = []

  var server = http.createServer(function (req, res) {
    if (req.url === '/file') {
      res.writeHead(200, {'content-type': 'image/jpg', 'content-length': 7187})
      res.end(fs.readFileSync(remoteFile), 'binary')
      return
    }

    t.ok(/multipart\/form-data; boundary=--------------------------\d+/
      .test(req.headers['content-type']))

    // temp workaround
    var data = ''
    req.setEncoding('utf8')

    req.on('data', function (d) {
      data += d
    })

    req.on('end', function () {
      var field
      // check for the fields' traces

      // 1st field : my_field
      field = FIELDS.shift()
      t.ok(data.indexOf('form-data; name="' + field.name + '"') !== -1)
      t.ok(data.indexOf(field.value) !== -1)

      // 2nd field : my_buffer
      field = FIELDS.shift()
      t.ok(data.indexOf('form-data; name="' + field.name + '"') !== -1)
      t.ok(data.indexOf(field.value) !== -1)

      // 3rd field : my_file
      field = FIELDS.shift()
      t.ok(data.indexOf('form-data; name="' + field.name + '"') !== -1)
      t.ok(data.indexOf('; filename="' + path.basename(field.value.path) + '"') !== -1)
      // check for unicycle.jpg traces
      t.ok(data.indexOf('2005:06:21 01:44:12') !== -1)
      t.ok(data.indexOf('Content-Type: ' + mime.lookup(field.value.path)) !== -1)

      // 4th field : remote_file
      field = FIELDS.shift()
      t.ok(data.indexOf('form-data; name="' + field.name + '"') !== -1)
      t.ok(data.indexOf('; filename="' + path.basename(field.value.path) + '"') !== -1)
      // check for http://localhost:nnnn/file traces
      t.ok(data.indexOf('Photoshop ICC') !== -1)
      t.ok(data.indexOf('Content-Type: ' + mime.lookup(remoteFile)) !== -1)

      t.ok(+req.headers['content-length'] === totalLength)

      res.writeHead(200)
      res.end('done')

      t.equal(FIELDS.length, 0)
    })
  })

  server.listen(0, function () {
    var url = 'http://localhost:' + this.address().port
    FIELDS = [
      { name: 'my_field', value: 'my_value' },
      { name: 'my_buffer', value: Buffer.from([1, 2, 3]) },
      { name: 'my_file', value: fs.createReadStream(localFile) },
      { name: 'remote_file', value: request(url + '/file') }
    ]

    var req = request.post(url + '/upload', function (err, res, body) {
      t.equal(err, null)
      t.equal(res.statusCode, 200)
      t.equal(body, 'done')
      server.close(function () {
        t.end()
      })
    })
    var form = req.form()

    FIELDS.forEach(function (field) {
      form.append(field.name, field.value)
    })

    form.getLength(function (err, length) {
      t.equal(err, null)
      totalLength = length
    })
  })
})