Merge pull request #22 from unclejustin/patch-1

Support array of default enums
luisrudge · Jan 25, 2024 · d7bc72f · d7bc72f
2 parents 9de7b5b + d68f1f4
commit d7bc72f
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/src/__tests__/__snapshots__/createMethods.test.ts.snap b/src/__tests__/__snapshots__/createMethods.test.ts.snap
@@ -76,7 +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],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -176,7 +176,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: null,
- enums: [Enum.A],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -276,7 +276,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: null,
- enums: [Enum.A],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -376,7 +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],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -476,7 +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],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -576,7 +576,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
- enums: [Enum.A],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {
@@ -676,7 +676,7 @@ export function fakeUserComplete() {
  enum: faker.helpers.arrayElement([Enum.A, Enum.B, Enum.C] as const),
  enumWithDefault: Enum.A,
  nullableEnum: undefined,
- enums: [Enum.A],
+ enums: [Enum.A, Enum.B],
  };
 }
 export function fakeUser2Complete() {

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

diff --git a/src/helpers/createMethods.ts b/src/helpers/createMethods.ts
@@ -29,7 +29,8 @@ function getFieldDefinition(
  );
  }
  else {
- return `${field.name}: [${enumName}.${field.default}]`;
+ const defaults = field.default.map((d) => `${enumName}.${d}`).join(', ');
+ return `${field.name}: [${defaults}]`;
  }
  }
  if (field.isList) {