import { test, expect, describe } from "vitest";
import {
  getConvexSize,
  getDocumentSize,
  SYSTEM_FIELD_CREATION_TIME_SIZE,
  SYSTEM_FIELD_ID_ESTIMATE,
} from "./size.js";

/**
 * Tests for getConvexSize JS-specific behavior.
 * Cross-language consistency with Rust is tested via proptests in
 * crates/isolate/src/tests/values.rs which call the JS implementation.
 */
describe("getConvexSize", () => {
  test("undefined returns 0", () => {
    expect(getConvexSize(undefined)).toBe(0);
  });

  test("long strings use TextEncoder path (>500 chars)", () => {
    // This tests the JS-specific optimization for long strings
    const longAscii = "a".repeat(1000);
    expect(getConvexSize(longAscii)).toBe(1002); // 1000 + 2 overhead

    const longUtf8 = "héllo".repeat(200); // 6 bytes each
    expect(getConvexSize(longUtf8)).toBe(1202); // 1200 + 2 overhead
  });
});

/**
 * Tests for getDocumentSize - the system fields handling logic.
 */
describe("getDocumentSize", () => {
  test("adds system fields overhead when not present", () => {
    const doc = { name: "Alice" };
    const baseSize = getConvexSize(doc);
    expect(getDocumentSize(doc)).toBe(
      baseSize + SYSTEM_FIELD_ID_ESTIMATE + SYSTEM_FIELD_CREATION_TIME_SIZE,
    );
  });

  test("empty document with system fields", () => {
    expect(getConvexSize({})).toBe(2);
    expect(getDocumentSize({})).toBe(
      2 + SYSTEM_FIELD_ID_ESTIMATE + SYSTEM_FIELD_CREATION_TIME_SIZE,
    );
  });

  test("does not double-count existing system fields", () => {
    const doc = {
      _id: "abc123def456ghi789jkl012mno345pq",
      _creationTime: 1234567890123,
      name: "Alice",
    };
    const baseSize = getConvexSize(doc);
    expect(getDocumentSize(doc)).toBe(baseSize);
  });

  test("handles partial system fields", () => {
    // Only _id present
    const docWithId = {
      _id: "abc123def456ghi789jkl012mno345pq",
      name: "Alice",
    };
    const sizeWithId = getConvexSize(docWithId);
    expect(getDocumentSize(docWithId)).toBe(
      sizeWithId + SYSTEM_FIELD_CREATION_TIME_SIZE,
    ); // adds _creationTime

    // Only _creationTime present
    const docWithTime = {
      _creationTime: 1234567890123,
      name: "Alice",
    };
    const sizeWithTime = getConvexSize(docWithTime);
    expect(getDocumentSize(docWithTime)).toBe(sizeWithTime + 38); // adds _id
  });

  test("customIdLength option", () => {
    const doc = { name: "Alice", _creationTime: 1234567890123 };
    const customIdLength = 40;

    // With custom ID length: 4 (field) + 2 (string overhead) + 40 = 46 bytes
    const withCustomId = getDocumentSize(doc, { customIdLength });

    // Verify by computing with actual _id of that length
    const _id = "a".repeat(customIdLength);
    const withActualId = getConvexSize({ ...doc, _id });
    expect(withCustomId).toBe(withActualId);
  });
});