updated tests. added client check based on client type

7nohe · Nov 4, 2024 · a4976ac · a4976ac
1 parent 2297531
commit a4976ac
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 5 deletions.
diff --git a/src/createImports.mts b/src/createImports.mts
@@ -45,12 +45,16 @@ export const createImports = ({
           ts.factory.createImportSpecifier(
             true,
             undefined,
-            ts.factory.createIdentifier("Options")
+            ts.factory.createIdentifier("Options"),
           ),
-        ])
+        ]),
+      ),
+      ts.factory.createStringLiteral(
+        client === "@hey-api/client-axios"
+          ? "@hey-api/client-axios"
+          : "@hey-api/client-fetch",
       ),
-      ts.factory.createStringLiteral("@hey-api/client-fetch"),
-      undefined
+      undefined,
     ),
     ts.factory.createImportDeclaration(
       undefined,
@@ -61,7 +65,7 @@ export const createImports = ({
           ts.factory.createImportSpecifier(
             true,
             undefined,
-            ts.factory.createIdentifier("QueryClient")
+            ts.factory.createIdentifier("QueryClient"),
           ),
           ts.factory.createImportSpecifier(
             false,

diff --git a/tests/__snapshots__/createSource.test.ts.snap b/tests/__snapshots__/createSource.test.ts.snap
@@ -11,6 +11,7 @@ export * from "./queries";
 exports[`createSource > createSource - @hey-api/client-axios 2`] = `
 "// generated with @7nohe/[email protected] 
 
+import { type Options } from "@hey-api/client-axios";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -47,6 +48,7 @@ exports[`createSource > createSource - @hey-api/client-axios 3`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-axios";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -65,6 +67,7 @@ exports[`createSource > createSource - @hey-api/client-axios 4`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-axios";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -80,6 +83,7 @@ exports[`createSource > createSource - @hey-api/client-axios 5`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-axios";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -102,6 +106,7 @@ export * from "./queries";
 exports[`createSource > createSource - @hey-api/client-fetch 2`] = `
 "// generated with @7nohe/[email protected] 
 
+import { type Options } from "@hey-api/client-fetch";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -137,6 +142,7 @@ exports[`createSource > createSource - @hey-api/client-fetch 3`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-fetch";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -154,6 +160,7 @@ exports[`createSource > createSource - @hey-api/client-fetch 4`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-fetch";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";
@@ -168,6 +175,7 @@ exports[`createSource > createSource - @hey-api/client-fetch 5`] = `
 "// generated with @7nohe/[email protected] 
 
 import * as Common from "./common";
+import { type Options } from "@hey-api/client-fetch";
 import { type QueryClient, useQuery, useSuspenseQuery, useMutation, UseQueryResult, UseQueryOptions, UseMutationOptions, UseMutationResult } from "@tanstack/react-query";
 import { client, findPets, addPet, getNotDefined, postNotDefined, findPetById, deletePet, findPaginatedPets } from "../requests/services.gen";
 import { Pet, NewPet, Error, FindPetsData, FindPetsResponse, FindPetsError, AddPetData, AddPetResponse, AddPetError, GetNotDefinedResponse, GetNotDefinedError, PostNotDefinedResponse, PostNotDefinedError, FindPetByIdData, FindPetByIdResponse, FindPetByIdError, DeletePetData, DeletePetResponse, DeletePetError, FindPaginatedPetsData, FindPaginatedPetsResponse, FindPaginatedPetsError } from "../requests/types.gen";

diff --git a/tests/createImports.test.ts b/tests/createImports.test.ts
@@ -20,6 +20,7 @@ describe(fileName, () => {
     // @ts-ignore
     const moduleNames = imports.map((i) => i.moduleSpecifier.text);
     expect(moduleNames).toStrictEqual([
+      "@hey-api/client-fetch",
       "@tanstack/react-query",
       "../requests/services.gen",
       "../requests/types.gen",
@@ -41,6 +42,7 @@ describe(fileName, () => {
     // @ts-ignore
     const moduleNames = imports.map((i) => i.moduleSpecifier.text);
     expect(moduleNames).toStrictEqual([
+      "@hey-api/client-fetch",
       "@tanstack/react-query",
       "../requests/services.gen",
       "../requests/types.gen",