diff --git a/src/__tests__/__snapshots__/createMethods.test.ts.snap b/src/__tests__/__snapshots__/createMethods.test.ts.snap index 2f2ec0a..3af825d 100644 --- 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 index 8b9b8ac..cffb1d3 100644 --- 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 index 5856bef..4b0158e 100644 --- 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) {