test(): added tests for enum capitalization

Author: kasvith

packages/apollo/tests/code-first/app.module.ts

@@ -2,15 +2,16 @@ import { Module } from '@nestjs/common';
 import { GraphQLModule } from '@nestjs/graphql';
 import { ApolloDriverConfig } from '../../lib';
 import { ApolloDriver } from '../../lib/drivers';
+import { CatsModule } from './cats/cats.module';
 import { DirectionsModule } from './directions/directions.module';
 import { RecipesModule } from './recipes/recipes.module';
 
 @Module({
   imports: [
     RecipesModule,
     DirectionsModule,
-    GraphQLModule.forRoot<ApolloDriverConfig>({
-      driver: ApolloDriver,
+    CatsModule,
+    GraphQLModule.forRoot({
       debug: false,
       installSubscriptionHandlers: true,
       autoSchemaFile: true,

packages/apollo/tests/code-first/cats/cats.resolver.ts

@@ -1,9 +1,19 @@
-import { Query, Resolver } from '@nestjs/graphql';
+import { Query, Resolver, Args } from '@nestjs/graphql';
+import { CatType } from '../enums/cat-type.enum';
 
 @Resolver()
 export class CatsResolver {
   @Query((returns) => String)
   getAnimalName(): string {
     return 'cat';
   }
+
+  @Resolver()
+  @Query((returns) => CatType)
+  catType(
+    @Args({ name: 'catType', type: () => CatType })
+    catType: CatType,
+  ): CatType {
+    return catType;
+  }
 }

packages/apollo/tests/code-first/enums/cat-type.enum.ts

@@ -0,0 +1,13 @@
+import { registerEnumType } from '@nestjs/graphql';
+
+export enum CatType {
+  PersianCat = 'persian-cat',
+  MaineCoon = 'maine-coon',
+  Ragdoll = 'ragdoll',
+}
+
+registerEnumType(CatType, {
+  name: 'CatType',
+  description: 'Distinguish cats',
+  mapToUppercase: true,
+});

packages/apollo/tests/e2e/code-first-schema.spec.ts

@@ -29,6 +29,7 @@ import {
   getSubscriptionByName,
 } from '../utils/introspection-schema.utils';
 import { printedSchemaSnapshot } from '../utils/printed-schema.snapshot';
+import { CatsResolver } from '../code-first/cats/cats.resolver';
 
 describe('Code-first - schema factory', () => {
   let schemaFactory: GraphQLSchemaFactory;
@@ -50,6 +51,7 @@ describe('Code-first - schema factory', () => {
         [
           RecipesResolver,
           DirectionsResolver,
+          CatsResolver,
           AbstractResolver,
           IRecipesResolver,
         ],
@@ -68,17 +70,18 @@ describe('Code-first - schema factory', () => {
         printedSchemaSnapshot,
       );
     });
-    it('should define 5 queries', async () => {
+    it('should define 6 queries', async () => {
       const type = getQuery(introspectionSchema);
 
-      expect(type.fields.length).toEqual(5);
+      expect(type.fields.length).toEqual(6);
       expect(type.fields.map((item) => item.name)).toEqual(
         expect.arrayContaining([
           'recipes',
           'search',
           'categories',
           'move',
           'recipe',
+          'catType',
         ]),
       );
     });
@@ -154,6 +157,40 @@ describe('Code-first - schema factory', () => {
       );
     });
 
+    it('should define "CatType" enum to use CAPITALIZED_UNDERSCORE', () => {
+      const type = introspectionSchema.types.find(
+        ({ name }) => name === 'CatType',
+      );
+
+      expect(type).toEqual(
+        expect.objectContaining({
+          kind: TypeKind.ENUM,
+          name: 'CatType',
+          description: 'Distinguish cats',
+          enumValues: [
+            {
+              deprecationReason: null,
+              description: null,
+              isDeprecated: false,
+              name: 'PERSIAN_CAT',
+            },
+            {
+              deprecationReason: null,
+              description: null,
+              isDeprecated: false,
+              name: 'MAINE_COON',
+            },
+            {
+              deprecationReason: null,
+              description: null,
+              isDeprecated: false,
+              name: 'RAGDOLL',
+            },
+          ],
+        }),
+      );
+    });
+
     it('should define "SearchResultUnion" union', () => {
       const type = introspectionSchema.types.find(
         ({ name }) => name === 'SearchResultUnion',

packages/apollo/tests/graphql/sort-auto-schema.module.ts

@@ -3,11 +3,13 @@ import { GraphQLModule } from '@nestjs/graphql';
 import { ApolloDriver } from '../../lib/drivers';
 import { DirectionsModule } from '../code-first/directions/directions.module';
 import { RecipesModule } from '../code-first/recipes/recipes.module';
+import { CatsModule } from '../code-first/cats/cats.module';
 
 @Module({
   imports: [
     RecipesModule,
     DirectionsModule,
+    CatsModule,
     GraphQLModule.forRoot({
       driver: ApolloDriver,
       autoSchemaFile: 'schema.graphql',

packages/apollo/tests/graphql/transform-auto-schema-file.module.ts

@@ -4,11 +4,13 @@ import { GraphQLSchema, lexicographicSortSchema } from 'graphql';
 import { ApolloDriver } from '../../lib/drivers';
 import { DirectionsModule } from '../code-first/directions/directions.module';
 import { RecipesModule } from '../code-first/recipes/recipes.module';
+import { CatsModule } from '../code-first/cats/cats.module';
 
 @Module({
   imports: [
     RecipesModule,
     DirectionsModule,
+    CatsModule,
     GraphQLModule.forRoot({
       driver: ApolloDriver,
       autoSchemaFile: 'schema.graphql',

packages/apollo/tests/utils/printed-schema.snapshot.ts

@@ -75,6 +75,7 @@ type Query {
     take: Int = 25
   ): [Recipe!]!
   move(direction: Direction!): Direction!
+  catType(catType: CatType!): CatType!
 }
 
 """Search result description"""
@@ -90,6 +91,13 @@ enum Direction {
   Sideways @deprecated(reason: "Replaced with Left or Right")
 }
 
+"""Distinguish cats"""
+enum CatType {
+  PERSIAN_CAT
+  MAINE_COON
+  RAGDOLL
+}
+
 type Mutation {
   addRecipe(newRecipeData: NewRecipeInput!): Recipe!
   removeRecipe(id: String!): Boolean!
@@ -113,6 +121,13 @@ export const sortedPrintedSchemaSnapshot = `# ----------------------------------
 # THIS FILE WAS AUTOMATICALLY GENERATED (DO NOT MODIFY)
 # ------------------------------------------------------
 
+"""Distinguish cats"""
+enum CatType {
+  MAINE_COON
+  PERSIAN_CAT
+  RAGDOLL
+}
+
 type Category {
   description: String!
   name: String!
@@ -162,6 +177,7 @@ input NewRecipeInput {
 }
 
 type Query {
+  catType(catType: CatType!): CatType!
   categories: [Category!]!
   move(direction: Direction!): Direction!