mirror of
https://github.com/denoland/deno.git
synced 2024-10-30 09:08:00 -04:00
521 lines
16 KiB
TypeScript
521 lines
16 KiB
TypeScript
// 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
|
|
|
|
import { TextProtoReader } from "../textproto/mod.ts";
|
|
import {
|
|
assert,
|
|
assertEquals,
|
|
assertNotEOF,
|
|
assertStrContains,
|
|
assertThrowsAsync,
|
|
} from "../testing/asserts.ts";
|
|
import { Response, ServerRequest, Server, serve } from "./server.ts";
|
|
import { BufReader, BufWriter } from "../io/bufio.ts";
|
|
import { delay } from "../util/async.ts";
|
|
import { encode, decode } from "../strings/mod.ts";
|
|
import { mockConn } from "./mock.ts";
|
|
|
|
const { Buffer, test } = Deno;
|
|
|
|
interface ResponseTest {
|
|
response: Response;
|
|
raw: string;
|
|
}
|
|
|
|
const responseTests: ResponseTest[] = [
|
|
// Default response
|
|
{
|
|
response: {},
|
|
raw: "HTTP/1.1 200 OK\r\n" + "content-length: 0" + "\r\n\r\n",
|
|
},
|
|
// Empty body with status
|
|
{
|
|
response: {
|
|
status: 404,
|
|
},
|
|
raw: "HTTP/1.1 404 Not Found\r\n" + "content-length: 0" + "\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<void> {
|
|
for (const testCase of responseTests) {
|
|
const buf = new Buffer();
|
|
const bufw = new BufWriter(buf);
|
|
const request = new ServerRequest();
|
|
request.w = bufw;
|
|
|
|
request.conn = mockConn();
|
|
|
|
await request.respond(testCase.response);
|
|
assertEquals(buf.toString(), testCase.raw);
|
|
await request.done;
|
|
}
|
|
});
|
|
|
|
test(function requestContentLength(): void {
|
|
// Has content length
|
|
{
|
|
const req = new ServerRequest();
|
|
req.headers = new Headers();
|
|
req.headers.set("content-length", "5");
|
|
const buf = new Buffer(encode("Hello"));
|
|
req.r = new BufReader(buf);
|
|
assertEquals(req.contentLength, 5);
|
|
}
|
|
// No content length
|
|
{
|
|
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(encode(chunksData));
|
|
req.r = new BufReader(buf);
|
|
assertEquals(req.contentLength, null);
|
|
}
|
|
});
|
|
|
|
interface TotalReader extends Deno.Reader {
|
|
total: number;
|
|
}
|
|
function totalReader(r: Deno.Reader): TotalReader {
|
|
let _total = 0;
|
|
async function read(p: Uint8Array): Promise<number | Deno.EOF> {
|
|
const result = await r.read(p);
|
|
if (typeof result === "number") {
|
|
_total += result;
|
|
}
|
|
return result;
|
|
}
|
|
return {
|
|
read,
|
|
get total(): number {
|
|
return _total;
|
|
},
|
|
};
|
|
}
|
|
test(async function requestBodyWithContentLength(): Promise<void> {
|
|
{
|
|
const req = new ServerRequest();
|
|
req.headers = new Headers();
|
|
req.headers.set("content-length", "5");
|
|
const buf = new Buffer(encode("Hello"));
|
|
req.r = new BufReader(buf);
|
|
const body = decode(await Deno.readAll(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(encode(longText));
|
|
req.r = new BufReader(buf);
|
|
const body = decode(await Deno.readAll(req.body));
|
|
assertEquals(body, longText);
|
|
}
|
|
// Handler ignored to consume body
|
|
});
|
|
test("ServerRequest.finalize() should consume unread body / content-length", async () => {
|
|
const text = "deno.land";
|
|
const req = new ServerRequest();
|
|
req.headers = new Headers();
|
|
req.headers.set("content-length", "" + text.length);
|
|
const tr = totalReader(new Buffer(encode(text)));
|
|
req.r = new BufReader(tr);
|
|
req.w = new BufWriter(new Buffer());
|
|
await req.respond({ status: 200, body: "ok" });
|
|
assertEquals(tr.total, 0);
|
|
await req.finalize();
|
|
assertEquals(tr.total, text.length);
|
|
});
|
|
test("ServerRequest.finalize() should consume unread body / chunked, trailers", async () => {
|
|
const text = [
|
|
"5",
|
|
"Hello",
|
|
"4",
|
|
"Deno",
|
|
"0",
|
|
"",
|
|
"deno: land",
|
|
"node: js",
|
|
"",
|
|
"",
|
|
].join("\r\n");
|
|
const req = new ServerRequest();
|
|
req.headers = new Headers();
|
|
req.headers.set("transfer-encoding", "chunked");
|
|
req.headers.set("trailer", "deno,node");
|
|
const body = encode(text);
|
|
const tr = totalReader(new Buffer(body));
|
|
req.r = new BufReader(tr);
|
|
req.w = new BufWriter(new Buffer());
|
|
await req.respond({ status: 200, body: "ok" });
|
|
assertEquals(tr.total, 0);
|
|
assertEquals(req.headers.has("trailer"), true);
|
|
assertEquals(req.headers.has("deno"), false);
|
|
assertEquals(req.headers.has("node"), false);
|
|
await req.finalize();
|
|
assertEquals(tr.total, body.byteLength);
|
|
assertEquals(req.headers.has("trailer"), false);
|
|
assertEquals(req.headers.get("deno"), "land");
|
|
assertEquals(req.headers.get("node"), "js");
|
|
});
|
|
test(async function requestBodyWithTransferEncoding(): Promise<void> {
|
|
{
|
|
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(encode(chunksData));
|
|
req.r = new BufReader(buf);
|
|
const body = decode(await Deno.readAll(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(encode(chunksData));
|
|
req.r = new BufReader(buf);
|
|
const body = decode(await Deno.readAll(req.body));
|
|
assertEquals(body, longText);
|
|
}
|
|
});
|
|
|
|
test(async function requestBodyReaderWithContentLength(): Promise<void> {
|
|
{
|
|
const shortText = "Hello";
|
|
const req = new ServerRequest();
|
|
req.headers = new Headers();
|
|
req.headers.set("content-length", "" + shortText.length);
|
|
const buf = new Buffer(encode(shortText));
|
|
req.r = new BufReader(buf);
|
|
const readBuf = new Uint8Array(6);
|
|
let offset = 0;
|
|
while (offset < shortText.length) {
|
|
const nread = await req.body.read(readBuf);
|
|
assertNotEOF(nread);
|
|
const s = decode(readBuf.subarray(0, nread as number));
|
|
assertEquals(shortText.substr(offset, nread as number), s);
|
|
offset += nread as number;
|
|
}
|
|
const nread = await req.body.read(readBuf);
|
|
assertEquals(nread, Deno.EOF);
|
|
}
|
|
|
|
// Larger than given 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(encode(longText));
|
|
req.r = new BufReader(buf);
|
|
const readBuf = new Uint8Array(1000);
|
|
let offset = 0;
|
|
while (offset < longText.length) {
|
|
const nread = await req.body.read(readBuf);
|
|
assertNotEOF(nread);
|
|
const s = decode(readBuf.subarray(0, nread as number));
|
|
assertEquals(longText.substr(offset, nread as number), s);
|
|
offset += nread as number;
|
|
}
|
|
const nread = await req.body.read(readBuf);
|
|
assertEquals(nread, Deno.EOF);
|
|
}
|
|
});
|
|
|
|
test(async function requestBodyReaderWithTransferEncoding(): Promise<void> {
|
|
{
|
|
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(encode(chunksData));
|
|
req.r = new BufReader(buf);
|
|
const readBuf = new Uint8Array(6);
|
|
let offset = 0;
|
|
while (offset < shortText.length) {
|
|
const nread = await req.body.read(readBuf);
|
|
assertNotEOF(nread);
|
|
const s = decode(readBuf.subarray(0, nread as number));
|
|
assertEquals(shortText.substr(offset, nread as number), s);
|
|
offset += nread as number;
|
|
}
|
|
const nread = await req.body.read(readBuf);
|
|
assertEquals(nread, Deno.EOF);
|
|
}
|
|
|
|
// 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(encode(chunksData));
|
|
req.r = new BufReader(buf);
|
|
const readBuf = new Uint8Array(1000);
|
|
let offset = 0;
|
|
while (offset < longText.length) {
|
|
const nread = await req.body.read(readBuf);
|
|
assertNotEOF(nread);
|
|
const s = decode(readBuf.subarray(0, nread as number));
|
|
assertEquals(longText.substr(offset, nread as number), s);
|
|
offset += nread as number;
|
|
}
|
|
const nread = await req.body.read(readBuf);
|
|
assertEquals(nread, Deno.EOF);
|
|
}
|
|
});
|
|
|
|
test({
|
|
name: "destroyed connection",
|
|
// FIXME(bartlomieju): hangs on windows, cause can't do `Deno.kill`
|
|
ignore: true,
|
|
fn: async (): Promise<void> => {
|
|
// Runs a simple server as another process
|
|
const p = Deno.run({
|
|
cmd: [Deno.execPath(), "--allow-net", "http/testdata/simple_server.ts"],
|
|
stdout: "piped",
|
|
});
|
|
|
|
let serverIsRunning = true;
|
|
const statusPromise = p
|
|
.status()
|
|
.then((): void => {
|
|
serverIsRunning = false;
|
|
})
|
|
.catch((_): void => {}); // Ignores the error when closing the process.
|
|
|
|
try {
|
|
const r = new TextProtoReader(new BufReader(p.stdout!));
|
|
const s = await r.readLine();
|
|
assert(s !== Deno.EOF && s.includes("server listening"));
|
|
await delay(100);
|
|
// Reqeusts to the server and immediately closes the connection
|
|
const conn = await Deno.connect({ port: 4502 });
|
|
await conn.write(new TextEncoder().encode("GET / HTTP/1.0\n\n"));
|
|
conn.close();
|
|
// Waits for the server to handle the above (broken) request
|
|
await delay(100);
|
|
assert(serverIsRunning);
|
|
} finally {
|
|
// Stops the sever and allows `p.status()` promise to resolve
|
|
Deno.kill(p.pid, Deno.Signal.SIGKILL);
|
|
await statusPromise;
|
|
p.stdout!.close();
|
|
p.close();
|
|
}
|
|
},
|
|
});
|
|
|
|
test({
|
|
name: "serveTLS",
|
|
// FIXME(bartlomieju): hangs on windows, cause can't do `Deno.kill`
|
|
ignore: true,
|
|
fn: async (): Promise<void> => {
|
|
// Runs a simple server as another process
|
|
const p = Deno.run({
|
|
cmd: [
|
|
Deno.execPath(),
|
|
"--allow-net",
|
|
"--allow-read",
|
|
"http/testdata/simple_https_server.ts",
|
|
],
|
|
stdout: "piped",
|
|
});
|
|
|
|
let serverIsRunning = true;
|
|
const statusPromise = p
|
|
.status()
|
|
.then((): void => {
|
|
serverIsRunning = false;
|
|
})
|
|
.catch((_): void => {}); // Ignores the error when closing the process.
|
|
|
|
try {
|
|
const r = new TextProtoReader(new BufReader(p.stdout!));
|
|
const s = await r.readLine();
|
|
assert(
|
|
s !== Deno.EOF && s.includes("server listening"),
|
|
"server must be started"
|
|
);
|
|
// Requests to the server and immediately closes the connection
|
|
const conn = await Deno.connectTLS({
|
|
hostname: "localhost",
|
|
port: 4503,
|
|
certFile: "http/testdata/tls/RootCA.pem",
|
|
});
|
|
await Deno.writeAll(
|
|
conn,
|
|
new TextEncoder().encode("GET / HTTP/1.0\r\n\r\n")
|
|
);
|
|
const res = new Uint8Array(100);
|
|
const nread = assertNotEOF(await conn.read(res));
|
|
conn.close();
|
|
const resStr = new TextDecoder().decode(res.subarray(0, nread));
|
|
assert(resStr.includes("Hello HTTPS"));
|
|
assert(serverIsRunning);
|
|
} finally {
|
|
// Stops the sever and allows `p.status()` promise to resolve
|
|
Deno.kill(p.pid, Deno.Signal.SIGKILL);
|
|
await statusPromise;
|
|
p.stdout!.close();
|
|
p.close();
|
|
}
|
|
},
|
|
});
|
|
|
|
test("close server while iterating", async (): Promise<void> => {
|
|
const server = serve(":8123");
|
|
const nextWhileClosing = server[Symbol.asyncIterator]().next();
|
|
server.close();
|
|
assertEquals(await nextWhileClosing, { value: undefined, done: true });
|
|
|
|
const nextAfterClosing = server[Symbol.asyncIterator]().next();
|
|
assertEquals(await nextAfterClosing, { value: undefined, done: true });
|
|
});
|
|
|
|
test({
|
|
name: "[http] close server while connection is open",
|
|
async fn(): Promise<void> {
|
|
async function iteratorReq(server: Server): Promise<void> {
|
|
for await (const req of server) {
|
|
await req.respond({ body: new TextEncoder().encode(req.url) });
|
|
}
|
|
}
|
|
|
|
const server = serve(":8123");
|
|
const p = iteratorReq(server);
|
|
const conn = await Deno.connect({ hostname: "127.0.0.1", port: 8123 });
|
|
await Deno.writeAll(
|
|
conn,
|
|
new TextEncoder().encode("GET /hello HTTP/1.1\r\n\r\n")
|
|
);
|
|
const res = new Uint8Array(100);
|
|
const nread = await conn.read(res);
|
|
assert(nread !== Deno.EOF);
|
|
const resStr = new TextDecoder().decode(res.subarray(0, nread));
|
|
assertStrContains(resStr, "/hello");
|
|
server.close();
|
|
await p;
|
|
// Client connection should still be open, verify that
|
|
// it's visible in resource table.
|
|
const resources = Deno.resources();
|
|
assertEquals(resources[conn.rid], "tcpStream");
|
|
conn.close();
|
|
},
|
|
});
|
|
|
|
test({
|
|
name: "respond error closes connection",
|
|
async fn(): Promise<void> {
|
|
const serverRoutine = async (): Promise<void> => {
|
|
const server = serve(":8124");
|
|
// @ts-ignore
|
|
for await (const req of server) {
|
|
await assertThrowsAsync(async () => {
|
|
await req.respond({
|
|
status: 12345,
|
|
body: new TextEncoder().encode("Hello World"),
|
|
});
|
|
}, Deno.errors.InvalidData);
|
|
// The connection should be destroyed
|
|
assert(!(req.conn.rid in Deno.resources()));
|
|
server.close();
|
|
}
|
|
};
|
|
const p = serverRoutine();
|
|
const conn = await Deno.connect({
|
|
hostname: "127.0.0.1",
|
|
port: 8124,
|
|
});
|
|
await Deno.writeAll(
|
|
conn,
|
|
new TextEncoder().encode("GET / HTTP/1.1\r\n\r\n")
|
|
);
|
|
conn.close();
|
|
await p;
|
|
},
|
|
});
|