yaak/src-web/lib/model_util.test.ts

import type { HttpResponseEvent } from "@yaakapp-internal/models";
import { describe, expect, test } from "vite-plus/test";
import { getCookieCounts } from "./model_util";

function makeEvent(type: string, name: string, value: string): HttpResponseEvent {
  return {
    id: "test",
    model: "http_response_event",
    responseId: "resp",
    workspaceId: "ws",
    createdAt: new Date().toISOString(),
    updatedAt: new Date().toISOString(),
    event: { type, name, value } as HttpResponseEvent["event"],
  };
}

describe("getCookieCounts", () => {
  test("returns zeros for undefined events", () => {
    expect(getCookieCounts(undefined)).toEqual({ sent: 0, received: 0 });
  });

  test("returns zeros for empty events", () => {
    expect(getCookieCounts([])).toEqual({ sent: 0, received: 0 });
  });

  test("counts single sent cookie", () => {
    const events = [makeEvent("header_up", "Cookie", "session=abc123")];
    expect(getCookieCounts(events)).toEqual({ sent: 1, received: 0 });
  });

  test("counts multiple sent cookies in one header", () => {
    const events = [makeEvent("header_up", "Cookie", "a=1; b=2; c=3")];
    expect(getCookieCounts(events)).toEqual({ sent: 3, received: 0 });
  });

  test("counts single received cookie", () => {
    const events = [makeEvent("header_down", "Set-Cookie", "session=abc123; Path=/")];
    expect(getCookieCounts(events)).toEqual({ sent: 0, received: 1 });
  });

  test("counts multiple received cookies from multiple headers", () => {
    const events = [
      makeEvent("header_down", "Set-Cookie", "a=1; Path=/"),
      makeEvent("header_down", "Set-Cookie", "b=2; HttpOnly"),
      makeEvent("header_down", "Set-Cookie", "c=3; Secure"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 0, received: 3 });
  });

  test("deduplicates sent cookies by name", () => {
    const events = [
      makeEvent("header_up", "Cookie", "session=old"),
      makeEvent("header_up", "Cookie", "session=new"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 1, received: 0 });
  });

  test("deduplicates received cookies by name", () => {
    const events = [
      makeEvent("header_down", "Set-Cookie", "token=abc; Path=/"),
      makeEvent("header_down", "Set-Cookie", "token=xyz; Path=/"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 0, received: 1 });
  });

  test("counts both sent and received cookies", () => {
    const events = [
      makeEvent("header_up", "Cookie", "a=1; b=2; c=3"),
      makeEvent("header_down", "Set-Cookie", "x=10; Path=/"),
      makeEvent("header_down", "Set-Cookie", "y=20; Path=/"),
      makeEvent("header_down", "Set-Cookie", "z=30; Path=/"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 3, received: 3 });
  });

  test("ignores non-cookie headers", () => {
    const events = [
      makeEvent("header_up", "Content-Type", "application/json"),
      makeEvent("header_down", "Content-Length", "123"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 0, received: 0 });
  });

  test("handles case-insensitive header names", () => {
    const events = [
      makeEvent("header_up", "COOKIE", "a=1"),
      makeEvent("header_down", "SET-COOKIE", "b=2; Path=/"),
    ];
    expect(getCookieCounts(events)).toEqual({ sent: 1, received: 1 });
  });
});