add tests

codegen/testserver/generated_test.go

@@ -519,6 +519,29 @@ func TestTracer(t *testing.T) {
 		require.NoError(t, err)
 		require.True(t, called)
 	})
+
+	t.Run("model methods", func(t *testing.T) {
+		t.Run("without context", func(t *testing.T) {
+			var resp struct {
+				ModelMethods struct {
+					NoContext bool
+				}
+			}
+			err := c.Post(`query { modelMethods{ noContext } }`, &resp)
+			require.NoError(t, err)
+			require.True(t, resp.ModelMethods.NoContext)
+		})
+		t.Run("with context", func(t *testing.T) {
+			var resp struct {
+				ModelMethods struct {
+					WithContext bool
+				}
+			}
+			err := c.Post(`query { modelMethods{ withContext } }`, &resp)
+			require.NoError(t, err)
+			require.True(t, resp.ModelMethods.WithContext)
+		})
+	})
 }
 
 func TestResponseExtension(t *testing.T) {
@@ -556,6 +579,9 @@ func (r *testResolver) User() UserResolver {
 func (r *testResolver) Query() QueryResolver {
 	return &testQueryResolver{}
 }
+func (r *testResolver) ModelMethods() ModelMethodsResolver {
+	return &modelMethodsResolver{}
+}
 
 type testQueryResolver struct{ queryResolver }
 

codegen/testserver/gqlgen.yml

@@ -11,6 +11,8 @@ resolver:
 models:
   It:
     model: "github.com/99designs/gqlgen/codegen/testserver/introspection.It"
+  ModelMethods:
+    model: "github.com/99designs/gqlgen/codegen/testserver.ModelMethods"
   InvalidIdentifier:
     model: "github.com/99designs/gqlgen/codegen/testserver/invalid-packagename.InvalidIdentifier"
   Changes:

codegen/testserver/models.go

@@ -1,11 +1,25 @@
 package testserver
 
-import "fmt"
+import (
+	context "context"
+	"fmt"
+)
 
 type ForcedResolver struct {
 	Field Circle
 }
 
+type ModelMethods struct {
+}
+
+func (m ModelMethods) NoContext() bool {
+	return true
+}
+
+func (m ModelMethods) WithContext(_ context.Context) bool {
+	return true
+}
+
 type Error struct {
 	ID string
 }

codegen/testserver/resolver.go

@@ -23,6 +23,15 @@ func (r *Resolver) Subscription() SubscriptionResolver {
 func (r *Resolver) User() UserResolver {
 	return &userResolver{r}
 }
+func (r *Resolver) ModelMethods() ModelMethodsResolver {
+	return &modelMethodsResolver{}
+}
+
+type modelMethodsResolver struct{}
+
+func (r *modelMethodsResolver) ResolverField(ctx context.Context, obj *ModelMethods) (bool, error) {
+	return true, nil
+}
 
 type forcedResolverResolver struct{ *Resolver }
 
@@ -68,6 +77,9 @@ func (r *queryResolver) User(ctx context.Context, id int) (User, error) {
 func (r *queryResolver) NullableArg(ctx context.Context, arg *int) (*string, error) {
 	panic("not implemented")
 }
+func (r *queryResolver) ModelMethods(ctx context.Context) (*ModelMethods, error) {
+	return &ModelMethods{}, nil
+}
 func (r *queryResolver) KeywordArgs(ctx context.Context, breakArg string, defaultArg string, funcArg string, interfaceArg string, selectArg string, caseArg string, deferArg string, goArg string, mapArg string, structArg string, chanArg string, elseArg string, gotoArg string, packageArg string, switchArg string, constArg string, fallthroughArg string, ifArg string, rangeArg string, typeArg string, continueArg string, forArg string, importArg string, returnArg string, varArg string) (bool, error) {
 	panic("not implemented")
 }