From 7ceb98616cf3054b277b7714e4ca6b9a48ee0b90 Mon Sep 17 00:00:00 2001
From: Josh Wilson <joshuef@gmail.com>
Date: Wed, 18 Oct 2023 10:23:11 +0200
Subject: [PATCH] test: update verify_data_location to use the test pay for
 chunks only

---
 sn_node/tests/verify_data_location.rs | 38 ++++-----------------------
 1 file changed, 5 insertions(+), 33 deletions(-)

diff --git a/sn_node/tests/verify_data_location.rs b/sn_node/tests/verify_data_location.rs
index 39371cd8c2..17d849c96d 100644
--- a/sn_node/tests/verify_data_location.rs
+++ b/sn_node/tests/verify_data_location.rs
@@ -21,10 +21,9 @@ use libp2p::{
     PeerId,
 };
 use rand::{rngs::OsRng, Rng};
-use sn_client::{Client, Files, WalletClient};
+use sn_client::{Client, Files};
 use sn_networking::{sort_peers_by_key, CLOSE_GROUP_SIZE};
-use sn_protocol::{storage::ChunkAddress, NetworkAddress, PrettyPrintRecordKey};
-use sn_transfers::LocalWallet;
+use sn_protocol::{NetworkAddress, PrettyPrintRecordKey};
 use std::{
     collections::{BTreeSet, HashMap, HashSet},
     fs::File,
@@ -86,16 +85,10 @@ async fn verify_data_location() -> Result<()> {
     println!("Creating a client and paying wallet...");
     let paying_wallet_dir = TempDir::new()?;
 
-    let (client, paying_wallet) =
+    let (client, _paying_wallet) =
         get_client_and_wallet(paying_wallet_dir.path(), PAYING_WALLET_INITIAL_BALANCE).await?;
 
-    store_chunks(
-        client,
-        paying_wallet,
-        chunk_count,
-        paying_wallet_dir.to_path_buf(),
-    )
-    .await?;
+    store_chunks(client, chunk_count, paying_wallet_dir.to_path_buf()).await?;
 
     // set of all the node indexes that stores a record key
     let record_holders = get_records_and_holders().await?;
@@ -302,15 +295,9 @@ async fn get_all_peer_ids() -> Result<Vec<PeerId>> {
 }
 
 // Generate random Chunks and store them to the Network
-async fn store_chunks(
-    client: Client,
-    paying_wallet: LocalWallet,
-    chunk_count: usize,
-    wallet_dir: PathBuf,
-) -> Result<()> {
+async fn store_chunks(client: Client, chunk_count: usize, wallet_dir: PathBuf) -> Result<()> {
     let start = Instant::now();
     let mut rng = OsRng;
-    let mut wallet_client = WalletClient::new(client.clone(), paying_wallet);
     let file_api = Files::new(client, wallet_dir);
 
     let mut uploaded_chunks_count = 0;
@@ -338,21 +325,6 @@ async fn store_chunks(
             random_bytes.len()
         );
 
-        let _cost = wallet_client
-            .pay_for_storage(
-                chunks
-                    .iter()
-                    .map(|(name, _)| NetworkAddress::ChunkAddress(ChunkAddress::new(*name))),
-            )
-            .await
-            .expect("Failed to pay for storage for new file at {file_addr:?}");
-
-        println!(
-            "Storing ({}) Chunk/s of file ({} bytes) at {file_addr:?}",
-            chunks.len(),
-            random_bytes.len()
-        );
-
         let key = PrettyPrintRecordKey::from(RecordKey::new(&file_addr));
         file_api
             .pay_and_upload_bytes_test(file_addr, chunks)