From 98ca6181a0390450b1e29c277d4670634695ce85 Mon Sep 17 00:00:00 2001
From: Szymon Marczak <36894700+szmarczak@users.noreply.github.com>
Date: Sun, 24 Apr 2022 22:04:02 +0200
Subject: [PATCH] Fix accidental test deletion

---
 test/progress.ts | 155 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 155 insertions(+)

diff --git a/test/progress.ts b/test/progress.ts
index 4ab10563a..279ae3a22 100644
--- a/test/progress.ts
+++ b/test/progress.ts
@@ -1,9 +1,13 @@
 import {Buffer} from 'buffer';
+import {promisify} from 'util';
 import stream from 'stream';
+import fs from 'fs';
 // @ts-expect-error Fails to find slow-stream/index.d.ts
 import SlowStream from 'slow-stream';
 import toReadableStream from 'to-readable-stream';
 import getStream from 'get-stream';
+import FormData from 'form-data';
+import tempy from 'tempy';
 import is from '@sindresorhus/is';
 import test, {ExecutionContext} from 'ava';
 import {Handler} from 'express';
@@ -41,6 +45,24 @@ const checkEvents = (t: ExecutionContext, events: Progress[], bodySize?: number)
 
 const file = Buffer.alloc(1024 * 1024 * 2);
 
+const downloadEndpoint: Handler = (_request, response) => {
+	response.setHeader('content-length', file.length);
+
+	stream.pipeline(
+		toReadableStream(file),
+		new SlowStream({maxWriteInterval: 50}),
+		response,
+		() => {
+			response.end();
+		},
+	);
+};
+
+const noTotalEndpoint: Handler = (_request, response) => {
+	response.write('hello');
+	response.end();
+};
+
 const uploadEndpoint: Handler = (request, response) => {
 	stream.pipeline(
 		request,
@@ -51,6 +73,118 @@ const uploadEndpoint: Handler = (request, response) => {
 	);
 };
 
+test('download progress', withServer, async (t, server, got) => {
+	server.get('/', downloadEndpoint);
+
+	const events: Progress[] = [];
+
+	const {body} = await got({responseType: 'buffer'})
+		.on('downloadProgress', event => events.push(event));
+
+	checkEvents(t, events, body.length);
+});
+
+test('download progress - missing total size', withServer, async (t, server, got) => {
+	server.get('/', noTotalEndpoint);
+
+	const events: Progress[] = [];
+
+	await got('').on('downloadProgress', (event: Progress) => events.push(event));
+
+	t.is(events[0]?.total, undefined);
+	checkEvents(t, events);
+});
+
+test('download progress - stream', withServer, async (t, server, got) => {
+	server.get('/', downloadEndpoint);
+
+	const events: Progress[] = [];
+
+	const stream = got.stream({responseType: 'buffer'})
+		.on('downloadProgress', event => events.push(event));
+
+	await getStream(stream);
+
+	checkEvents(t, events, file.length);
+});
+
+test('upload progress - file', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const events: Progress[] = [];
+
+	await got.post({body: file}).on('uploadProgress', (event: Progress) => events.push(event));
+
+	checkEvents(t, events, file.length);
+});
+
+test('upload progress - file stream', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const path = tempy.file();
+	fs.writeFileSync(path, file);
+
+	const {size} = await promisify(fs.stat)(path);
+
+	const events: Progress[] = [];
+
+	await got.post({
+		body: fs.createReadStream(path),
+		headers: {
+			'content-length': size.toString(),
+		},
+	})
+		.on('uploadProgress', (event: Progress) => events.push(event));
+
+	checkEvents(t, events, file.length);
+});
+
+test('upload progress - form data', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const events: Progress[] = [];
+
+	const body = new FormData();
+	body.append('key', 'value');
+	body.append('file', file);
+
+	const size = await promisify(body.getLength.bind(body))();
+
+	await got.post({body}).on('uploadProgress', (event: Progress) => events.push(event));
+
+	checkEvents(t, events, size);
+});
+
+test('upload progress - json', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const body = JSON.stringify({key: 'value'});
+	const size = Buffer.byteLength(body);
+	const events: Progress[] = [];
+
+	await got.post({body}).on('uploadProgress', (event: Progress) => events.push(event));
+
+	checkEvents(t, events, size);
+});
+
+test('upload progress - stream with known body size', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const events: Progress[] = [];
+	const options = {
+		headers: {'content-length': file.length.toString()},
+	};
+
+	const request = got.stream.post(options)
+		.on('uploadProgress', event => events.push(event));
+
+	await getStream(
+		stream.pipeline(toReadableStream(file), request, () => {}),
+	);
+
+	checkEvents(t, events, file.length);
+});
+
 test('upload progress - stream with unknown body size', withServer, async (t, server, got) => {
 	server.post('/', uploadEndpoint);
 
@@ -66,3 +200,24 @@ test('upload progress - stream with unknown body size', withServer, async (t, se
 	t.is(events[0]?.total, undefined);
 	checkEvents(t, events);
 });
+
+test('upload progress - no body', withServer, async (t, server, got) => {
+	server.post('/', uploadEndpoint);
+
+	const events: Progress[] = [];
+
+	await got.post('').on('uploadProgress', (event: Progress) => events.push(event));
+
+	t.deepEqual(events, [
+		{
+			percent: 0,
+			transferred: 0,
+			total: undefined,
+		},
+		{
+			percent: 1,
+			transferred: 0,
+			total: 0,
+		},
+	]);
+});