From 068f697aa5e827ff14297d31111f6112edcd8ab7 Mon Sep 17 00:00:00 2001
From: Jeremias Diaz <monitz87@gmail.com>
Date: Tue, 13 Oct 2020 18:33:33 -0300
Subject: [PATCH] feat: support new status codes

---
 src/types/index.ts           | 2 ++
 src/utils/__tests__/index.ts | 8 ++++++++
 src/utils/index.ts           | 6 ++++++
 3 files changed, 16 insertions(+)

diff --git a/src/types/index.ts b/src/types/index.ts
index 6622cb3c54..f42c9f0de3 100644
--- a/src/types/index.ts
+++ b/src/types/index.ts
@@ -358,6 +358,8 @@ export enum TransferStatus {
   VolumeLimitReached = 'VolumeLimitReached', // 165
   BlockedTransaction = 'BlockedTransaction', // 166
   FundsLimitReached = 'FundsLimitReached', // 168
+  PortfolioFailure = 'PortfolioFailure', // 169
+  CustodianError = 'CustodianError', // 170
 }
 
 export interface ClaimTarget {
diff --git a/src/utils/__tests__/index.ts b/src/utils/__tests__/index.ts
index e8f0ef2450..d261840f75 100644
--- a/src/utils/__tests__/index.ts
+++ b/src/utils/__tests__/index.ts
@@ -2428,6 +2428,14 @@ describe('u8ToTransferStatus', () => {
 
     expect(result).toBe(TransferStatus.FundsLimitReached);
 
+    result = u8ToTransferStatus(dsMockUtils.createMockU8(169));
+
+    expect(result).toBe(TransferStatus.PortfolioFailure);
+
+    result = u8ToTransferStatus(dsMockUtils.createMockU8(170));
+
+    expect(result).toBe(TransferStatus.CustodianError);
+
     const fakeStatusCode = 1;
     expect(() => u8ToTransferStatus(dsMockUtils.createMockU8(fakeStatusCode))).toThrow(
       `Unsupported status code "${fakeStatusCode}". Please report this issue to the Polymath team`
diff --git a/src/utils/index.ts b/src/utils/index.ts
index 376a367f97..3a61a641ea 100644
--- a/src/utils/index.ts
+++ b/src/utils/index.ts
@@ -640,6 +640,12 @@ export function u8ToTransferStatus(status: u8): TransferStatus {
     case 168: {
       return TransferStatus.FundsLimitReached;
     }
+    case 169: {
+      return TransferStatus.PortfolioFailure;
+    }
+    case 170: {
+      return TransferStatus.CustodianError;
+    }
     case 80: {
       return TransferStatus.Failure;
     }