Fix enums initial value

src/FactoriesVisitor.ts

@@ -76,7 +76,10 @@ export class FactoriesVisitor extends BaseVisitor<
           return this.config.enumsAsTypes
             ? `"${this.enums[name].getValues()[0].value}"`
             : `${name}.${this.convertName(
-                this.enums[name].getValues()[0].name
+                this.enums[name].getValues()[0].name,
+                {
+                  transformUnderscore: true,
+                }
               )}`;
         }
 

src/__integration__/schema.graphql

@@ -1,5 +1,11 @@
+enum UserRole {
+  SUPER_ADMIN
+  ADMIN
+}
+
 type User {
   id: ID!
+  role: UserRole!
 }
 
 type Query {

src/__integration__/types.ts

@@ -1,5 +1,7 @@
 export type Maybe<T> = T | null;
-export type Exact<T extends { [key: string]: any }> = { [K in keyof T]: T[K] };
+export type Exact<T extends { [key: string]: unknown }> = { [K in keyof T]: T[K] };
+export type MakeOptional<T, K extends keyof T> = Omit<T, K> & { [SubKey in K]?: Maybe<T[SubKey]> };
+export type MakeMaybe<T, K extends keyof T> = Omit<T, K> & { [SubKey in K]: Maybe<T[SubKey]> };
 /** All built-in and custom scalars, mapped to their actual values */
 export type Scalars = {
   ID: string;
@@ -9,9 +11,15 @@ export type Scalars = {
   Float: number;
 };
 
+export enum UserRole {
+  SuperAdmin = 'SUPER_ADMIN',
+  Admin = 'ADMIN'
+}
+
 export type User = {
   __typename?: 'User';
   id: Scalars['ID'];
+  role: UserRole;
 };
 
 export type Query = {
@@ -23,6 +31,7 @@ export function createUserMock(props: Partial<User>): User {
   return {
     __typename: "User",
     id: "",
+    role: UserRole.SuperAdmin,
     ...props,
   };
 }

src/__tests__/__snapshots__/plugin.ts.snap

@@ -73,6 +73,17 @@ exports[`plugin should not create factories for Query and Mutation 1`] = `
 "
 `;
 
+exports[`plugin should support enums with an underscore 1`] = `
+"export function createUserMock(props: Partial<User>): User {
+  return {
+    __typename: \\"User\\",
+    role: UserRole.SuperAdmin,
+    ...props,
+  };
+}
+"
+`;
+
 exports[`plugin should use enums as types 1`] = `
 "export function createUserMock(props: Partial<User>): User {
   return {

src/__tests__/plugin.ts

@@ -112,4 +112,17 @@ describe("plugin", () => {
       })
     ).toMatchSnapshot();
   });
+
+  it("should support enums with an underscore", () => {
+    const schema = buildSchema(`
+      enum UserRole {
+        SUPER_ADMIN
+        ADMIN
+      }
+      type User {
+        role: UserRole!
+      }
+    `);
+    expect(plugin(schema, [], {})).toMatchSnapshot();
+  });
 });