123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.import { assertEquals, assertRejects } from "../assert/mod.ts";import {  type ServerSentEventMessage,  ServerSentEventStream,} from "./server_sent_event_stream.ts";
function createStream(  messages: ServerSentEventMessage[],): ReadableStream<string> {  return ReadableStream    .from<ServerSentEventMessage>(messages)    .pipeThrough(new ServerSentEventStream())    .pipeThrough(new TextDecoderStream());}
Deno.test("ServerSentEventStream() enqueues a stringified server-sent event message object", async () => {  const stream = createStream([    {      comment: "a",      event: "b",      data: "c\nd\re\r\nf",      id: "123",      retry: 456,    },    {      comment: "a",    },    {      event: "b",    },    {      data: "c\nd\re\r\nf",    },    {      id: "123",    },    {      id: 123,    },    {      retry: 456,    },  ]);  const clientMessages = await Array.fromAsync(stream);
  assertEquals(    clientMessages,    [      ":a\nevent:b\ndata:c\ndata:d\ndata:e\ndata:f\nid:123\nretry:456\n\n",      ":a\n\n",      "event:b\n\n",      "data:c\ndata:d\ndata:e\ndata:f\n\n",      "id:123\n\n",      "id:123\n\n",      "retry:456\n\n",    ],  );});
Deno.test("ServerSentEventStream() throws if single-line fields contain a newline", async () => {  // Comment  await assertRejects(    async () => await createStream([{ comment: "a\n" }]).getReader().read(),    RangeError,    "`message.comment` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ comment: "a\r" }]).getReader().read(),    RangeError,    "`message.comment` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ comment: "a\n\r" }]).getReader().read(),    RangeError,    "`message.comment` cannot contain a newline",  );
  // Event  await assertRejects(    async () => await createStream([{ event: "a\n" }]).getReader().read(),    RangeError,    "`message.event` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ event: "a\r" }]).getReader().read(),    RangeError,    "`message.event` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ event: "a\n\r" }]).getReader().read(),    RangeError,    "`message.event` cannot contain a newline",  );
  // ID  await assertRejects(    async () => await createStream([{ id: "a\n" }]).getReader().read(),    RangeError,    "`message.id` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ id: "a\r" }]).getReader().read(),    RangeError,    "`message.id` cannot contain a newline",  );
  await assertRejects(    async () => await createStream([{ id: "a\n\r" }]).getReader().read(),    RangeError,    "`message.id` cannot contain a newline",  );});