From 6c1409b00db68dcd88b42806efef1e91bb1abed2 Mon Sep 17 00:00:00 2001
From: Fabio Berger <me@fabioberger.com>
Date: Mon, 29 Jan 2018 13:25:51 +0100
Subject: Fix bug related to balance/allowance fetching being async

---
 packages/website/ts/components/token_balances.tsx | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'packages/website')

diff --git a/packages/website/ts/components/token_balances.tsx b/packages/website/ts/components/token_balances.tsx
index ecafe5432..776ccb16e 100644
--- a/packages/website/ts/components/token_balances.tsx
+++ b/packages/website/ts/components/token_balances.tsx
@@ -133,6 +133,20 @@ export class TokenBalances extends React.Component<TokenBalancesProps, TokenBala
         if (!_.isEqual(nextProps.trackedTokens, this.props.trackedTokens)) {
             const newTokens = _.difference(nextProps.trackedTokens, this.props.trackedTokens);
             const newTokenAddresses = _.map(newTokens, token => token.address);
+            // Add placeholder entry for this token to the state, since fetching the
+            // balance/allowance is asynchronous
+            const trackedTokenStateByAddress = this.state.trackedTokenStateByAddress;
+            for (const tokenAddress of newTokenAddresses) {
+                trackedTokenStateByAddress[tokenAddress] = {
+                    balance: new BigNumber(0),
+                    allowance: new BigNumber(0),
+                    isLoaded: false,
+                };
+            }
+            this.setState({
+                trackedTokenStateByAddress,
+            });
+            // Fetch the actual balance/allowance.
             this._fetchBalancesAndAllowancesAsync(newTokenAddresses);
         }
     }
-- 
cgit v1.2.3