feat: add support for enum arrays with default value

fix enum list with default value
luisrudge · Nov 15, 2023 · eab337c · eab337c
2 parents b7ec96c + 8743766
commit eab337c
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/src/__tests__/__snapshots__/createMethods.test.ts.snap b/src/__tests__/__snapshots__/createMethods.test.ts.snap
@@ -76,6 +76,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: null,
+ enums: [Enum.A],
  };
 }
 export function fakeUser2Complete() {
@@ -175,6 +176,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
+ enums: [Enum.A],
  };
 }
 export function fakeUser2Complete() {
@@ -274,6 +276,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
+ enums: [Enum.A],
  };
 }
 export function fakeUser2Complete() {
@@ -373,6 +376,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
+ enums: [Enum.A],
  };
 }
 export function fakeUser2Complete() {
@@ -472,6 +476,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
+ enums: [Enum.A],
  };
 }
 export function fakeUser2Complete() {

diff --git a/src/__tests__/sample.prisma b/src/__tests__/sample.prisma
@@ -46,6 +46,7 @@ model User {
  nullableEnum Enum?
  relation UserRelation? @relation(name: "UserRelationToUser1")
  relation2 UserRelation? @relation(name: "UserRelationToUser2")
+ enums Enum[] @default([A])
 }
 
 model User2 {

diff --git a/src/helpers/createMethods.ts b/src/helpers/createMethods.ts
@@ -20,6 +20,18 @@ function getFieldDefinition(
  }`;
  }
  if (field.hasDefaultValue) {
+ if (field.isList && field.kind === 'enum') {
+ const enumName = field.type;
+ const enumValues = enums.find((it) => it.name === enumName)?.values || [];
+ if (enumValues.length === 0) {
+ logger.warn(
+ `Enum ${enumName} has no enum values. Field ${field.name} won't be generated.`,
+ );
+ }
+ else {
+ return `${field.name}: [${enumName}.${field.default}]`;
+ }
+ }
  if (field.isList) {
  return `${field.name}: ${field.default?.toString() || '[]'}`;
  }