// Copyright 2010 The Go Authors. All rights reserved. // Use of this source code is governed by a BSD-style // license that can be found in the LICENSE file. // Ported from // https://github.com/golang/go/blob/master/src/net/http/responsewrite_test.go const { Buffer } = Deno; import { test, runIfMain } from "../testing/mod.ts"; import { assert, assertEquals } from "../testing/asserts.ts"; import { Response, ServerRequest, writeResponse, readRequest } from "./server.ts"; import { BufReader, BufWriter } from "../io/bufio.ts"; import { StringReader } from "../io/readers.ts"; interface ResponseTest { response: Response; raw: string; } const enc = new TextEncoder(); const dec = new TextDecoder(); type Handler = () => void; const responseTests: ResponseTest[] = [ // Default response { response: {}, raw: "HTTP/1.1 200 OK\r\n" + "\r\n" }, // HTTP/1.1, chunked coding; empty trailer; close { response: { status: 200, body: new Buffer(new TextEncoder().encode("abcdef")) }, raw: "HTTP/1.1 200 OK\r\n" + "transfer-encoding: chunked\r\n\r\n" + "6\r\nabcdef\r\n0\r\n\r\n" } ]; test(async function responseWrite(): Promise { for (const testCase of responseTests) { const buf = new Buffer(); const bufw = new BufWriter(buf); const request = new ServerRequest(); request.w = bufw; await request.respond(testCase.response); assertEquals(buf.toString(), testCase.raw); await request.done; } }); test(async function requestBodyWithContentLength(): Promise { { const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("content-length", "5"); const buf = new Buffer(enc.encode("Hello")); req.r = new BufReader(buf); const body = dec.decode(await req.body()); assertEquals(body, "Hello"); } // Larger than internal buf { const longText = "1234\n".repeat(1000); const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("Content-Length", "5000"); const buf = new Buffer(enc.encode(longText)); req.r = new BufReader(buf); const body = dec.decode(await req.body()); assertEquals(body, longText); } }); test(async function requestBodyWithTransferEncoding(): Promise { { const shortText = "Hello"; const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("transfer-encoding", "chunked"); let chunksData = ""; let chunkOffset = 0; const maxChunkSize = 70; while (chunkOffset < shortText.length) { const chunkSize = Math.min(maxChunkSize, shortText.length - chunkOffset); chunksData += `${chunkSize.toString(16)}\r\n${shortText.substr( chunkOffset, chunkSize )}\r\n`; chunkOffset += chunkSize; } chunksData += "0\r\n\r\n"; const buf = new Buffer(enc.encode(chunksData)); req.r = new BufReader(buf); const body = dec.decode(await req.body()); assertEquals(body, shortText); } // Larger than internal buf { const longText = "1234\n".repeat(1000); const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("transfer-encoding", "chunked"); let chunksData = ""; let chunkOffset = 0; const maxChunkSize = 70; while (chunkOffset < longText.length) { const chunkSize = Math.min(maxChunkSize, longText.length - chunkOffset); chunksData += `${chunkSize.toString(16)}\r\n${longText.substr( chunkOffset, chunkSize )}\r\n`; chunkOffset += chunkSize; } chunksData += "0\r\n\r\n"; const buf = new Buffer(enc.encode(chunksData)); req.r = new BufReader(buf); const body = dec.decode(await req.body()); assertEquals(body, longText); } }); test(async function requestBodyStreamWithContentLength(): Promise { { const shortText = "Hello"; const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("content-length", "" + shortText.length); const buf = new Buffer(enc.encode(shortText)); req.r = new BufReader(buf); const it = await req.bodyStream(); let offset = 0; for await (const chunk of it) { const s = dec.decode(chunk); assertEquals(shortText.substr(offset, s.length), s); offset += s.length; } } // Larger than internal buf { const longText = "1234\n".repeat(1000); const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("Content-Length", "5000"); const buf = new Buffer(enc.encode(longText)); req.r = new BufReader(buf); const it = await req.bodyStream(); let offset = 0; for await (const chunk of it) { const s = dec.decode(chunk); assertEquals(longText.substr(offset, s.length), s); offset += s.length; } } }); test(async function requestBodyStreamWithTransferEncoding(): Promise { { const shortText = "Hello"; const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("transfer-encoding", "chunked"); let chunksData = ""; let chunkOffset = 0; const maxChunkSize = 70; while (chunkOffset < shortText.length) { const chunkSize = Math.min(maxChunkSize, shortText.length - chunkOffset); chunksData += `${chunkSize.toString(16)}\r\n${shortText.substr( chunkOffset, chunkSize )}\r\n`; chunkOffset += chunkSize; } chunksData += "0\r\n\r\n"; const buf = new Buffer(enc.encode(chunksData)); req.r = new BufReader(buf); const it = await req.bodyStream(); let offset = 0; for await (const chunk of it) { const s = dec.decode(chunk); assertEquals(shortText.substr(offset, s.length), s); offset += s.length; } } // Larger than internal buf { const longText = "1234\n".repeat(1000); const req = new ServerRequest(); req.headers = new Headers(); req.headers.set("transfer-encoding", "chunked"); let chunksData = ""; let chunkOffset = 0; const maxChunkSize = 70; while (chunkOffset < longText.length) { const chunkSize = Math.min(maxChunkSize, longText.length - chunkOffset); chunksData += `${chunkSize.toString(16)}\r\n${longText.substr( chunkOffset, chunkSize )}\r\n`; chunkOffset += chunkSize; } chunksData += "0\r\n\r\n"; const buf = new Buffer(enc.encode(chunksData)); req.r = new BufReader(buf); const it = await req.bodyStream(); let offset = 0; for await (const chunk of it) { const s = dec.decode(chunk); assertEquals(longText.substr(offset, s.length), s); offset += s.length; } } }); test(async function writeUint8ArrayResponse(): Promise { const shortText = "Hello"; const body = new TextEncoder().encode(shortText); const res: Response = { body }; const buf = new Deno.Buffer(); await writeResponse(buf, res); const decoder = new TextDecoder("utf-8"); const reader = new BufReader(buf); let line: Uint8Array; line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), "HTTP/1.1 200 OK"); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), `content-length: ${shortText.length}`); line = (await reader.readLine())[0]; assertEquals(line.byteLength, 0); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), shortText); line = (await reader.readLine())[0]; assertEquals(line.byteLength, 0); }); test(async function writeStringReaderResponse(): Promise { const shortText = "Hello"; const body = new StringReader(shortText); const res: Response = { body }; const buf = new Deno.Buffer(); await writeResponse(buf, res); const decoder = new TextDecoder("utf-8"); const reader = new BufReader(buf); let line: Uint8Array; line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), "HTTP/1.1 200 OK"); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), "transfer-encoding: chunked"); line = (await reader.readLine())[0]; assertEquals(line.byteLength, 0); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), shortText.length.toString()); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), shortText); line = (await reader.readLine())[0]; assertEquals(decoder.decode(line), "0"); }); test(async function readRequestError(): Promise { let input = `GET / HTTP/1.1 malformedHeader `; const reader = new BufReader(new StringReader(input)); let err; try { await readRequest(reader); } catch (e) { err = e; } assert(err instanceof Error); assertEquals(err.message, "malformed MIME header line: malformedHeader"); }); // Ported from Go // https://github.com/golang/go/blob/go1.12.5/src/net/http/request_test.go#L377-L443 // TODO(zekth) fix tests test(async function testReadRequestError(): Promise { const testCases = { 0: { in: "GET / HTTP/1.1\r\nheader: foo\r\n\r\n", headers: [{ key: "header", value: "foo" }], err: null }, 1: { in: "GET / HTTP/1.1\r\nheader:foo\r\n", err: "EOF", headers: [] }, 2: { in: "", err: "EOF", headers: [] }, // 3: { // in: "HEAD / HTTP/1.1\r\nContent-Length:4\r\n\r\n", // err: "http: method cannot contain a Content-Length" // }, 4: { in: "HEAD / HTTP/1.1\r\n\r\n", headers: [], err: null } // Multiple Content-Length values should either be // deduplicated if same or reject otherwise // See Issue 16490. // 5: { // in: // "POST / HTTP/1.1\r\nContent-Length: 10\r\nContent-Length: 0\r\n\r\nGopher hey\r\n", // err: "cannot contain multiple Content-Length headers" // }, // 6: { // in: // "POST / HTTP/1.1\r\nContent-Length: 10\r\nContent-Length: 6\r\n\r\nGopher\r\n", // err: "cannot contain multiple Content-Length headers" // }, // 7: { // in: // "PUT / HTTP/1.1\r\nContent-Length: 6 \r\nContent-Length: 6\r\nContent-Length:6\r\n\r\nGopher\r\n", // err: null, // headers: [{ key: "Content-Length", value: "6" }] // }, // 8: { // in: "PUT / HTTP/1.1\r\nContent-Length: 1\r\nContent-Length: 6 \r\n\r\n", // err: "cannot contain multiple Content-Length headers" // }, // 9: { // in: "POST / HTTP/1.1\r\nContent-Length:\r\nContent-Length: 3\r\n\r\n", // err: "cannot contain multiple Content-Length headers" // }, // 10: { // in: "HEAD / HTTP/1.1\r\nContent-Length:0\r\nContent-Length: 0\r\n\r\n", // headers: [{ key: "Content-Length", value: "0" }], // err: null // } }; for (const p in testCases) { const test = testCases[p]; const reader = new BufReader(new StringReader(test.in)); const [req, err] = await readRequest(reader); assertEquals(test.err, err); for (const h of test.headers) { assertEquals(req.headers.get(h.key), h.value); } } }); runIfMain(import.meta);