123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254
// Copyright 2018-2021 the oak authors. All rights reserved. MIT license.
// deno-lint-ignore-file
import { assertEquals } from "./test_deps.ts";
import { Cookies } from "./cookies.ts";import { KeyStack } from "./keyStack.ts";import type { Request } from "./request.ts";import type { Response } from "./response.ts";
const { test } = Deno;
function createMockRequest(cookieValue?: string[]): Request {  return {    headers: new Headers(      cookieValue ? [["Cookie", cookieValue.join("; ")]] : undefined,    ),  } as any;}
function createMockResponse(): Response {  return {    headers: new Headers(),  } as any;}
test({  name: "get cookie value",  fn() {    const request = createMockRequest(["foo=bar"]);    const response = createMockResponse();    const cookies = new Cookies(request, response);    assertEquals(cookies.get("foo"), "bar");    assertEquals(cookies.get("bar"), undefined);    assertEquals([...response.headers], []);  },});
test({  name: "get signed cookie",  fn() {    const request = createMockRequest(      ["bar=foo", "bar.sig=S7GhXzJF3n4j8JwTupr7H-h25qtt_vs0stdETXZb-Ro"],    );    const response = createMockResponse();    const cookies = new Cookies(      request,      response,      { keys: new KeyStack(["secret1"]) },    );    assertEquals(cookies.get("bar"), "foo");    assertEquals([...response.headers], []);  },});
test({  name: "get signed cookie requiring re-signing",  fn() {    const request = createMockRequest(      ["bar=foo", "bar.sig=S7GhXzJF3n4j8JwTupr7H-h25qtt_vs0stdETXZb-Ro"],    );    const response = createMockResponse();    const cookies = new Cookies(      request,      response,      { keys: new KeyStack(["secret2", "secret1"]) },    );    assertEquals(cookies.get("bar"), "foo");    assertEquals([...response.headers], [[      "set-cookie",      "bar.sig=ar46bgP3n0ZRazFOfiZ4SyZVFxKUvG1-zQZCb9lbcPI; path=/; httponly",    ]]);  },});
test({  name: "get invalid signed cookie",  fn() {    const request = createMockRequest(      ["bar=foo", "bar.sig=tampered", "foo=baz"],    );    const response = createMockResponse();    const cookies = new Cookies(      request,      response,      { keys: new KeyStack(["secret1"]) },    );    assertEquals(cookies.get("bar"), undefined);    assertEquals(cookies.get("foo"), undefined);    assertEquals([...response.headers], [      [        "set-cookie",        "bar.sig=; path=/; expires=Thu, 01 Jan 1970 00:00:00 GMT; httponly",      ],    ]);  },});
test({  name: "set cookie",  fn() {    const request = createMockRequest();    const response = createMockResponse();    const cookies = new Cookies(request, response);    cookies.set("foo", "bar");    assertEquals([...response.headers], [      ["set-cookie", "foo=bar; path=/; httponly"],    ]);  },});
test({  name: "set multiple cookies",  fn() {    const request = createMockRequest();    const response = createMockResponse();    const cookies = new Cookies(request, response);    cookies.set("a", "a");    cookies.set("b", "b");    cookies.set("c", "c");    assertEquals([...response.headers], [      ["set-cookie", "a=a; path=/; httponly"],      ["set-cookie", "b=b; path=/; httponly"],      ["set-cookie", "c=c; path=/; httponly"],    ]);  },});
test({  name: "set cookie with options",  fn() {    const request = createMockRequest();    const response = createMockResponse();    const cookies = new Cookies(request, response);    cookies.set("foo", "bar", {      domain: "*.example.com",      expires: new Date("2020-01-01T00:00:00+00:00"),      httpOnly: false,      overwrite: false,      path: "/foo",      sameSite: "strict",    });    assertEquals(      response.headers.get("set-cookie"),      "foo=bar; path=/foo; expires=Wed, 01 Jan 2020 00:00:00 GMT; domain=*.example.com; samesite=strict",    );  },});
test({  name: "set signed cookie",  fn() {    const request = createMockRequest();    const response = createMockResponse();    const cookies = new Cookies(      request,      response,      { keys: new KeyStack(["secret1"]) },    );    cookies.set("bar", "foo");
    assertEquals(      response.headers.get("set-cookie"),      "bar=foo; path=/; httponly, bar.sig=S7GhXzJF3n4j8JwTupr7H-h25qtt_vs0stdETXZb-Ro; path=/; httponly",    );  },});
test({  name: "iterate cookies",  fn() {    const request = createMockRequest(      ["bar=foo", "foo=baz", "baz=1234"],    );    const response = createMockResponse();    const cookies = new Cookies(      request,      response,    );    assertEquals(      [...cookies],      [["bar", "foo"], ["foo", "baz"], ["baz", "1234"]],    );  },});
test({  name: "iterate signed cookie",  fn() {    const request = createMockRequest(      ["bar=foo", "bar.sig=S7GhXzJF3n4j8JwTupr7H-h25qtt_vs0stdETXZb-Ro"],    );    const response = createMockResponse();    const cookies = new Cookies(      request,      response,      { keys: new KeyStack(["secret1"]) },    );    assertEquals([...cookies], [["bar", "foo"]]);  },});
test({  name: "Cookies - inspecting",  fn() {    const request = createMockRequest(      ["bar=foo", "foo=baz", "baz=1234"],    );    const response = createMockResponse();    assertEquals(      Deno.inspect(new Cookies(request, response)),      `Cookies [ [ "bar", "foo" ], [ "foo", "baz" ], [ "baz", "1234" ] ]`,    );  },});
test({  name: "set multiple cookies with options",  fn() {    const request = createMockRequest();    const response = createMockResponse();    const cookies = new Cookies(request, response);    cookies.set("foo", "bar", {      domain: "*.example.com",      expires: new Date("2020-01-01T00:00:00+00:00"),      httpOnly: false,      overwrite: false,      path: "/foo",      sameSite: "strict",    });    cookies.set("a", "b", {      domain: "*.example.com",      expires: new Date("2020-01-01T00:00:00+00:00"),      httpOnly: false,      overwrite: false,      path: "/a",      sameSite: "strict",    });    cookies.set("foo", "baz", {      domain: "*.example.com",      expires: new Date("2020-01-01T00:00:00+00:00"),      httpOnly: false,      overwrite: true,      path: "/baz",      sameSite: "strict",    });    assertEquals(      response.headers.get("set-cookie"),      "a=b; path=/a; expires=Wed, 01 Jan 2020 00:00:00 GMT; domain=*.example.com; samesite=strict, foo=baz; path=/baz; expires=Wed, 01 Jan 2020 00:00:00 GMT; domain=*.example.com; samesite=strict",    );  },});