Fix test case

99designs · Oct 2, 2018 · 20add12 · 20add12
1 parent f5c0340
commit 20add12
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 8 deletions.
diff --git a/codegen/testserver/generated.go b/codegen/testserver/generated.go
diff --git a/codegen/testserver/generated_test.go b/codegen/testserver/generated_test.go
@@ -182,7 +182,7 @@ func TestGeneratedServer(t *testing.T) {
 		}
 		err := c.Post(`query { nullableArg(arg: null) }`, &resp)
 		require.Nil(t, err)
-		require.Nil(t, resp.NullableArg)
+		require.Equal(t, "Ok", *resp.NullableArg)
 	})
 }
 
@@ -236,6 +236,11 @@ func (r *testQueryResolver) User(ctx context.Context, id int) (User, error) {
 	return User{ID: 1}, nil
 }
 
+func (r *testQueryResolver) NullableArg(ctx context.Context, arg *int) (*string, error) {
+	s := "Ok"
+	return &s, nil
+}
+
 func (r *testResolver) Subscription() SubscriptionResolver {
 	return &testSubscriptionResolver{r}
 }

diff --git a/codegen/testserver/resolver.go b/codegen/testserver/resolver.go
@@ -65,7 +65,7 @@ func (r *queryResolver) Valid(ctx context.Context) (string, error) {
 func (r *queryResolver) User(ctx context.Context, id int) (User, error) {
 	panic("not implemented")
 }
-func (r *queryResolver) NullableArg(ctx context.Context, arg int) (*string, error) {
+func (r *queryResolver) NullableArg(ctx context.Context, arg *int) (*string, error) {
 	panic("not implemented")
 }
 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) {