Import useMutate on generated components

contiamo · May 17, 2019 · ed78ff2 · ed78ff2
1 parent 17bd22e
commit ed78ff2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/scripts/import-open-api.ts b/src/scripts/import-open-api.ts
@@ -558,7 +558,7 @@ const importOpenApi = async (
     imports.push("Get", "GetProps", "useGet", "UseGetProps");
   }
   if (haveMutate) {
-    imports.push("Mutate", "MutateProps");
+    imports.push("Mutate", "MutateProps", "useMutate", "UseMutateProps");
   }
   if (havePoll) {
     imports.push("Poll", "PollProps");

diff --git a/src/scripts/tests/__snapshots__/import-open-api.test.ts.snap b/src/scripts/tests/__snapshots__/import-open-api.test.ts.snap
@@ -4,7 +4,7 @@ exports[`scripts/import-open-api should parse correctly petstore-expanded.yaml 1
 "/* Generated by restful-react */
 
 import React from \\"react\\";
-import { Get, GetProps, useGet, UseGetProps, Mutate, MutateProps } from \\"restful-react\\";
+import { Get, GetProps, useGet, UseGetProps, Mutate, MutateProps, useMutate, UseMutateProps } from \\"restful-react\\";
 
 export type Omit<T, K extends keyof T> = Pick<T, Exclude<keyof T, K>>;