From d89099d90a7765fcb12beda57676172b62aef241 Mon Sep 17 00:00:00 2001
From: clawbot <35+clawbot@noreply.example.org>
Date: Tue, 3 Nov 2026 02:25:25 +0100
Subject: [PATCH] fix: the address screen shows a token just added at its
fetched balance, not 0.0000 (closes #516)
A balance refresh that returns now draws the balances on AddressDetail or
AddressToken again when that screen is up, not only the wallet list. Their
transaction lists are left alone.
Every add path on AddToken and SettingsAddToken tracks a token through
trackToken(), which gives the tracked record `balanceNotFetched: true`. While
the field is set, the token's line reads `quantity unknown` instead of 0.0000,
and Send reads its balance as unknown. refreshBalances() now resolves with the
tracked tokens whose balance it read at every address. The popup's refresh
deletes the field from those tokens in the stored record and leaves the rest of
the stored list as it is. Removing the token removes the field, and each network
keeps its own records.
Model: opus-5-5
---
README.md | 28 +-
src/popup/index.js | 17 +-
src/popup/views/addToken.js | 11 +-
src/popup/views/addressDetail.js | 29 +-
src/popup/views/addressToken.js | 80 ++--
src/popup/views/helpers.js | 12 +-
src/popup/views/send.js | 12 +-
src/popup/views/settingsAddToken.js | 15 +-
src/shared/balances.js | 24 +-
src/shared/state.js | 26 +-
tests/addTokenLateAnswer.test.js | 1 +
tests/addedTokenBalance.test.js | 458 ++++++++++++++++++++++
tests/balanceRefreshNetworkSwitch.test.js | 1 +
tests/blockscoutEtherscanSettings.test.js | 2 +
tests/e2e/run.js | 3 +-
tests/firstWalletRefresh.test.js | 1 +
tests/flashLine.test.js | 1 +
tests/support/popupBoot.js | 5 +-
tests/trackedTokenBalances.test.js | 28 ++
tests/trackedTokensPerNetwork.test.js | 11 +-
20 files changed, 699 insertions(+), 66 deletions(-)
create mode 100644 tests/addedTokenBalance.test.js
diff --git a/README.md b/README.md
index ee98310..de33fb1 100644
--- a/README.md
+++ b/README.md
@@ -1490,7 +1490,9 @@ lists, the send-screen token selector, the address total and the remove-address
warning (`isBelowOneMillionth()` in `src/shared/amountDisplay.js`). The Send and
confirmation screens show it when its token is the one being sent. Tracked
tokens with a zero balance are listed as well while "Show tracked tokens with
-zero balance" is on.
+zero balance" is on. While that setting is on, a token just added reads
+`quantity unknown` at an address with no entry for it, until a refresh has read
+its balance at every address (see AddToken).
#### Stored state and its version
@@ -1674,6 +1676,13 @@ network switch (below). That popup then redraws its top screen as a reopened
popup would. A screen that a reopened popup cannot render, such as Send, is
replaced by Home.
+The popup refreshes balances every ten seconds while it is open
+(`doRefreshAndRender()` in `src/popup/index.js`). Each refresh that returns
+draws the wallet list again. When AddressDetail or AddressToken is up, the
+refresh also draws that screen's USD total and balance lines again, but not its
+transaction list. A refresh draws no other screen again, DeleteAddress and its
+balance lines included.
+
A network switch leaves every screen on the page showing the previous network.
The user makes it in Settings, or approves a site's request and the background
saves it. Any other popup open meanwhile takes the switch from the stored
@@ -1933,7 +1942,8 @@ view would leave a wallet one click from deletion.
- To: address or ENS name input, with an inline validation message
- Amount input with current balance display, which reads
`Current balance: unknown (SYMBOL)` for a token whose scale is unknown, as
- ConfirmTx's balance line does (see Unknown token scale)
+ ConfirmTx's balance line does (see Unknown token scale). It reads the same
+ for a token whose balance no refresh has read yet (see AddToken)
- "Max" button beside the amount input, always in place. It fills in a
token's balance, cut down to the 18 decimal places ConfirmTx accepts for a
token that has more, or for ETH the exact balance minus the network fee
@@ -2236,7 +2246,15 @@ view would leave a wallet one click from deletion.
- "Add" button
- **Transitions**:
- "Add" (valid contract) → tracks the token, pops the stack, and re-renders
- **AddressDetail**
+ **AddressDetail**. Until a balance refresh on the token's network has read
+ the token's balance at every address, its line never reads `0.0000`. The
+ line reads `quantity unknown` instead, or is left out while "Show tracked
+ tokens with zero balance" is off. Where Blockscout already listed a
+ holding of the token, the line shows that holding. Until that refresh, the
+ tracked token's record carries `balanceNotFetched`, so the line still
+ reads `quantity unknown` after the popup closes and opens again. A refresh
+ that returns draws AddressDetail's balances again (see Navigation), so the
+ fetched balance shows without the screen being reopened
- "Add" with a token already tracked, a scam-listed address, or a failed
contract lookup → flash message, no screen change
- "Add" with a token that the spoof rule refuses → nothing is tracked; "Not
@@ -2548,7 +2566,9 @@ view would leave a wallet one click from deletion.
add path
- **Transitions**:
- Any of the add paths shown, on success → adds the token and shows an
- "Added SYMBOL" flash message (no screen change)
+ "Added SYMBOL" flash message (no screen change). Until a refresh has read
+ the token's balance at every address, its lines read as after "Add" on
+ AddToken
- A duplicate, a scam-listed address, or a failed contract lookup → flash
message, no screen change
- "Add" by contract address with a token that the spoof rule refuses →
diff --git a/src/popup/index.js b/src/popup/index.js
index 22ea27b..f4db425 100644
--- a/src/popup/index.js
+++ b/src/popup/index.js
@@ -6,6 +6,7 @@ const {
saveState,
onSaveFailure,
takeStoredNetwork,
+ clearBalanceNotFetched,
loadState,
} = require("../shared/state");
const { StateUnusableError } = require("../shared/stateSchema");
@@ -95,6 +96,17 @@ let refreshTimer = null;
// removed meanwhile is left as it is. Blockscout may have answered before the
// user turned "Use Blockscout" off. So the save drops every token the user has
// not tracked if the stored setting is off by then.
+//
+// The save also deletes `balanceNotFetched` from each tracked token whose
+// balance this refresh read at every address. trackToken() in
+// src/popup/views/helpers.js sets that field. The save takes the field off the
+// matching tokens of the stored record (clearBalanceNotFetched() in
+// src/shared/state.js). The rest of the stored tracked token list stays as it
+// is.
+//
+// Once the save is done, the wallet list is drawn again. So are the balances
+// on AddressDetail or AddressToken when that screen is up. No other screen is
+// drawn again, DeleteAddress and its balance lines included.
async function doRefreshAndRender() {
if (refreshInFlight) {
refreshAgain = true;
@@ -105,7 +117,7 @@ async function doRefreshAndRender() {
await followStoredSettings();
const networkId = state.networkId;
const wallets = structuredClone(state.wallets);
- await Promise.all([
+ const [, readTokens] = await Promise.all([
refreshPrices(pageClosed.signal),
refreshBalances(
wallets,
@@ -137,10 +149,13 @@ async function doRefreshAndRender() {
if (!stored.useBlockscout) {
dropUntrackedTokens(state.wallets, stored.trackedTokens);
}
+ clearBalanceNotFetched(stored, readTokens);
state.lastBalanceRefresh = Date.now();
return true;
});
renderWalletList();
+ if (state.currentView === "address") addressDetail.renderBalances();
+ if (state.currentView === "address-token") addressToken.renderBalance();
} catch (e) {
// No caller waits for this function, so it must never reject. An
// unhandled rejection reports nothing. The save inside it reports its
diff --git a/src/popup/views/addToken.js b/src/popup/views/addToken.js
index d299a48..83be6b3 100644
--- a/src/popup/views/addToken.js
+++ b/src/popup/views/addToken.js
@@ -1,4 +1,11 @@
-const { $, showView, showFlash, escapeHtml, goBack } = require("./helpers");
+const {
+ $,
+ showView,
+ showFlash,
+ escapeHtml,
+ goBack,
+ trackToken,
+} = require("./helpers");
const { getTopTokens } = require("../../shared/tokenList");
const { state, saveState, currentNetwork } = require("../../shared/state");
const { lookupTokenInfo } = require("../../shared/balances");
@@ -88,7 +95,7 @@ function init(ctx) {
return;
}
log.infof("Adding token", info.symbol, contractAddr);
- state.trackedTokens.push({
+ trackToken({
address: contractAddr,
symbol: info.symbol,
decimals: info.decimals,
diff --git a/src/popup/views/addressDetail.js b/src/popup/views/addressDetail.js
index bc5808e..671edbf 100644
--- a/src/popup/views/addressDetail.js
+++ b/src/popup/views/addressDetail.js
@@ -67,11 +67,27 @@ function show() {
});
$("address-line").dataset.full = addr.address;
attachCopyHandlers($("address-line"));
- const usdTotal = formatAddressTotal(getAddressValue(addr));
- $("address-usd-total").innerHTML = escapeHtml(usdTotal) || " ";
const ensEl = $("address-ens");
// ENS is now shown inside renderAddressHtml, hide the separate element
ensEl.classList.add("hidden");
+ renderBalances();
+ // Only Blockscout reports the tokens the user has not tracked.
+ $("address-tokens-note").classList.toggle("hidden", !!state.useBlockscout);
+ renderSendTokenSelect(addr);
+ $("tx-list").innerHTML =
+ '
Loading...
';
+ showView("address");
+ loadTransactions(addr.address);
+}
+
+// The USD total and the balance list. Drawn again by each balance refresh
+// that returns while this screen is up (doRefreshAndRender() in
+// src/popup/index.js), which leaves the rest of the screen as it is.
+function renderBalances() {
+ const addr =
+ state.wallets[state.selectedWallet].addresses[state.selectedAddress];
+ const usdTotal = formatAddressTotal(getAddressValue(addr));
+ $("address-usd-total").innerHTML = escapeHtml(usdTotal) || " ";
$("address-balances").innerHTML = balanceLinesForAddress(
addr,
state.trackedTokens,
@@ -85,13 +101,6 @@ function show() {
ctx.showAddressToken();
});
});
- // Only Blockscout reports the tokens the user has not tracked.
- $("address-tokens-note").classList.toggle("hidden", !!state.useBlockscout);
- renderSendTokenSelect(addr);
- $("tx-list").innerHTML =
- 'Loading...
';
- showView("address");
- loadTransactions(addr.address);
}
let loadedTxs = [];
@@ -320,4 +329,4 @@ function init(_ctx) {
exportPrivkey.init();
}
-module.exports = { init, show };
+module.exports = { init, show, renderBalances };
diff --git a/src/popup/views/addressToken.js b/src/popup/views/addressToken.js
index 33584cf..40545a5 100644
--- a/src/popup/views/addressToken.js
+++ b/src/popup/views/addressToken.js
@@ -50,37 +50,18 @@ function show() {
const addr = wallet.addresses[state.selectedAddress];
const ai = state.selectedAddress;
const tokenId = state.selectedToken;
-
- // Determine token symbol and balance
- let symbol, amount, price;
const knownToken = TOKEN_BY_ADDRESS.get(tokenId.toLowerCase());
- if (tokenId === "ETH") {
- symbol = nativeCurrency();
- amount = parseFloat(addr.balance || "0");
- price = getPrice("ETH");
- } else {
- const tb = (addr.tokenBalances || []).find(
- (t) => t.address.toLowerCase() === tokenId.toLowerCase(),
- );
- symbol = resolveSymbol(
- tokenId,
- addr.tokenBalances,
- state.trackedTokens,
- );
- // null when the scale is unknown: no quantity to show, and none to
- // price. balanceLine() states that rather than printing 0.0000.
- amount = tb ? unknownableAmount(tb.balance) : 0;
- price = getPrice(tokenId);
- }
-
- currentSymbol = symbol;
+ currentSymbol =
+ tokenId === "ETH"
+ ? nativeCurrency()
+ : resolveSymbol(tokenId, addr.tokenBalances, state.trackedTokens);
$("address-token-title").textContent =
wallet.name +
" \u2014 Address " +
(ai + 1) +
" \u2014 " +
- displaySymbol(symbol);
+ displaySymbol(currentSymbol);
// Blockie
const blockieEl = $("address-token-jazzicon");
@@ -102,13 +83,7 @@ function show() {
$("address-token-line").dataset.full = addr.address;
attachCopyHandlers($("address-token-line"));
- // USD total for this token only
- const usdVal = price && amount !== null ? amount * price : null;
- const usdStr = formatUsd(usdVal);
- $("address-token-usd-total").innerHTML = escapeHtml(usdStr) || " ";
-
- // Single token balance line (no tokenId — not clickable here)
- $("address-token-balance").innerHTML = balanceLine(symbol, amount, price);
+ renderBalance();
// Token contract details (ERC-20 only)
const contractInfo = $("address-token-contract-info");
@@ -170,6 +145,47 @@ function show() {
loadTransactions(addr.address, tokenId);
}
+// The USD total and the balance line of the selected token. Drawn again by
+// each balance refresh that returns while this screen is up
+// (doRefreshAndRender() in src/popup/index.js), which leaves the rest of the
+// screen as it is.
+function renderBalance() {
+ const addr =
+ state.wallets[state.selectedWallet].addresses[state.selectedAddress];
+ const tokenId = state.selectedToken;
+ let amount, price;
+ if (tokenId === "ETH") {
+ amount = parseFloat(addr.balance || "0");
+ price = getPrice("ETH");
+ } else {
+ const id = tokenId.toLowerCase();
+ const tb = (addr.tokenBalances || []).find(
+ (t) => t.address.toLowerCase() === id,
+ );
+ const tracked = state.trackedTokens.find(
+ (t) => t.address.toLowerCase() === id,
+ );
+ // null when the scale is unknown, or when no refresh has read the
+ // balance of a token just added: no quantity to show, and none to
+ // price. balanceLine() states that rather than printing 0.0000.
+ if (tb) amount = unknownableAmount(tb.balance);
+ else amount = tracked && tracked.balanceNotFetched ? null : 0;
+ price = getPrice(tokenId);
+ }
+
+ // USD total for this token only
+ const usdVal = price && amount !== null ? amount * price : null;
+ const usdStr = formatUsd(usdVal);
+ $("address-token-usd-total").innerHTML = escapeHtml(usdStr) || " ";
+
+ // Single token balance line (no tokenId — not clickable here)
+ $("address-token-balance").innerHTML = balanceLine(
+ currentSymbol,
+ amount,
+ price,
+ );
+}
+
async function loadTransactions(address, tokenId) {
const networkId = state.networkId;
try {
@@ -383,4 +399,4 @@ function init(_ctx) {
});
}
-module.exports = { init, show };
+module.exports = { init, show, renderBalance };
diff --git a/src/popup/views/helpers.js b/src/popup/views/helpers.js
index 261fb8e..2753b1f 100644
--- a/src/popup/views/helpers.js
+++ b/src/popup/views/helpers.js
@@ -418,6 +418,15 @@ function balanceLine(symbol, amount, price, tokenId) {
);
}
+// Tracks `token` on the active network, as every add path on AddToken and
+// SettingsAddToken does. The tracked record carries `balanceNotFetched: true`
+// until a balance refresh has read the token's balance at every address
+// (doRefreshAndRender() in src/popup/index.js). Until then, an address with no
+// balance row for the token shows `quantity unknown` for it, not 0.0000.
+function trackToken(token) {
+ state.trackedTokens.push({ ...token, balanceNotFetched: true });
+}
+
function balanceLinesForAddress(addr, trackedTokens, showZero) {
let html = balanceLine(
nativeCurrency(),
@@ -453,7 +462,7 @@ function balanceLinesForAddress(addr, trackedTokens, showZero) {
if (isSpoofedSymbol(t.symbol, t.address)) continue;
html += balanceLine(
t.symbol,
- 0,
+ t.balanceNotFetched ? null : 0,
getPrice(t.address),
t.address.toLowerCase(),
);
@@ -833,6 +842,7 @@ module.exports = {
flashCopyFeedback,
balanceLine,
balanceLinesForAddress,
+ trackToken,
addressHoldsFunds,
keyMaterialName,
unknownableAmount,
diff --git a/src/popup/views/send.js b/src/popup/views/send.js
index a699562..4228bd6 100644
--- a/src/popup/views/send.js
+++ b/src/popup/views/send.js
@@ -198,13 +198,21 @@ function renderSendTokenSelect(addr) {
// null rather than "0": this screen states it as unknown, as the confirmation
// screen does, and Max fills nothing in for it. Only a stored quantity is
// withdrawn: the "0" for a token that has no row at all is an absence of
-// holdings, which is true at every scale.
+// holdings, which is true at every scale. A token with no row whose tracked
+// record carries `balanceNotFetched` gets null instead, because no refresh has
+// read its balance.
function tokenBalanceAndDecimals(addr, token) {
const tb = (addr.tokenBalances || []).find(
(t) => t.address.toLowerCase() === token.toLowerCase(),
);
const tokenDecimals = resolveTokenDecimals(token, tokenSourcesByChainId());
- if (!tb) return { tokenBalance: "0", tokenDecimals };
+ if (!tb) {
+ const tracked = state.trackedTokens.find(
+ (t) => t.address.toLowerCase() === token.toLowerCase(),
+ );
+ const notRead = tracked && tracked.balanceNotFetched;
+ return { tokenBalance: notRead ? null : "0", tokenDecimals };
+ }
if (tokenDecimals === null) return { tokenBalance: null, tokenDecimals };
return { tokenBalance: tb.balance ?? null, tokenDecimals };
}
diff --git a/src/popup/views/settingsAddToken.js b/src/popup/views/settingsAddToken.js
index 38ed93f..0aeeddd 100644
--- a/src/popup/views/settingsAddToken.js
+++ b/src/popup/views/settingsAddToken.js
@@ -1,4 +1,11 @@
-const { $, showView, showFlash, escapeHtml, goBack } = require("./helpers");
+const {
+ $,
+ showView,
+ showFlash,
+ escapeHtml,
+ goBack,
+ trackToken,
+} = require("./helpers");
const { getTopTokens } = require("../../shared/tokenList");
const { state, saveState, currentNetwork } = require("../../shared/state");
const { lookupTokenInfo } = require("../../shared/balances");
@@ -44,7 +51,7 @@ function renderTop10() {
decimals: parseInt(btn.dataset.decimals, 10),
name: btn.dataset.name || btn.dataset.symbol,
};
- state.trackedTokens.push(token);
+ trackToken(token);
await saveState();
showFlash("Added " + token.symbol);
renderTop10();
@@ -111,7 +118,7 @@ function init(_ctx) {
decimals: parseInt(opt.dataset.decimals, 10),
name: opt.dataset.name || opt.dataset.symbol,
};
- state.trackedTokens.push(token);
+ trackToken(token);
await saveState();
showFlash("Added " + token.symbol);
renderTop10();
@@ -167,7 +174,7 @@ function init(_ctx) {
return;
}
log.infof("Adding token", info.symbol, addr);
- state.trackedTokens.push({
+ trackToken({
address: addr,
symbol: info.symbol,
decimals: info.decimals,
diff --git a/src/shared/balances.js b/src/shared/balances.js
index fe4e4bc..33f73ae 100644
--- a/src/shared/balances.js
+++ b/src/shared/balances.js
@@ -247,7 +247,11 @@ async function fetchTokenBalances(
// it has none. The failure is logged. A balance of zero gets no entry, as a
// zero Blockscout reports gets none: the balance lists take a tracked token's
// zero from the tracked list. A token whose symbol the spoof rule refuses gets
-// no entry either, as in fetchTokenBalances().
+// no entry either, as in fetchTokenBalances(), and its balance is not read.
+//
+// Resolves with `rows`, the entries, and `unread`, the tracked tokens whose
+// balance was not read, as lowercased contract addresses. A token is unread
+// when the spoof rule refuses it or its balanceOf fails.
//
// `signal` is as for fetchTokenBalances().
async function fetchTrackedTokenBalances(
@@ -288,12 +292,16 @@ async function fetchTrackedTokenBalances(
const reported = explorerRows ?? previous;
const rows = [];
+ const unread = trackedTokens
+ .filter((token) => !listed.includes(token))
+ .map((token) => token.address.toLowerCase());
listed.forEach((token, i) => {
const lower = token.address.toLowerCase();
const entryIn = (list) =>
list.find((row) => String(row.address).toLowerCase() === lower);
const raw = balances[i];
if (raw === null) {
+ unread.push(lower);
const kept = entryIn(previous) || entryIn(explorerRows || []);
if (kept) rows.push(kept);
return;
@@ -314,12 +322,15 @@ async function fetchTrackedTokenBalances(
holders: entry ? entry.holders : null,
});
});
- return rows;
+ return { rows, unread };
}
// Fetch ETH balances, ENS names, and ERC-20 token balances for all addresses.
// `trackedTokens` are the tokens tracked on the active network. `signal` is as
// for fetchTokenBalances().
+//
+// Resolves with the lowercased contract addresses of the tracked tokens whose
+// balance it read at every address.
async function refreshBalances(
wallets,
rpcUrl,
@@ -332,6 +343,8 @@ async function refreshBalances(
const provider = getProvider(rpcUrl, networkId);
const tracked = trackedTokens || [];
const trackedSet = new Set(tracked.map((t) => t.address.toLowerCase()));
+ // The tracked tokens whose balance was not read at some address.
+ const unread = new Set();
const updates = [];
for (const wallet of wallets) {
@@ -406,12 +419,14 @@ async function refreshBalances(
signal,
),
explorerRequest,
- ]).then(([trackedRows, explorerRows]) => {
+ ]).then(([trackedResult, explorerRows]) => {
+ for (const lower of trackedResult.unread) unread.add(lower);
const untrackedRows = (explorerRows ?? previous).filter(
(row) =>
!trackedSet.has(String(row.address).toLowerCase()),
);
- addr.tokenBalances = trackedRows.concat(untrackedRows);
+ addr.tokenBalances =
+ trackedResult.rows.concat(untrackedRows);
log.debugf(
"Token balances",
addr.address,
@@ -425,6 +440,7 @@ async function refreshBalances(
await Promise.all(updates);
log.debugf("refreshBalances done");
+ return [...trackedSet].filter((lower) => !unread.has(lower));
}
// Drops the balance of each token not in `trackedTokens` from every address.
diff --git a/src/shared/state.js b/src/shared/state.js
index 45cacd4..78cc9e6 100644
--- a/src/shared/state.js
+++ b/src/shared/state.js
@@ -129,7 +129,9 @@ function currentNetwork() {
// loadState() finished, or when its last successful saveState() began.
// takeStoredNetwork(), takeStoredTrackedTokens() and
// takeStoredNetworkTrackedTokens() also write into it what this page has taken
-// from the stored record since then. saveState() diffs the live state against
+// from the stored record since then. `baseline` also lacks any
+// `balanceNotFetched` field that a balance refresh has deleted since
+// (clearBalanceNotFetched()). saveState() diffs the live state against
// `baseline` to find only the fields THIS page changed since.
//
// Deep-cloned, not a reference: callers mutate persisted objects and arrays
@@ -701,6 +703,27 @@ function takeStoredNetworkTrackedTokens(stored) {
);
}
+// Deletes `balanceNotFetched` from each tracked token whose lowercased
+// contract address is in `addresses`. The field goes from `stored`, from
+// `state` and from `baseline` alike. `stored` is the record a save of this
+// page has just read (doRefreshAndRender() in src/popup/index.js calls this
+// from `beforeWrite`). So the deletion is not a change this page made to its
+// tracked tokens. Unless this page has changed its tracked tokens itself, the
+// save then writes the stored list with only the field gone.
+function clearBalanceNotFetched(stored, addresses) {
+ for (const list of [
+ stored.trackedTokens,
+ rawState.trackedTokens,
+ baseline.trackedTokens,
+ ]) {
+ for (const token of list) {
+ if (addresses.includes(token.address.toLowerCase())) {
+ delete token.balanceNotFetched;
+ }
+ }
+ }
+}
+
// Rejects with StateUnusableError for a stored record this build cannot make
// sense of. Nothing is assigned and `loaded` stays false in that case, so a
// caller that ignores the rejection gets StateNotLoadedError on the first
@@ -749,6 +772,7 @@ module.exports = {
takeStoredNetwork,
takeStoredTrackedTokens,
takeStoredNetworkTrackedTokens,
+ clearBalanceNotFetched,
loadState,
currentAddress,
currentNetwork,
diff --git a/tests/addTokenLateAnswer.test.js b/tests/addTokenLateAnswer.test.js
index 8d0bda9..84ae220 100644
--- a/tests/addTokenLateAnswer.test.js
+++ b/tests/addTokenLateAnswer.test.js
@@ -16,6 +16,7 @@ const TOKEN = {
symbol: "TST",
decimals: 18,
name: "Test Token",
+ balanceNotFetched: true,
};
// A second contract address, typed after AddToken was opened again.
diff --git a/tests/addedTokenBalance.test.js b/tests/addedTokenBalance.test.js
new file mode 100644
index 0000000..c3ef97b
--- /dev/null
+++ b/tests/addedTokenBalance.test.js
@@ -0,0 +1,458 @@
+// After "Add" on AddToken, AddressDetail lists the new token. Its line reads
+// `quantity unknown`, not 0.0000, until a balance refresh has read the token's
+// balance at every address, even if the popup closes and opens again
+// meanwhile. A token added on SettingsAddToken reads the same wherever it is
+// listed. A refresh that returns draws the balances on AddressDetail or
+// AddressToken again, so the fetched balance shows without the screen being
+// reopened (https://git.eeqj.de/sneak/AutistMask/issues/516).
+//
+// The tracked token's record carries `balanceNotFetched` until a refresh has
+// read the token's balance. Removing the token removes that record. A network
+// switch keeps each network's records apart. A refresh takes the field off the
+// stored record's tokens and keeps another page's changes to the stored list.
+//
+// Driven through the real popup over a stored profile on Ethereum Mainnet,
+// with "Show tracked tokens with zero balance" on, as it is by default.
+
+const {
+ bootPopup,
+ cleanupPopup,
+ unversionedValidProfile,
+} = require("./support/popupBoot");
+
+// Tracked by nobody until a test adds it.
+const NEW_TOKEN = "0xBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB";
+
+// A holding of a million NEW_TOKEN, as refreshBalances() stores it.
+const MILLION = {
+ address: NEW_TOKEN,
+ name: "New Token",
+ symbol: "NEW",
+ decimals: 18,
+ balance: "1000000.0",
+ holders: 3,
+};
+
+// NEW_TOKEN as "Add" on AddToken tracks it, once a refresh has read its
+// balance.
+const NEW_TRACKED = {
+ address: NEW_TOKEN,
+ symbol: "NEW",
+ name: "New Token",
+ decimals: 18,
+};
+
+// A token that another popup page tracks.
+const OTHER_TOKEN = {
+ address: "0xCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCCC",
+ symbol: "OTH",
+ name: "Other Token",
+ decimals: 18,
+};
+
+afterEach(() => {
+ cleanupPopup();
+});
+
+// The popup reopened onto AddressDetail for the one address.
+function bootOnAddress(extra) {
+ return bootPopup(
+ unversionedValidProfile({
+ currentView: "address",
+ selectedWallet: 0,
+ selectedAddress: 0,
+ ...extra,
+ }),
+ );
+}
+
+// Holds the next balance refresh open. The returned function lets it finish
+// with the address's ETH balance at 2 and `tokenBalances` as its token
+// entries, having read the balance of every token tracked when it started.
+// Given null, it finishes having read no token's balance, as when every
+// balanceOf fails, and leaves the token entries as they were.
+function holdNextRefresh(page) {
+ const { refreshBalances } = require("../src/shared/balances");
+ let release;
+ refreshBalances.mockImplementationOnce(
+ async (wallets, rpcUrl, explorerUrl, trackedTokens) => {
+ const read = trackedTokens.map((t) => t.address.toLowerCase());
+ const tokenBalances = await new Promise((resolve) => {
+ release = resolve;
+ });
+ wallets[0].addresses[0].balance = "2.0";
+ if (tokenBalances === null) return [];
+ wallets[0].addresses[0].tokenBalances = tokenBalances;
+ return read;
+ },
+ );
+ return async (tokenBalances) => {
+ release(tokenBalances);
+ await page.settle();
+ };
+}
+
+// Runs the ten-second refresh to the end, finishing it as holdNextRefresh()
+// does with `tokenBalances`.
+async function refreshOnce(page, tokenBalances) {
+ const finish = holdNextRefresh(page);
+ const ticking = page.tick();
+ await page.settle();
+ await finish(tokenBalances);
+ await ticking;
+}
+
+// "+ Token", NEW_TOKEN typed in, and "Add", with the contract lookup
+// answering at once.
+async function addNewToken(page) {
+ const { lookupTokenInfo } = require("../src/shared/balances");
+ lookupTokenInfo.mockResolvedValueOnce({
+ symbol: "NEW",
+ decimals: 18,
+ name: "New Token",
+ });
+ await page.click("btn-add-token");
+ page.node("add-token-address").value = NEW_TOKEN;
+ await page.click("btn-add-token-confirm");
+}
+
+// Settings' "+ Add token", NEW_TOKEN typed in, and "Add", with the contract
+// lookup answering at once. It ends on SettingsAddToken.
+async function addOnSettingsAddToken(page) {
+ const { lookupTokenInfo } = require("../src/shared/balances");
+ lookupTokenInfo.mockResolvedValueOnce({
+ symbol: "NEW",
+ decimals: 18,
+ name: "New Token",
+ });
+ await page.click("btn-settings-add-token");
+ page.node("settings-addtoken-address").value = NEW_TOKEN;
+ await page.click("btn-settings-addtoken-manual");
+}
+
+// The "[x]" button Settings draws beside the first tracked token. The page
+// stub does not turn the markup Settings draws into elements. So the list
+// hands Settings a stand-in for that button each time it is drawn, and
+// `press()` presses the last one drawn. Call it before Settings opens.
+function firstRemoveButton(page) {
+ const button = { press: null };
+ page.node("settings-tracked-tokens").querySelectorAll = () => [
+ {
+ dataset: { idx: "0" },
+ addEventListener: (name, fn) => {
+ button.press = fn;
+ },
+ },
+ ];
+ return button;
+}
+
+// Picks `networkId` in the Settings network selector, as the user does.
+async function chooseNetwork(page, networkId) {
+ const select = page.node("settings-network");
+ select.value = networkId;
+ for (const fn of select.listeners.change) await fn();
+ await page.settle();
+}
+
+// Another popup page saves `trackedTokens` as the tracked token list. This
+// page is not told of the write.
+function anotherPageTracks(page, trackedTokens) {
+ page.storage.write("autistmask", {
+ ...page.storage.read("autistmask"),
+ trackedTokens,
+ });
+}
+
+// NEW_TOKEN's row in the balance list in element `id`, AddressDetail's unless
+// another is named, as markup.
+function newTokenRow(page, id = "address-balances") {
+ return page
+ .node(id)
+ .innerHTML.split("
+ row.includes('data-token="' + NEW_TOKEN.toLowerCase() + '"'),
+ );
+}
+
+// AddressDetail for the one address opened from Home, as its `[info]` button
+// does.
+async function openAddress(page) {
+ const { state } = require("../src/shared/state");
+ state.selectedWallet = 0;
+ state.selectedAddress = 0;
+ require("../src/popup/views/helpers").pushCurrentView();
+ require("../src/popup/views/addressDetail").show();
+ await page.settle();
+}
+
+// AddressToken for `token` opened from AddressDetail, as tapping the token's
+// row there does.
+async function openToken(page, token) {
+ const { state } = require("../src/shared/state");
+ state.selectedToken = token;
+ require("../src/popup/views/helpers").pushCurrentView();
+ require("../src/popup/views/addressToken").show();
+ await page.settle();
+}
+
+test("until the refresh Add started returns, the new token's line reads quantity unknown", async () => {
+ const page = await bootOnAddress();
+ expect(page.pageErrors).toEqual([]);
+ const finish = holdNextRefresh(page);
+
+ await addNewToken(page);
+
+ expect(page.visibleViews()).toEqual(["address"]);
+ const row = newTokenRow(page);
+ expect(row).toContain("quantity unknown");
+ expect(row).not.toContain("0.0000");
+ await finish([MILLION]);
+});
+
+test("once that refresh returns, AddressDetail shows the fetched balance without being reopened", async () => {
+ const page = await bootOnAddress();
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+ // Stands for the transactions on the screen, which only show() replaces.
+ page.node("tx-list").innerHTML = "transactions on the screen";
+
+ await finish([MILLION]);
+
+ expect(page.visibleViews()).toEqual(["address"]);
+ expect(newTokenRow(page)).toContain("1000000.0000");
+ // Only the balances are drawn again, not the transaction list.
+ expect(page.node("tx-list").innerHTML).toBe("transactions on the screen");
+});
+
+test("a refresh already running when the token is added does not count as reading its balance", async () => {
+ const page = await bootOnAddress();
+ const finishRunning = holdNextRefresh(page);
+ const ticking = page.tick();
+ await page.settle();
+
+ await addNewToken(page);
+ await finishRunning([]);
+ await ticking;
+
+ expect(newTokenRow(page)).toContain("quantity unknown");
+
+ // The next one reads the token's balance, which is zero.
+ await refreshOnce(page, []);
+
+ expect(newTokenRow(page)).toContain("0.0000");
+});
+
+test("closed and opened again before any refresh has read its balance, the new token's line still reads quantity unknown", async () => {
+ const closed = await bootOnAddress();
+ holdNextRefresh(closed);
+ await addNewToken(closed);
+ cleanupPopup();
+
+ // The refresh this boot starts reads no token's balance: the boot's stub
+ // of refreshBalances() reads none.
+ const page = await bootPopup(undefined, { storage: closed.storage });
+ expect(page.visibleViews()).toEqual(["address"]);
+ const row = newTokenRow(page);
+ expect(row).toContain("quantity unknown");
+ expect(row).not.toContain("0.0000");
+
+ await refreshOnce(page, [MILLION]);
+
+ expect(newTokenRow(page)).toContain("1000000.0000");
+});
+
+test("a refresh that could not read the new token's balance leaves its line quantity unknown", async () => {
+ const page = await bootOnAddress();
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+
+ await finish(null);
+
+ // Drawn again: the ETH balance it fetched is shown.
+ expect(page.node("address-balances").innerHTML).toContain("2.0000");
+ expect(newTokenRow(page)).toContain("quantity unknown");
+});
+
+test("AddressToken for the new token reads quantity unknown until the refresh returns, then the fetched balance", async () => {
+ const page = await bootOnAddress();
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+
+ await openToken(page, NEW_TOKEN.toLowerCase());
+
+ expect(page.visibleViews()).toEqual(["address-token"]);
+ const before = page.node("address-token-balance").innerHTML;
+ expect(before).toContain("quantity unknown");
+ expect(before).not.toContain("0.0000");
+
+ await finish([MILLION]);
+
+ expect(page.node("address-token-balance").innerHTML).toContain(
+ "1000000.0000",
+ );
+});
+
+test("Send for the new token reads its balance as unknown before any refresh has read it, and Max says so", async () => {
+ const page = await bootOnAddress();
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+ await openToken(page, NEW_TOKEN.toLowerCase());
+
+ await page.click("btn-address-token-send");
+
+ expect(page.visibleViews()).toEqual(["send"]);
+ expect(page.text("send-balance")).toBe("Current balance: unknown (NEW)");
+
+ await page.click("btn-send-max");
+
+ expect(page.text("flash-msg")).toBe("This token's balance is unknown.");
+ await finish([MILLION]);
+});
+
+test("a token added by contract address on SettingsAddToken reads quantity unknown until the refresh returns", async () => {
+ const page = await bootPopup(unversionedValidProfile());
+ expect(page.visibleViews()).toEqual(["main"]);
+ const finish = holdNextRefresh(page);
+
+ await page.click("btn-settings");
+ await addOnSettingsAddToken(page);
+ await page.click("btn-settings-addtoken-back");
+ await page.click("btn-settings-back");
+ expect(page.visibleViews()).toEqual(["main"]);
+ const onHome = newTokenRow(page, "wallet-list");
+ expect(onHome).toContain("quantity unknown");
+ expect(onHome).not.toContain("0.0000");
+
+ await openAddress(page);
+ const onAddress = newTokenRow(page);
+ expect(onAddress).toContain("quantity unknown");
+ expect(onAddress).not.toContain("0.0000");
+
+ await finish([MILLION]);
+
+ expect(newTokenRow(page)).toContain("1000000.0000");
+});
+
+test("a refresh that returns while AddressToken is up draws its balance again", async () => {
+ const page = await bootPopup(
+ unversionedValidProfile({
+ currentView: "address-token",
+ viewStack: ["main", "address"],
+ selectedWallet: 0,
+ selectedAddress: 0,
+ selectedToken: "ETH",
+ }),
+ );
+ expect(page.visibleViews()).toEqual(["address-token"]);
+ expect(page.node("address-token-balance").innerHTML).toContain("1.5000");
+
+ await refreshOnce(page, null);
+
+ expect(page.node("address-token-balance").innerHTML).toContain("2.0000");
+});
+
+test("a token removed before any refresh has read its balance leaves nothing stored, and adding it again tracks it once", async () => {
+ const page = await bootPopup(unversionedValidProfile());
+ const removeButton = firstRemoveButton(page);
+ const stored = () => page.storage.read("autistmask");
+
+ await page.click("btn-settings");
+ await addOnSettingsAddToken(page);
+ // Settings opened again from Home draws its list with the token in it.
+ await page.click("btn-settings-addtoken-back");
+ await page.click("btn-settings-back");
+ await page.click("btn-settings");
+ await removeButton.press();
+ await page.settle();
+
+ expect(stored().trackedTokens).toEqual([]);
+ // Nothing in the stored record names the token any more.
+ expect(JSON.stringify(stored()).toLowerCase()).not.toContain(
+ NEW_TOKEN.toLowerCase(),
+ );
+
+ await addOnSettingsAddToken(page);
+ // A second add of the same token is refused.
+ page.node("settings-addtoken-address").value = NEW_TOKEN;
+ await page.click("btn-settings-addtoken-manual");
+
+ const added = { address: NEW_TOKEN, symbol: "NEW", decimals: 18 };
+ expect(stored().trackedTokens).toEqual([
+ { ...added, name: "New Token", balanceNotFetched: true },
+ ]);
+
+ await refreshOnce(page, []);
+
+ expect(stored().trackedTokens).toEqual([{ ...added, name: "New Token" }]);
+});
+
+test("a refresh on another network leaves the new token reading quantity unknown on its own", async () => {
+ // Sepolia tracks a token at the same contract address, also just added.
+ const sepoliaToken = {
+ address: NEW_TOKEN,
+ symbol: "SEP",
+ name: "Sepolia Token",
+ decimals: 18,
+ };
+ const page = await bootOnAddress({
+ networkTrackedTokens: {
+ sepolia: [{ ...sepoliaToken, balanceNotFetched: true }],
+ },
+ });
+ const finishMainnet = holdNextRefresh(page);
+ await addNewToken(page);
+
+ await page.click("btn-settings");
+ await chooseNetwork(page, "sepolia");
+ // The refresh Add started returns after the switch and writes nothing.
+ await finishMainnet([MILLION]);
+ // Sepolia's refresh reads the balance of its own token.
+ await refreshOnce(page, []);
+
+ const stored = page.storage.read("autistmask");
+ expect(stored.trackedTokens).toEqual([sepoliaToken]);
+ expect(stored.networkTrackedTokens.mainnet[0].balanceNotFetched).toBe(true);
+
+ await chooseNetwork(page, "mainnet");
+ await page.click("btn-settings-back");
+
+ expect(page.visibleViews()).toEqual(["address"]);
+ expect(newTokenRow(page)).toContain("quantity unknown");
+
+ await refreshOnce(page, [MILLION]);
+
+ expect(newTokenRow(page)).toContain("1000000.0000");
+});
+
+test("a refresh keeps a token that another page added after this page loaded", async () => {
+ const page = await bootOnAddress();
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+ const otherAdded = { ...OTHER_TOKEN, balanceNotFetched: true };
+ anotherPageTracks(page, [
+ { ...NEW_TRACKED, balanceNotFetched: true },
+ otherAdded,
+ ]);
+
+ await finish([MILLION]);
+
+ expect(page.storage.read("autistmask").trackedTokens).toEqual([
+ NEW_TRACKED,
+ otherAdded,
+ ]);
+ expect(newTokenRow(page)).toContain("1000000.0000");
+});
+
+test("a refresh does not bring back a token that another page removed after this page loaded", async () => {
+ const page = await bootOnAddress({ trackedTokens: [OTHER_TOKEN] });
+ const finish = holdNextRefresh(page);
+ await addNewToken(page);
+ anotherPageTracks(page, [{ ...NEW_TRACKED, balanceNotFetched: true }]);
+
+ await finish([MILLION]);
+
+ expect(page.storage.read("autistmask").trackedTokens).toEqual([
+ NEW_TRACKED,
+ ]);
+});
diff --git a/tests/balanceRefreshNetworkSwitch.test.js b/tests/balanceRefreshNetworkSwitch.test.js
index 7a6677d..920a6f8 100644
--- a/tests/balanceRefreshNetworkSwitch.test.js
+++ b/tests/balanceRefreshNetworkSwitch.test.js
@@ -45,6 +45,7 @@ async function holdRefresh(page) {
holders: null,
},
];
+ return [];
});
const ticking = page.tick();
await page.settle();
diff --git a/tests/blockscoutEtherscanSettings.test.js b/tests/blockscoutEtherscanSettings.test.js
index 747a445..de8908a 100644
--- a/tests/blockscoutEtherscanSettings.test.js
+++ b/tests/blockscoutEtherscanSettings.test.js
@@ -720,6 +720,7 @@ describe("the popup", () => {
tokenRow(TRACKED, "TRK"),
tokenRow(UNTRACKED, "TST"),
];
+ return [];
});
const ticking = page.tick();
await page.settle();
@@ -764,6 +765,7 @@ describe("the popup", () => {
addr.tokenBalances = after
? [tokenRow(TRACKED, "TRK"), tokenRow(UNTRACKED, "TST")]
: [tokenRow(TRACKED, "TRK")];
+ return [];
});
anotherPageSaves(page, { useBlockscout: after });
diff --git a/tests/e2e/run.js b/tests/e2e/run.js
index a2112e1..f8f5fa6 100644
--- a/tests/e2e/run.js
+++ b/tests/e2e/run.js
@@ -969,7 +969,8 @@ test("a token added by its contract address is listed on the address screen (#15
await visible(env.page, "#view-address");
// No wait: the confirm renders the balance list before it shows the
- // screen, and nothing renders the list again while the screen is up.
+ // screen. The refresh it starts draws the list again, with the row still
+ // in it.
const row = env.page.locator(
'#address-balances [data-token="' + STUB_TOKEN.address + '"]',
{ hasText: STUB_TOKEN.symbol },
diff --git a/tests/firstWalletRefresh.test.js b/tests/firstWalletRefresh.test.js
index 232cb30..5bd9972 100644
--- a/tests/firstWalletRefresh.test.js
+++ b/tests/firstWalletRefresh.test.js
@@ -48,6 +48,7 @@ async function boot(stored) {
for (const wallet of wallets) {
for (const addr of wallet.addresses) addr.balance = chainBalance;
}
+ return [];
});
return { page, refreshBalances };
}
diff --git a/tests/flashLine.test.js b/tests/flashLine.test.js
index ce00781..dd76583 100644
--- a/tests/flashLine.test.js
+++ b/tests/flashLine.test.js
@@ -88,6 +88,7 @@ describe.each([
showFlash: (msg) => flashes.push(msg),
escapeHtml: (s) => s,
goBack: () => {},
+ trackToken: () => {},
}));
require("../src/popup/views/" + view).init({
diff --git a/tests/support/popupBoot.js b/tests/support/popupBoot.js
index 22d79b6..128d0fe 100644
--- a/tests/support/popupBoot.js
+++ b/tests/support/popupBoot.js
@@ -279,10 +279,11 @@ async function bootPopup(stored, options) {
getAddressValue: () => ({ usd: null, partial: false }),
}));
// scanForAddresses() answers in the real one's shape, a scan that found
- // no used address: AddWallet reads `addresses` from it.
+ // no used address: AddWallet reads `addresses` from it. refreshBalances()
+ // answers as a refresh that read no tracked token's balance.
jest.doMock("../../src/shared/balances", () => ({
fetchTokenBalances: jest.fn(async () => []),
- refreshBalances: jest.fn(async () => {}),
+ refreshBalances: jest.fn(async () => []),
dropUntrackedTokens: jest.requireActual("../../src/shared/balances")
.dropUntrackedTokens,
lookupTokenInfo: jest.fn(async () => null),
diff --git a/tests/trackedTokenBalances.test.js b/tests/trackedTokenBalances.test.js
index d00291f..11245f1 100644
--- a/tests/trackedTokenBalances.test.js
+++ b/tests/trackedTokenBalances.test.js
@@ -307,3 +307,31 @@ test("a tracked token wearing a known symbol from another contract is not listed
);
expect(first.tokenBalances).toEqual([]);
});
+
+test("the refresh resolves with each tracked token whose balance it read at every address", async () => {
+ const flaky = "0x9999999999999999999999999999999999999999";
+ holdings[TOKEN] = { [FIRST]: 2000000n, [SECOND]: 0n };
+ // balanceOf answers at FIRST and fails at SECOND.
+ holdings[flaky] = { [FIRST]: 2000000n };
+ const read = await refreshBalances(
+ [
+ {
+ addresses: [
+ { address: FIRST, tokenBalances: [] },
+ { address: SECOND, tokenBalances: [] },
+ ],
+ },
+ ],
+ RPC_URL,
+ EXPLORER_URL,
+ [
+ TRACKED,
+ { address: flaky, symbol: "FLK", name: "Flaky", decimals: 6 },
+ // A known symbol from another contract: its balance is not read.
+ { address: OTHER, symbol: "USDC", name: "Fake", decimals: 6 },
+ ],
+ "mainnet",
+ new AbortController().signal,
+ );
+ expect(read).toEqual([TOKEN]);
+});
diff --git a/tests/trackedTokensPerNetwork.test.js b/tests/trackedTokensPerNetwork.test.js
index 922ae29..d15fd6e 100644
--- a/tests/trackedTokensPerNetwork.test.js
+++ b/tests/trackedTokensPerNetwork.test.js
@@ -162,7 +162,9 @@ test("a token added by contract address on Sepolia is tracked on Sepolia only",
"sepolia",
expect.anything(),
);
- expect(stored().trackedTokens).toEqual([SEPOLIA_TOKEN]);
+ // No refresh has read the new token's balance yet.
+ const added = { ...SEPOLIA_TOKEN, balanceNotFetched: true };
+ expect(stored().trackedTokens).toEqual([added]);
await boot.click("btn-settings");
await chooseNetwork(boot, "mainnet");
@@ -174,7 +176,7 @@ test("a token added by contract address on Sepolia is tracked on Sepolia only",
await chooseNetwork(boot, "sepolia");
expect(stored().networkId).toBe("sepolia");
- expect(stored().trackedTokens).toEqual([SEPOLIA_TOKEN]);
+ expect(stored().trackedTokens).toEqual([added]);
const onSepolia = boot.node("settings-tracked-tokens").innerHTML;
expect(onSepolia).toContain("TST");
expect(onSepolia).not.toContain("DAI");
@@ -306,6 +308,7 @@ test("a token added before this page takes another page's switch is tracked on E
symbol: "NEW",
decimals: 18,
name: "New Token",
+ balanceNotFetched: true,
},
]);
});
@@ -343,6 +346,7 @@ test("a token added before this page takes another page's switch joins that page
symbol: "NEW",
decimals: 18,
name: "New Token",
+ balanceNotFetched: true,
},
]);
});
@@ -518,6 +522,7 @@ test("a token added while a switch in Settings waits for the lock stays tracked
symbol: "NEW",
decimals: 18,
name: "New Token",
+ balanceNotFetched: true,
},
]);
});
@@ -578,6 +583,7 @@ test("a token added while a switch in Settings waits for another page's switch s
symbol: "NEW",
decimals: 18,
name: "New Token",
+ balanceNotFetched: true,
},
]);
});
@@ -632,6 +638,7 @@ test("a token added while this page waits to take another page's switch stays tr
symbol: "NEW",
decimals: 18,
name: "New Token",
+ balanceNotFetched: true,
},
]);
});