chore: import setHeaders and onNoMatch suites

lukeed · May 26, 2020 · bdfc9b0 · bdfc9b0
1 parent d248d52
commit bdfc9b0
Showing 1 changed file with 118 additions and 20 deletions.
diff --git a/tests/sirv.js b/tests/sirv.js
@@ -429,10 +429,10 @@ etag.run();
 
 // ---
 
-const gzip = suite('gzip');
+const brotli = suite('brotli');
 
-gzip('should require "Accept-Encoding" match to do anything', async () => {
-	let server = utils.http({ gzip: true });
+brotli('should require "Accept-Encoding" match to do anything', async () => {
+	let server = utils.http({ brotli: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
 	try {
@@ -443,31 +443,31 @@ gzip('should require "Accept-Encoding" match to do anything', async () => {
 		// the `matches` helper assumes wrong mime type
 		let res = await server.send('GET', '/data.js', { headers });
 		assert.is(res.headers['content-type'], 'application/javascript');
-		assert.is(res.data, 'gzip js file\n');
+		assert.is(res.data, 'brotli js file\n');
 		assert.is(res.statusCode, 200);
 	} finally {
 		server.close();
 	}
 });
 
-gzip('should serve prepared `.gz` file of any asset, if found', async () => {
-	let server = utils.http({ gzip: true });
+brotli('should serve prepared `.gz` file of any asset, if found', async () => {
+	let server = utils.http({ brotli: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
 	try {
 		let res1 = await server.send('GET', '/', { headers });
 		assert.is(res1.headers['content-type'], 'text/html');
-		assert.is(res1.data, 'gzip html\n');
+		assert.is(res1.data, 'brotli html\n');
 		assert.is(res1.statusCode, 200);
 
 		let res2 = await server.send('GET', '/bundle.67329.js', { headers });
-		await utils.matches(res2, 200, 'bundle.67329.js', 'utf8'); // no gz
+		await utils.matches(res2, 200, 'bundle.67329.js', 'utf8'); // no brotli
 	} finally {
 		server.close();
 	}
 });
 
-gzip('should defer to brotli when "Accept-Encoding" allows both', async () => {
+brotli('should be preferred when "Accept-Encoding" allows both', async () => {
 	let server = utils.http({ gzip: true, brotli: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
@@ -486,14 +486,14 @@ gzip('should defer to brotli when "Accept-Encoding" allows both', async () => {
 	}
 });
 
-gzip.run();
+brotli.run();
 
 // ---
 
-const brotli = suite('brotli');
+const gzip = suite('gzip');
 
-brotli('should require "Accept-Encoding" match to do anything', async () => {
-	let server = utils.http({ brotli: true });
+gzip('should require "Accept-Encoding" match to do anything', async () => {
+	let server = utils.http({ gzip: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
 	try {
@@ -504,31 +504,31 @@ brotli('should require "Accept-Encoding" match to do anything', async () => {
 		// the `matches` helper assumes wrong mime type
 		let res = await server.send('GET', '/data.js', { headers });
 		assert.is(res.headers['content-type'], 'application/javascript');
-		assert.is(res.data, 'brotli js file\n');
+		assert.is(res.data, 'gzip js file\n');
 		assert.is(res.statusCode, 200);
 	} finally {
 		server.close();
 	}
 });
 
-brotli('should serve prepared `.gz` file of any asset, if found', async () => {
-	let server = utils.http({ brotli: true });
+gzip('should serve prepared `.gz` file of any asset, if found', async () => {
+	let server = utils.http({ gzip: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
 	try {
 		let res1 = await server.send('GET', '/', { headers });
 		assert.is(res1.headers['content-type'], 'text/html');
-		assert.is(res1.data, 'brotli html\n');
+		assert.is(res1.data, 'gzip html\n');
 		assert.is(res1.statusCode, 200);
 
 		let res2 = await server.send('GET', '/bundle.67329.js', { headers });
-		await utils.matches(res2, 200, 'bundle.67329.js', 'utf8'); // no brotli
+		await utils.matches(res2, 200, 'bundle.67329.js', 'utf8'); // no gz
 	} finally {
 		server.close();
 	}
 });
 
-brotli('should be preferred when "Accept-Encoding" allows both', async () => {
+gzip('should defer to brotli when "Accept-Encoding" allows both', async () => {
 	let server = utils.http({ gzip: true, brotli: true });
 	let headers = { 'Accept-Encoding': 'br,gzip' };
 
@@ -547,7 +547,7 @@ brotli('should be preferred when "Accept-Encoding" allows both', async () => {
 	}
 });
 
-brotli.run();
+gzip.run();
 
 // ---
 
@@ -749,3 +749,101 @@ ranges('should throw `416` when range cannot be met (overflow)', async () => {
 });
 
 ranges.run();
+
+// ---
+
+const setHeaders = suite('setHeaders');
+
+setHeaders('should be able to set new response headers', async () => {
+	let server = utils.http({
+		setHeaders(res) {
+			res.setHeader('x-foo', 'bar');
+		}
+	});
+
+	try {
+		let res = await server.send('GET', '/sw.js');
+		assert.is(res.headers['x-foo'], 'bar');
+		assert.is(res.statusCode, 200);
+	} finally {
+		server.close();
+	}
+});
+
+setHeaders('should be able to customize "Content-Type" header', async () => {
+	let server = utils.http({
+		setHeaders(res) {
+			res.setHeader('Content-Type', 'text/foobar');
+		}
+	});
+
+	try {
+		let res = await server.send('GET', '/sw.js');
+		assert.is(res.headers['content-type'], 'text/foobar');
+		assert.is(res.statusCode, 200);
+	} finally {
+		server.close();
+	}
+});
+
+setHeaders('should receive "path" argument', async () => {
+	let server = utils.http({
+		setHeaders(res, path) {
+			res.setHeader('Cache-Control', path === '/sw.js' ? 'private' : 'public');
+		}
+	});
+
+	try {
+		let res1 = await server.send('GET', '/sw.js');
+		assert.is(res1.headers['cache-control'], 'private');
+		await utils.matches(res1, 200, 'sw.js', 'utf8');
+
+		let res2 = await server.send('GET', '/test.svg');
+		assert.is(res2.headers['cache-control'], 'public');
+		await utils.matches(res2, 200, 'test.svg', 'utf8');
+	} finally {
+		server.close();
+	}
+});
+
+setHeaders('should receive "stats" argument', async () => {
+	let server = utils.http({
+		setHeaders(res, path, stats) {
+			res.setHeader('x-filesize', stats.size);
+		}
+	});
+
+	try {
+		let res = await server.send('GET', '/sw.js');
+		let file = await utils.lookup('sw.js', 'utf8');
+		assert.is(res.headers['x-filesize'], String(file.size));
+	} finally {
+		server.close();
+	}
+});
+
+setHeaders.run();
+
+// ---
+
+const onNoMatch = suite('onNoMatch');
+
+onNoMatch('should be called instead of default 404 response', async () => {
+	let server = utils.http({
+		onNoMatch(req, res) {
+			res.setHeader('x-foo', 'bar');
+			res.end('not found');
+		}
+	});
+
+	try {
+		let res = await server.send('GET', '/1234');
+		assert.is(res.headers['x-foo'], 'bar');
+		assert.is(res.data, 'not found');
+		assert.is(res.statusCode, 200);
+	} finally {
+		server.close();
+	}
+});
+
+onNoMatch.run();