Merge pull request #9 from Teneff/bugfix-issue-#8

Bugfix issue #8

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@teneff/with-retry",
-  "version": "1.1.2",
+  "version": "1.1.3",
   "description": "Decorator for retrying async operations",
   "main": "index.js",
   "types": "index.d.ts",

src/decorator.spec.ts

@@ -30,6 +30,13 @@ class Example {
   getData2(): Promise<string> {
     return this.mockCallback("arg1", "arg2");
   }
+
+  @withRetry({
+    maxCalls: 4,
+  })
+  async getData3() {
+    console.log('3')
+  }
 }
 
 describe("decorator", () => {

src/decorator.ts

@@ -1,10 +1,10 @@
 import withRetry from "./withRetry";
 
 export default function <T>(...args: Parameters<typeof withRetry>) {
-  return function <K extends string>(
-    target: { [key in K]: (...args: any[]) => Promise<T> },
+  return function <K extends string, V extends Record<K, (...args: any[]) => Promise<T>>>(
+    target: V,
     propertyKey: K,
-    descriptor?: TypedPropertyDescriptor<() => Promise<T>>
+    descriptor?: TypedPropertyDescriptor<V[K]>
   ): any {
     const method = target[propertyKey];
     return Object.assign({}, descriptor, {

src/decorator2.spec.ts

@@ -0,0 +1,59 @@
+import withRetry from "./decorator";
+
+type Config = {
+  method: string;
+  url: string;
+  headers: {
+    "user-api-key": string;
+  };
+};
+
+class ErrorOnWhichWeShouldRetry extends Error {}
+
+class Example {
+  configService = {
+    get: <T extends string>(v: T): T => v,
+  };
+  httpService = {
+    axiosRef: {
+      request: async (config: Config) => ({
+        data: JSON.stringify(config),
+      }),
+    },
+  };
+
+  @withRetry({
+    errors: [ErrorOnWhichWeShouldRetry],
+    maxCalls: 10,
+    delay: 1000,
+  })
+  async acceptCredentialOffer(
+    userCredentialOfferId: string,
+    userApiKey: string,
+    subjectId: string
+  ) {
+    const config = {
+      method: "post",
+      url: `${this.configService.get<string>(
+        "prism.prismApiUrl"
+      )}/accept-credential-offer/${userCredentialOfferId}?subjectId=${subjectId}`,
+      headers: {
+        "user-api-key": userApiKey,
+      },
+    };
+
+    return this.httpService.axiosRef.request(config);
+  }
+}
+
+describe("Example", () => {
+  describe("acceptCredentialOffer", () => {
+    it("should acceptCredentialOffer", () => {
+      return expect(
+        new Example().acceptCredentialOffer("a", "b", "c")
+      ).resolves.toEqual({
+        data: '{\"method\":\"post\",\"url\":\"prism.prismApiUrl/accept-credential-offer/a?subjectId=c\",\"headers\":{\"user-api-key\":\"b\"}}',
+      });
+    });
+  });
+});