Fix naming issues with custom query root type (#77)

* Fix operation name for custom root query

* Update change log

Author: zhouzi

changelog.md

@@ -2,6 +2,10 @@
 
 ## Unreleased
 
+### Fixed
+
+- Fix operations factories name when using a custom root query #75
+
 ## 1.2.0 - 2023-01-20
 
 ### Added

packages/graphql-codegen-factories/src/operations/FactoriesOperationsVisitor.ts

@@ -1,7 +1,7 @@
 import path from "path";
 import { getBaseType } from "@graphql-codegen/plugin-helpers";
 import { getConfigValue } from "@graphql-codegen/visitor-plugin-common";
-import { camelCase } from "change-case-all";
+import { camelCase, pascalCase } from "change-case-all";
 import {
   FragmentDefinitionNode,
   GraphQLCompositeType,
@@ -173,7 +173,10 @@ export class FactoriesOperationsVisitor extends FactoriesBaseVisitor<
     selection: OperationDefinitionNode | SelectionNode
   ): NormalizedSelection[] {
     if (selection.kind === Kind.OPERATION_DEFINITION) {
-      const operationSuffix = this.getOperationSuffix(selection, parent.name);
+      const operationSuffix = this.getOperationSuffix(
+        selection,
+        pascalCase(selection.operation)
+      );
       const name = this.convertName(this.handleAnonymousOperation(selection), {
         suffix: operationSuffix,
       });

packages/graphql-codegen-factories/src/operations/__tests__/__snapshots__/plugin.ts.snap

@@ -297,6 +297,32 @@ export function createCreateUserMutationMock_createUser(props: Partial<CreateUse
 }
 `;
 
+exports[`plugin should support custom root Query 1`] = `
+Object {
+  "content": "export function createGetUserQueryMock(props: Partial<GetUserQuery> = {}): GetUserQuery {
+  return {
+    __typename: \\"CustomQuery\\",
+    user: null,
+    ...props,
+  };
+}
+
+export function createGetUserQueryMock_user(props: Partial<NonNullable<GetUserQuery[\\"user\\"]>> = {}): NonNullable<GetUserQuery[\\"user\\"]> {
+  const user = schemaFactories.createUserMock({
+    username: props.username,
+  });
+  return {
+    __typename: \\"User\\",
+    username: user.username,
+    ...props,
+  };
+}",
+  "prepend": Array [
+    "import * as schemaFactories from \\"./factories\\";",
+  ],
+}
+`;
+
 exports[`plugin should support external fragments 1`] = `
 Object {
   "content": "export function createGetMeQueryMock(props: Partial<GetMeQuery> = {}): GetMeQuery {

packages/graphql-codegen-factories/src/operations/__tests__/plugin.ts

@@ -587,4 +587,34 @@ describe("plugin", () => {
     );
     expect(output).toMatchSnapshot();
   });
+
+  it("should support custom root Query", async () => {
+    const schema = buildSchema(/* GraphQL */ `
+      type User {
+        username: String!
+      }
+
+      type CustomQuery {
+        user: User
+      }
+
+      schema {
+        query: CustomQuery
+      }
+    `);
+    const ast = parse(/* GraphQL */ `
+      query GetUser {
+        user {
+          username
+        }
+      }
+    `);
+
+    const output = await plugin(
+      schema,
+      [{ location: "GetUser.graphql", document: ast }],
+      { schemaFactoriesPath: "./factories" }
+    );
+    expect(output).toMatchSnapshot();
+  });
 });