Update address_balances.rb

coinbase · Sep 9, 2024 · 813487f · 813487f
1 parent 8425c60
commit 813487f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/spec/support/shared_examples/address_balances.rb b/spec/support/shared_examples/address_balances.rb
@@ -2,9 +2,11 @@
 
 shared_examples 'an address that supports balance queries' do |_operation|
   let(:external_addresses_api) { instance_double(Coinbase::Client::ExternalAddressesApi) }
+  let(:balance_history_api) { instance_double(Coinbase::Client::BalanceHistoryApi) }
 
   before do
     allow(Coinbase::Client::ExternalAddressesApi).to receive(:new).and_return(external_addresses_api)
+    allow(Coinbase::Client::BalanceHistoryApi).to receive(:new).and_return(balance_history_api)
   end
 
   describe '#balances' do
@@ -114,7 +116,7 @@
     end
 
     before do
-      allow(external_addresses_api)
+      allow(balance_history_api)
         .to receive(:list_address_historical_balance)
         .with(normalized_network_id, address_id, primary_denomination, { limit: 100, page: nil })
         .and_return(response)