strapi/test/response/attachment.js

'use strict';

const request = require('supertest');

const context = require('../helpers/context');

const Koa = require('../..').server;

describe('ctx.attachment([filename])', function () {
  describe('when given a filename', function () {
    it('should set the filename param', function () {
      const ctx = context();
      ctx.attachment('path/to/tobi.png');
      const str = 'attachment; filename="tobi.png"';
      ctx.response.header['content-disposition'].should.equal(str);
    });
  });

  describe('when omitting filename', function () {
    it('should not set filename param', function () {
      const ctx = context();
      ctx.attachment();
      ctx.response.header['content-disposition'].should.equal('attachment');
    });
  });

  describe('when given a no-ascii filename', function () {
    it('should set the encodeURI filename param', function () {
      const ctx = context();
      ctx.attachment('path/to/include-no-ascii-char-中文名-ok.png');
      const str = 'attachment; filename=\"include-no-ascii-char-???-ok.png\"; filename*=UTF-8\'\'include-no-ascii-char-%E4%B8%AD%E6%96%87%E5%90%8D-ok.png';
      ctx.response.header['content-disposition'].should.equal(str);
    });

    it('should work with http client', function (done) {
      const app = new Koa();

      app.use(function * () {
        this.attachment('path/to/include-no-ascii-char-中文名-ok.json');
        this.body = {
          foo: 'bar'
        };
      });

      request(app.listen())
        .get('/')
        .expect('content-disposition', 'attachment; filename="include-no-ascii-char-???-ok.json"; filename*=UTF-8\'\'include-no-ascii-char-%E4%B8%AD%E6%96%87%E5%90%8D-ok.json')
        .expect({
          foo: 'bar'
        })
        .expect(200, done);
    });
  });
});