Brunobkr/llama.cpp_AlgMor24_github

1// @ts-ignore TS61332import { expect, test } from "vitest";3 4import * as z from "zod/v3";5import { util } from "../helpers/util.js";6 7test("create enum", () => {8  const MyEnum = z.enum(["Red", "Green", "Blue"]);9  expect(MyEnum.Values.Red).toEqual("Red");10  expect(MyEnum.Enum.Red).toEqual("Red");11  expect(MyEnum.enum.Red).toEqual("Red");12});13 14test("infer enum", () => {15  const MyEnum = z.enum(["Red", "Green", "Blue"]);16  type MyEnum = z.infer<typeof MyEnum>;17  util.assertEqual<MyEnum, "Red" | "Green" | "Blue">(true);18});19 20test("get options", () => {21  expect(z.enum(["tuna", "trout"]).options).toEqual(["tuna", "trout"]);22});23 24test("readonly enum", () => {25  const HTTP_SUCCESS = ["200", "201"] as const;26  const arg = z.enum(HTTP_SUCCESS);27  type arg = z.infer<typeof arg>;28  util.assertEqual<arg, "200" | "201">(true);29 30  arg.parse("201");31  expect(() => arg.parse("202")).toThrow();32});33 34test("error params", () => {35  const result = z.enum(["test"], { required_error: "REQUIRED" }).safeParse(undefined);36  expect(result.success).toEqual(false);37  if (!result.success) {38    expect(result.error.issues[0].message).toEqual("REQUIRED");39  }40});41 42test("extract/exclude", () => {43  const foods = ["Pasta", "Pizza", "Tacos", "Burgers", "Salad"] as const;44  const FoodEnum = z.enum(foods);45  const ItalianEnum = FoodEnum.extract(["Pasta", "Pizza"]);46  const UnhealthyEnum = FoodEnum.exclude(["Salad"]);47  const EmptyFoodEnum = FoodEnum.exclude(foods);48 49  util.assertEqual<z.infer<typeof ItalianEnum>, "Pasta" | "Pizza">(true);50  util.assertEqual<z.infer<typeof UnhealthyEnum>, "Pasta" | "Pizza" | "Tacos" | "Burgers">(true);51  // @ts-expect-error TS234452  util.assertEqual<typeof EmptyFoodEnum, z.ZodEnum<[]>>(true);53  util.assertEqual<z.infer<typeof EmptyFoodEnum>, never>(true);54});55 56test("error map in extract/exclude", () => {57  const foods = ["Pasta", "Pizza", "Tacos", "Burgers", "Salad"] as const;58  const FoodEnum = z.enum(foods, {59    errorMap: () => ({ message: "This is not food!" }),60  });61  const ItalianEnum = FoodEnum.extract(["Pasta", "Pizza"]);62  const foodsError = FoodEnum.safeParse("Cucumbers");63  const italianError = ItalianEnum.safeParse("Tacos");64  if (!foodsError.success && !italianError.success) {65    expect(foodsError.error.issues[0].message).toEqual(italianError.error.issues[0].message);66  }67 68  const UnhealthyEnum = FoodEnum.exclude(["Salad"], {69    errorMap: () => ({ message: "This is not healthy food!" }),70  });71  const unhealthyError = UnhealthyEnum.safeParse("Salad");72  if (!unhealthyError.success) {73    expect(unhealthyError.error.issues[0].message).toEqual("This is not healthy food!");74  }75});76 77test("readonly in ZodEnumDef", () => {78  let _t!: z.ZodEnumDef<readonly ["a", "b"]>;79  _t;80});81