'use strict'; const fs = require('fs'); const path = require('path'); // Helpers. const { createTestBuilder } = require('../../../../../test/helpers/builder'); const { createStrapiInstance } = require('../../../../../test/helpers/strapi'); const { createContentAPIRequest } = require('../../../../../test/helpers/request'); const builder = createTestBuilder(); const data = { dogs: [] }; let strapi; let rq; const dogModel = { displayName: 'Dog', singularName: 'dog', pluralName: 'dogs', kind: 'collectionType', attributes: { profilePicture: { type: 'media', }, }, }; describe('Upload plugin', () => { beforeAll(async () => { await builder.addContentType(dogModel).build(); strapi = await createStrapiInstance(); rq = await createContentAPIRequest({ strapi }); }); afterAll(async () => { await strapi.destroy(); await builder.cleanup(); }); describe('Create', () => { test('Simple image upload', async () => { const res = await rq({ method: 'POST', url: '/upload', formData: { files: fs.createReadStream(path.join(__dirname, '../utils/rec.jpg')), }, }); expect(res.statusCode).toBe(200); expect(Array.isArray(res.body)).toBe(true); expect(res.body.length).toBe(1); expect(res.body[0]).toEqual( expect.objectContaining({ id: expect.anything(), name: 'rec.jpg', ext: '.jpg', mime: 'image/jpeg', hash: expect.any(String), size: expect.any(Number), width: expect.any(Number), height: expect.any(Number), url: expect.any(String), provider: 'local', }) ); }); test('Rejects when no files are provided', async () => { const res = await rq({ method: 'POST', url: '/upload', formData: {} }); expect(res.statusCode).toBe(400); }); test('Generates a thumbnail on large enough files', async () => { const res = await rq({ method: 'POST', url: '/upload', formData: { files: fs.createReadStream(path.join(__dirname, '../utils/thumbnail_target.png')), }, }); expect(res.statusCode).toBe(200); expect(Array.isArray(res.body)).toBe(true); expect(res.body.length).toBe(1); expect(res.body[0]).toEqual( expect.objectContaining({ id: expect.anything(), name: 'thumbnail_target.png', ext: '.png', mime: 'image/png', hash: expect.any(String), size: expect.any(Number), width: expect.any(Number), height: expect.any(Number), url: expect.any(String), provider: 'local', formats: { thumbnail: { name: 'thumbnail_thumbnail_target.png', hash: expect.any(String), ext: '.png', mime: 'image/png', size: expect.any(Number), width: expect.any(Number), height: expect.any(Number), url: expect.any(String), path: null, }, }, }) ); }); }); describe('Read', () => { test('Get files', async () => { const getRes = await rq({ method: 'GET', url: '/upload/files' }); expect(getRes.statusCode).toBe(200); expect(getRes.body).toEqual( expect.arrayContaining([ expect.objectContaining({ id: expect.anything(), url: expect.any(String), }), ]) ); }); }); describe('Create an entity with a file', () => { test('With an image', async () => { const res = await rq({ method: 'POST', url: '/dogs?populate=*', formData: { data: '{}', 'files.profilePicture': fs.createReadStream(path.join(__dirname, '../utils/rec.jpg')), }, }); expect(res.statusCode).toBe(200); expect(res.body).toMatchObject({ data: { attributes: { profilePicture: { data: { id: expect.anything(), attributes: { provider: 'local', }, }, }, }, id: expect.anything(), }, }); data.dogs.push(res.body); }); test('With a pdf', async () => { const res = await rq({ method: 'POST', url: '/dogs?populate=*', formData: { data: '{}', 'files.profilePicture': fs.createReadStream(path.join(__dirname, '../utils/rec.pdf')), }, }); expect(res.statusCode).toBe(200); expect(res.body).toMatchObject({ data: { attributes: { profilePicture: { data: { id: expect.anything(), attributes: { provider: 'local', }, }, }, }, id: expect.anything(), }, }); data.dogs.push(res.body); }); }); // see https://github.com/strapi/strapi/issues/14125 describe('File relations are correctly removed', () => { test('Update an entity with a file correctly removes the relation between the entity and its old file', async () => { const res = await rq({ method: 'PUT', url: `/dogs/${data.dogs[0].data.id}?populate=*`, formData: { data: '{}', 'files.profilePicture': fs.createReadStream(path.join(__dirname, '../utils/strapi.jpg')), }, }); expect(res.statusCode).toBe(200); expect(res.body.data.attributes.profilePicture.data.id).not.toBe( data.dogs[0].data.attributes.profilePicture.data.id ); data.dogs[0] = res.body; }); test('Update a file with an entity correctly removes the relation between the entity and its old file', async () => { const fileId = data.dogs[1].data.attributes.profilePicture.data.id; await strapi.entityService.update('plugin::upload.file', fileId, { data: { related: [ { id: data.dogs[0].data.id, __type: 'api::dog.dog', __pivot: { field: 'profilePicture' }, }, ], }, }); const res = await rq({ method: 'GET', url: `/dogs/${data.dogs[0].data.id}?populate=*`, }); expect(res.body.data.attributes.profilePicture.data.id).toBe(fileId); data.dogs[0] = res.body; }); }); });