Merge pull request #79 from PolymathNetwork/feature/reservation-data

fix: return proper variables from toPojo
PolymathNetwork · Sep 2, 2019 · 38ea583 · 38ea583
2 parents 96be2ba + 1307153
commit 38ea583
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/entities/SecurityTokenReservation.ts b/src/entities/SecurityTokenReservation.ts
@@ -116,13 +116,13 @@ export class SecurityTokenReservation extends Entity<Params> {
   };
 
   public toPojo() {
-    const { uid, symbol, expiry, securityTokenAddress } = this;
+    const { uid, symbol, expiry, securityTokenAddress, reservedAt, ownerAddress } = this;
 
-    return { uid, symbol, expiry, securityTokenAddress };
+    return { uid, symbol, expiry, securityTokenAddress, reservedAt, ownerAddress };
   }
 
   public _refresh(params: Partial<Params>) {
-    const { expiry, securityTokenAddress } = params;
+    const { expiry, securityTokenAddress, reservedAt, ownerAddress } = params;
 
     if (expiry) {
       this.expiry = expiry;
@@ -131,5 +131,13 @@ export class SecurityTokenReservation extends Entity<Params> {
     if (securityTokenAddress) {
       this.securityTokenAddress = securityTokenAddress;
     }
+
+    if (reservedAt) {
+      this.reservedAt = reservedAt;
+    }
+
+    if (ownerAddress) {
+      this.ownerAddress = ownerAddress;
+    }
   }
 }