add mutation tests

Author: danadajian

src/definitions/object.ts

@@ -27,7 +27,7 @@ import {
 import { buildFieldDefinition } from "../helpers/build-field-definition";
 import { CodegenConfigWithDefaults } from "../helpers/build-config-with-defaults";
 import { inputTypeHasMatchingOutputType } from "../helpers/input-type-has-matching-output-type";
-import { findTypeInResolverInterfacesConfig } from "../helpers/findTypeInResolverInterfacesConfig";
+import { findTypeInResolverInterfacesConfig } from "../helpers/find-type-in-resolver-interfaces-config";
 import { titleCase } from "title-case";
 
 export function buildObjectTypeDefinition(
@@ -78,7 +78,7 @@ export function buildObjectTypeDefinition(
 
   if (node.name.value === "Query" || node.name.value === "Mutation") {
     const individualQueryClasses = node.fields?.map((fieldNode) => {
-      const className = `${titleCase(fieldNode.name.value)}Query`;
+      const className = `${titleCase(fieldNode.name.value)}${node.name.value}`;
       return `${annotations}${outputRestrictionAnnotation}open class ${className}${interfaceInheritance} {
 ${getDataClassMembers({ node, fieldNodes: [fieldNode], schema, config, shouldGenerateFunctions })}
 }`;

src/helpers/build-config-with-defaults.ts

@@ -12,7 +12,7 @@ export function buildConfigWithDefaults(
     includeDependentTypes: true,
     unionGeneration: "MARKER_INTERFACE",
     extraImports: ["com.expediagroup.graphql.generator.annotations.*"],
-    resolverInterfaces: [{ typeName: "Query" }],
+    resolverInterfaces: [{ typeName: "Query" }, { typeName: "Mutation" }],
   } as const satisfies GraphQLKotlinCodegenConfig;
 
   return merge(defaultConfig, config) as GraphQLKotlinCodegenConfig &

src/helpers/build-field-definition.ts

@@ -23,7 +23,7 @@ import {
 import { CodegenConfigWithDefaults } from "./build-config-with-defaults";
 import { indent } from "@graphql-codegen/visitor-plugin-common";
 import { buildAnnotations } from "./build-annotations";
-import { findTypeInResolverInterfacesConfig } from "./findTypeInResolverInterfacesConfig";
+import { findTypeInResolverInterfacesConfig } from "./find-type-in-resolver-interfaces-config";
 
 export function buildFieldDefinition(
   node: ObjectTypeDefinitionNode | InterfaceTypeDefinitionNode,

src/helpers/findTypeInResolverInterfacesConfig.ts renamed to src/helpers/find-type-in-resolver-interfaces-config.ts

@@ -5,7 +5,7 @@ export function findTypeInResolverInterfacesConfig(
   node: ObjectTypeDefinitionNode | InterfaceTypeDefinitionNode,
   config: CodegenConfigWithDefaults,
 ) {
-  return config.resolverInterfaces.find(
+  return config.resolverInterfaces.findLast(
     (resolverInterface) => resolverInterface.typeName === node.name.value,
   );
 }

test/unit/should_handle_query_type_properly/codegen.config.ts

@@ -0,0 +1,5 @@
+import { GraphQLKotlinCodegenConfig } from "../../../src/plugin";
+
+export default {
+  resolverInterfaces: [{ typeName: "Mutation", classMethods: "SUSPEND" }],
+} satisfies GraphQLKotlinCodegenConfig;

test/unit/should_handle_query_type_properly/expected.kt

@@ -17,3 +17,19 @@ open class GetStuffQuery {
 open class GetStuffWithInputQuery {
     open fun getStuffWithInput(input: String, dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Query.getStuffWithInput must be implemented.")
 }
+
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.OBJECT])
+open class Mutation {
+    open suspend fun mutateStuff(dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Mutation.mutateStuff must be implemented.")
+    open suspend fun mutateStuffWithInput(input: String, dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Mutation.mutateStuffWithInput must be implemented.")
+}
+
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.OBJECT])
+open class MutateStuffMutation {
+    open suspend fun mutateStuff(dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Mutation.mutateStuff must be implemented.")
+}
+
+@GraphQLValidObjectLocations(locations = [GraphQLValidObjectLocations.Locations.OBJECT])
+open class MutateStuffWithInputMutation {
+    open suspend fun mutateStuffWithInput(input: String, dataFetchingEnvironment: graphql.schema.DataFetchingEnvironment): String = throw NotImplementedError("Mutation.mutateStuffWithInput must be implemented.")
+}

test/unit/should_handle_query_type_properly/schema.graphql

@@ -2,3 +2,8 @@ type Query {
   getStuff: String!
   getStuffWithInput(input: String!): String!
 }
+
+type Mutation {
+  mutateStuff: String!
+  mutateStuffWithInput(input: String!): String!
+}