test: Native string should not be imported

unlight · Apr 16, 2021 · 2c470f8 · 2c470f8
1 parent aba712c
commit 2c470f8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/src/helpers/get-graphql-import.ts b/src/helpers/get-graphql-import.ts
@@ -38,7 +38,6 @@ export function getGraphqlImport(args: {
         switch (typeName) {
             case 'Float':
             case 'Int':
-            case 'String':
                 return { name: typeName, specifier: '@nestjs/graphql' };
             case 'DateTime':
                 return { name: 'Date', specifier: undefined };

diff --git a/src/test/generate.spec.ts b/src/test/generate.spec.ts
@@ -496,13 +496,9 @@ describe('one model with scalar types', () => {
 
     describe('user create input', () => {
         before(() => {
-            sourceFile = project.getSourceFile(s =>
-                s.getFilePath().endsWith('/user-create.input.ts'),
-            )!;
+            setSourceFile('/user-create.input.ts');
         });
 
-        // it('', () => console.log(sourceFile.getText()));
-
         it('valid imports', () => {
             const sourceText = sourceFile.getText();
             expect(sourceText).not.toContain("import ';");
@@ -534,6 +530,12 @@ describe('one model with scalar types', () => {
             expect(getFieldType(sourceFile, 'data')).toEqual('() => GraphQLJSON');
         });
 
+        it('native string should not be imported', () => {
+            expect(imports).not.toContainEqual(
+                expect.objectContaining({ name: 'String' }),
+            );
+        });
+
         // it('', () => console.log(sourceFile.getText()));
     });