Merge pull request #398 from pranavkparti/transfers-post-fix

Created transfer returns token_count
Greenstand · Aug 21, 2023 · 6678ac8 · 6678ac8
2 parents c360847 + 1e41d78
commit 6678ac8
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/server/handlers/transferHandler.spec.js b/server/handlers/transferHandler.spec.js
@@ -114,6 +114,11 @@ describe('transferRouter', () => {
         result: {
           id: tokenId,
           state: TransferEnums.STATE.completed,
+          parameters: {
+            bundle: {
+              bundleSize: 1,
+            },
+          },
         },
         status: 201,
       });
@@ -128,6 +133,12 @@ describe('transferRouter', () => {
     expect(res.body).eql({
       id: tokenId,
       state: TransferEnums.STATE.completed,
+      parameters: {
+        bundle: {
+          bundleSize: 1,
+        },
+      },
+      token_count: 1
     });
     expect(
       initiateTranferStub.calledOnceWithExactly(
@@ -151,6 +162,11 @@ describe('transferRouter', () => {
         result: {
           id: tokenId,
           state: TransferEnums.STATE.completed,
+          parameters: {
+            bundle: {
+              bundleSize: 1,
+            },
+          },
         },
         status: 202,
       });
@@ -168,6 +184,12 @@ describe('transferRouter', () => {
     expect(res.body).eql({
       id: tokenId,
       state: TransferEnums.STATE.completed,
+      parameters: {
+        bundle: {
+          bundleSize: 1,
+        },
+      },
+      token_count: 1
     });
     expect(
       initiateTranferStub.calledOnceWithExactly(

diff --git a/server/handlers/transferHandler/index.js b/server/handlers/transferHandler/index.js
@@ -16,7 +16,13 @@ const transferPost = async (req, res) => {
     req.wallet_id,
   );
 
-  res.status(status).send(result);
+  const modifiedTransfer = {
+    ...result,
+    token_count:
+        +result.parameters?.bundle?.bundleSize || +result.parameters?.tokens?.length,
+  }
+
+  res.status(status).send(modifiedTransfer);
 };
 
 const transferIdAcceptPost = async (req, res) => {