123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272
// Copyright 2018-2021 the Deno authors. All rights reserved. MIT license.// Based on https://github.com/golang/go/blob/master/src/net/textproto/reader_test.go// Copyright 2009 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.import { BufReader } from "../io/bufio.ts";import { TextProtoReader } from "./mod.ts";import { StringReader } from "../io/readers.ts";import { assert, assertEquals, assertThrows } from "../testing/asserts.ts";
function reader(s: string): TextProtoReader {  return new TextProtoReader(new BufReader(new StringReader(s)));}
Deno.test({  ignore: true,  name: "[textproto] Reader : DotBytes",  fn() {    const _input =      "dotlines\r\n.foo\r\n..bar\n...baz\nquux\r\n\r\n.\r\nanot.her\r\n";    return Promise.resolve();  },});
Deno.test("[textproto] ReadEmpty", async () => {  const r = reader("");  const m = await r.readMIMEHeader();  assertEquals(m, null);});
Deno.test("[textproto] Reader", async () => {  const r = reader("line1\nline2\n");  let s = await r.readLine();  assertEquals(s, "line1");
  s = await r.readLine();  assertEquals(s, "line2");
  s = await r.readLine();  assert(s === null);});
Deno.test({  name: "[textproto] Reader : MIME Header",  async fn() {    const input =      "my-key: Value 1  \r\nLong-key: Even Longer Value\r\nmy-Key: " +      "Value 2\r\n\n";    const r = reader(input);    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("My-Key"), "Value 1, Value 2");    assertEquals(m.get("Long-key"), "Even Longer Value");  },});
Deno.test({  name: "[textproto] Reader : MIME Header Single",  async fn() {    const input = "Foo: bar\n\n";    const r = reader(input);    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("Foo"), "bar");  },});
Deno.test({  name: "[textproto] Reader : MIME Header No Key",  async fn() {    const input = ": bar\ntest-1: 1\n\n";    const r = reader(input);    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("Test-1"), "1");  },});
Deno.test({  name: "[textproto] Reader : Large MIME Header",  async fn() {    const data: string[] = [];    // Go test is 16*1024. But seems it can't handle more    for (let i = 0; i < 1024; i++) {      data.push("x");    }    const sdata = data.join("");    const r = reader(`Cookie: ${sdata}\r\n\r\n`);    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("Cookie"), sdata);  },});
// Test that we don't read MIME headers seen in the wild,// with spaces before colons, and spaces in keys.Deno.test({  name: "[textproto] Reader : MIME Header Non compliant",  async fn() {    const input = "Foo: bar\r\n" +      "Content-Language: en\r\n" +      "SID : 0\r\n" +      "Audio Mode : None\r\n" +      "Privilege : 127\r\n\r\n";    const r = reader(input);    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("Foo"), "bar");    assertEquals(m.get("Content-Language"), "en");    // Make sure we drop headers with trailing whitespace    assertEquals(m.get("SID"), null);    assertEquals(m.get("Privilege"), null);    // Not legal http header    assertThrows((): void => {      assertEquals(m.get("Audio Mode"), "None");    });  },});
Deno.test({  name: "[textproto] Reader : MIME Header Malformed",  async fn() {    const input = [      "No colon first line\r\nFoo: foo\r\n\r\n",      " No colon first line with leading space\r\nFoo: foo\r\n\r\n",      "\tNo colon first line with leading tab\r\nFoo: foo\r\n\r\n",      " First: line with leading space\r\nFoo: foo\r\n\r\n",      "\tFirst: line with leading tab\r\nFoo: foo\r\n\r\n",      "Foo: foo\r\nNo colon second line\r\n\r\n",    ];    const r = reader(input.join(""));
    let err;    try {      await r.readMIMEHeader();    } catch (e) {      err = e;    }    assert(err instanceof Deno.errors.InvalidData);  },});
Deno.test({  name: "[textproto] Reader : MIME Header Trim Continued",  async fn() {    const input = "a:\n" +      " 0 \r\n" +      "b:1 \t\r\n" +      "c: 2\r\n" +      " 3\t\n" +      "  \t 4  \r\n\n";    const r = reader(input);    let err;    try {      await r.readMIMEHeader();    } catch (e) {      err = e;    }    assert(err instanceof Deno.errors.InvalidData);  },});
Deno.test({  name: "[textproto] #409 issue : multipart form boundary",  async fn() {    const input = [      "Accept: */*\r\n",      'Content-Disposition: form-data; name="test"\r\n',      " \r\n",      "------WebKitFormBoundaryimeZ2Le9LjohiUiG--\r\n\n",    ];    const r = reader(input.join(""));    const m = await r.readMIMEHeader();    assert(m !== null);    assertEquals(m.get("Accept"), "*/*");    assertEquals(m.get("Content-Disposition"), 'form-data; name="test"');  },});
/* TODO(kt3k): Enable this testDeno.test({  name: "[textproto] #4521 issue",  async fn() {    const input = "abcdefghijklmnopqrstuvwxyz";    const bufSize = 25;    const tp = new TextProtoReader(      new BufReader(new StringReader(input), bufSize),    );    const line = await tp.readLine();    assertEquals(line, input);  },});*/
Deno.test({  name: "[textproto] PR #859",  async fn() {    const TESTS: Array<string> = [      "Hello, World!",      "Hello, World!\n",      "Hello,\0 World!",      "Hello,\n World!",      "\nHello, World!",      "   \t   ",      "   \t   \n",      "   \n   \t",      "\n   ",    ];
    const EXPECTED_OUTPUTS = [      new Uint8Array([        72,        101,        108,        108,        111,        44,        32,        87,        111,        114,        108,        100,        33,      ]),      new Uint8Array([        72,        101,        108,        108,        111,        44,        32,        87,        111,        114,        108,        100,        33,      ]),      new Uint8Array([        72,        101,        108,        108,        111,        44,        0,        32,        87,        111,        114,        108,        100,        33,      ]),      new Uint8Array([72, 101, 108, 108, 111, 44]),      new Uint8Array(0),      new Uint8Array(0),      new Uint8Array(0),      new Uint8Array(0),      new Uint8Array(0),    ];
    for (let i = 0; i < TESTS.length; ++i) {      const READER = reader(TESTS[i]);      const RESULT = await READER.readLineSlice();      assertEquals(RESULT, EXPECTED_OUTPUTS[i]);    }  },});