@@ -212,10 +212,9 @@ describe('spot balance helpers', () => {
212212 ] ,
213213 } as never ) ;
214214
215- // spotBalance is 0 (no USDC), so totalBalance is unchanged but
216- // availableToTradeBalance is computed from availableBalance.
217- expect ( result . totalBalance ) . toBe ( '50' ) ;
218- expect ( result . availableToTradeBalance ) . toBe ( '0' ) ;
215+ expect ( result . totalBalance ) . toBe ( accountState . totalBalance ) ;
216+ expect ( result . availableBalance ) . toBe ( accountState . availableBalance ) ;
217+ expect ( result . availableToTradeBalance ) . toBe ( accountState . availableBalance ) ;
219218 } ) ;
220219
221220 it ( 'excludes USDH-only spot balance from funded-state totals' , ( ) => {
@@ -234,10 +233,9 @@ describe('spot balance helpers', () => {
234233 ] ,
235234 } as never ) ;
236235
237- // spotBalance is 0 (no USDC), so totalBalance is unchanged but
238- // availableToTradeBalance is computed from availableBalance.
239- expect ( result . totalBalance ) . toBe ( '0' ) ;
240- expect ( result . availableToTradeBalance ) . toBe ( '0' ) ;
236+ expect ( result . totalBalance ) . toBe ( accountState . totalBalance ) ;
237+ expect ( result . availableBalance ) . toBe ( accountState . availableBalance ) ;
238+ expect ( result . availableToTradeBalance ) . toBe ( accountState . availableBalance ) ;
241239 } ) ;
242240
243241 it ( 'adds only the USDC portion when USDC and USDH are both present' , ( ) => {
@@ -260,7 +258,7 @@ describe('spot balance helpers', () => {
260258 expect ( result . totalBalance ) . toBe ( '30' ) ;
261259 } ) ;
262260
263- it ( 'returns the original account state when spot balance is zero' , ( ) => {
261+ it ( 'preserves numeric fields and defaults availableToTradeBalance when spot balance is zero' , ( ) => {
264262 const accountState : AccountState = {
265263 availableBalance : '1' ,
266264 totalBalance : '2' ,
@@ -269,9 +267,14 @@ describe('spot balance helpers', () => {
269267 returnOnEquity : '5' ,
270268 } ;
271269
272- expect (
273- addSpotBalanceToAccountState ( accountState , { balances : [ ] } as never ) ,
274- ) . toEqual ( expect . objectContaining ( accountState ) ) ;
270+ const result = addSpotBalanceToAccountState ( accountState , {
271+ balances : [ ] ,
272+ } as never ) ;
273+
274+ expect ( result . totalBalance ) . toBe ( accountState . totalBalance ) ;
275+ expect ( result . availableBalance ) . toBe ( accountState . availableBalance ) ;
276+ expect ( result . marginUsed ) . toBe ( accountState . marginUsed ) ;
277+ expect ( result . availableToTradeBalance ) . toBe ( accountState . availableBalance ) ;
275278 } ) ;
276279} ) ;
277280
0 commit comments