From 2a81156520bd10c7926ff12e330fb9f9858b3cea Mon Sep 17 00:00:00 2001
From: Jack Grigg <jack@electriccoin.co>
Date: Sun, 10 Mar 2024 19:53:31 +0000
Subject: [PATCH] zcash_client_sqlite: Fix balance APIs in `TestState` to
 include Orchard

---
 zcash_client_sqlite/src/testing.rs | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/zcash_client_sqlite/src/testing.rs b/zcash_client_sqlite/src/testing.rs
index 7d63401a8..1fd22e1c5 100644
--- a/zcash_client_sqlite/src/testing.rs
+++ b/zcash_client_sqlite/src/testing.rs
@@ -806,7 +806,7 @@ impl<Cache> TestState<Cache> {
         min_confirmations: u32,
     ) -> NonNegativeAmount {
         self.with_account_balance(account, min_confirmations, |balance| {
-            balance.sapling_balance().spendable_value()
+            balance.spendable_value()
         })
     }
 
@@ -816,8 +816,7 @@ impl<Cache> TestState<Cache> {
         min_confirmations: u32,
     ) -> NonNegativeAmount {
         self.with_account_balance(account, min_confirmations, |balance| {
-            balance.sapling_balance().value_pending_spendability()
-                + balance.sapling_balance().change_pending_confirmation()
+            balance.value_pending_spendability() + balance.change_pending_confirmation()
         })
         .unwrap()
     }
@@ -829,7 +828,7 @@ impl<Cache> TestState<Cache> {
         min_confirmations: u32,
     ) -> NonNegativeAmount {
         self.with_account_balance(account, min_confirmations, |balance| {
-            balance.sapling_balance().change_pending_confirmation()
+            balance.change_pending_confirmation()
         })
     }