Make types pascal cased (#467)

* Make types pascal cased * Update test spec to include union types
dotansimha · Aug 7, 2018 · 8f14052 · 8f14052
1 parent caf3e98
commit 8f14052
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 1 deletion.
diff --git a/packages/templates/typescript/src/schema.handlebars b/packages/templates/typescript/src/schema.handlebars
@@ -60,7 +60,7 @@ export interface {{ toPascalCase name }}{{ toPascalCase ../name }}Args {
 {{/each}}
 {{#each unions}}
 {{ toComment description }}
-export type {{ name }} = {{#each possibleTypes}}{{this}}{{#unless @last}} | {{/unless}}{{/each}};
+export type {{ toPascalCase name }} = {{#each possibleTypes}}{{ toPascalCase this}}{{#unless @last}} | {{/unless}}{{/each}};
 
 {{/each}}
 {{#if @root.config.schemaNamespace ~}} } {{~/if}}

diff --git a/packages/templates/typescript/tests/typescript.spec.ts b/packages/templates/typescript/tests/typescript.spec.ts
@@ -558,6 +558,41 @@ describe('TypeScript template', () => {
       `);
     });
 
+    it('should transform correctly name of union', async () => {
+      const { context } = compileAndBuildContext(`
+        type Query {
+          fieldTest: [CBText]
+        }
+        union CBText = ABText | BBText
+        scalar ABText
+        scalar BBText
+      `);
+
+      const compiled = await compileTemplate(
+        {
+          ...config
+        } as GeneratorConfig,
+        context
+      );
+
+      const content = compiled[0].content;
+
+      expect(content).toBeSimilarStringTo(`
+        export type AbText = any;
+      `);
+      expect(content).toBeSimilarStringTo(`
+        export type BbText = any;
+      `);
+      expect(content).toBeSimilarStringTo(`
+        export interface Query {
+          fieldTest?: (CbText | null)[] | null;
+        }
+      `);
+      expect(content).toBeSimilarStringTo(`
+        export type CbText = AbText | BbText;
+      `);
+    });
+
     it('should generate enums correctly', async () => {
       const { context } = compileAndBuildContext(`
         type Query {