feat(easy): Add wrap function that takes a async function and return …

…that result of that function
thisisagile · Jul 28, 2023 · 292ba7c · 292ba7c
1 parent 48f40b4
commit 292ba7c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/packages/easy/src/types/Context.ts b/packages/easy/src/types/Context.ts
@@ -38,6 +38,7 @@ export interface RequestContext {
   get<T>(key: string): T;
   set<T>(key: string, value: T): T;
   create: (f: () => void) => void;
+  wrap<T>(f: () => Promise<T>): Promise<T>;
 }
 
 export class BaseRequestContext implements RequestContext {
@@ -88,6 +89,8 @@ export class BaseRequestContext implements RequestContext {
   }
 
   public readonly create = (f: () => void): void => f();
+
+  public readonly wrap = <T>(f: () => Promise<T>): Promise<T> => f();
 }
 
 /**

diff --git a/packages/easy/test/types/Context.test.ts b/packages/easy/test/types/Context.test.ts
@@ -44,6 +44,11 @@ describe('Environment Context', () => {
     expect(ctx.request.lastError).toBe('Wrong');
   });
 
+  test('simple wrap on request context', async () => {
+    const r = await ctx.request.wrap(async () => Promise.resolve(42));
+    expect(r).toBe(42);
+  });
+
   test('token should be an any in the context', () => {
     const c = new Context({ request: new BaseRequestContext() });
     c.request.token = { tenant: 42 };