SnowLuma/.pnpm-store/v11/files/ae/6375648ae621d63f29f20db69b7a787b956b6847ba87c3a0c90afebb1455bb598e4b2fe7f9b86ffa177e423257fda28183c950faf1421fda33107d14d30329

// @ts-ignore TS6133
import { expect, test } from "vitest";

import * as z from "zod/v3";

test("object augmentation", () => {
  const Animal = z
    .object({
      species: z.string(),
    })
    .augment({
      population: z.number(),
    });
  // overwrites `species`
  const ModifiedAnimal = Animal.augment({
    species: z.array(z.string()),
  });
  ModifiedAnimal.parse({
    species: ["asd"],
    population: 1324,
  });

  const bad = () =>
    ModifiedAnimal.parse({
      species: "asdf",
      population: 1324,
    } as any);
  expect(bad).toThrow();
});