diff --git a/packages/plugins/typescript/urql/src/visitor.ts b/packages/plugins/typescript/urql/src/visitor.ts index 8daba7c4dc8..c6fc9059ad6 100644 --- a/packages/plugins/typescript/urql/src/visitor.ts +++ b/packages/plugins/typescript/urql/src/visitor.ts @@ -84,7 +84,7 @@ export const ${componentName} = (props: Omit { + const docs = [{ location: '', document: basicDoc }]; + const content = (await plugin( + schema, + docs, + { omitOperationSuffix: true }, + { + outputFile: 'graphql.tsx', + } + )) as Types.ComplexPluginOutput; + + expect(content.content).not.toContain(`export class TestComponent`); + }); + it('should add three generics if operation type is subscription', async () => { const documents = parse(/* GraphQL */ ` subscription ListenToComments($name: String) { @@ -601,5 +615,19 @@ export function useSubmitRepositoryMutation() { expect(content.content).toContain(`export function useTestQuery`); }); + + it('Should respect omitOperationSuffix for hooks', async () => { + const docs = [{ location: '', document: basicDoc }]; + const content = (await plugin( + schema, + docs, + { withHooks: true, omitOperationSuffix: true }, + { + outputFile: 'graphql.tsx', + } + )) as Types.ComplexPluginOutput; + + expect(content.content).toContain(`export function useTest(`); + }); }); });