fix(e2e tests): add missing fields in select

matter-labs · Apr 15, 2024 · d3916ed · d3916ed
1 parent 7fa6341
commit d3916ed
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/packages/api/src/balance/balance.service.ts b/packages/api/src/balance/balance.service.ts
@@ -63,6 +63,9 @@ export class BalanceService {
       },
       select: {
         balance: true,
+        // We need this two other fields for the `where`.
+        address: true,
+        tokenAddress: true,
       },
       order: {
         blockNumber: "DESC",
@@ -84,6 +87,8 @@ export class BalanceService {
     const balancesQuery = this.balanceRepository.createQueryBuilder("balances");
     balancesQuery.select("balances.address");
     balancesQuery.addSelect("balances.balance");
+    balancesQuery.addSelect("balances.tokenAddress");
+    balancesQuery.addSelect("balances.blockNumber");
     balancesQuery.innerJoin(
       `(${latestBalancesQuery.getQuery()})`,
       "latest_balances",