denoland-deno/std/examples/chat/server_test.ts

// Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
import { assert, assertEquals } from "../../testing/asserts.ts";
import { TextProtoReader } from "../../textproto/mod.ts";
import { BufReader } from "../../io/bufio.ts";
import { connectWebSocket, WebSocket } from "../../ws/mod.ts";

let server: Deno.Process | undefined;
async function startServer(): Promise<void> {
  server = Deno.run({
    args: [Deno.execPath(), "--allow-net", "--allow-read", "server.ts"],
    cwd: "examples/chat",
    stdout: "piped"
  });
  try {
    assert(server.stdout != null);
    const r = new TextProtoReader(new BufReader(server.stdout));
    const s = await r.readLine();
    assert(s !== Deno.EOF && s.includes("chat server starting"));
  } catch {
    server.close();
  }
}

const { test, build } = Deno;

// TODO: https://github.com/denoland/deno/issues/4108
if (build.os !== "win") {
  test("beforeAll", async () => {
    await startServer();
  });

  test("GET / should serve html", async () => {
    const resp = await fetch("http://127.0.0.1:8080/");
    assertEquals(resp.status, 200);
    assertEquals(resp.headers.get("content-type"), "text/html");
    const html = await resp.body.text();
    assert(html.includes("ws chat example"), "body is ok");
  });

  let ws: WebSocket | undefined;
  test("GET /ws should upgrade conn to ws", async () => {
    ws = await connectWebSocket("http://127.0.0.1:8080/ws");
    const it = ws.receive();
    assertEquals((await it.next()).value, "Connected: [1]");
    ws.send("Hello");
    assertEquals((await it.next()).value, "[1]: Hello");
  });
  test("afterAll", () => {
    server?.close();
    ws?.conn.close();
  });
}