From 335fbcb046ddf50000423f6758d7ed33b154c62e Mon Sep 17 00:00:00 2001
From: Lucas Soriano del Pino <lucas_soriano@fastmail.com>
Date: Fri, 14 Oct 2022 16:31:30 +1100
Subject: [PATCH] Test to showcase high memory consumption of
 `batch_script_get_history`

When running `batch_script_get_history` on a Linux machine with processing many scripts the memory consumption is enormous and does not go down.
This test showcases this behavior. When triggering the test with 7000 scripts we see a memory consumption of about 4GB.
This problem could be reproduced on multiple Linux machine, but not on MacOS.

Note that you can add a loop to re-trigger the batching to see that the memory does not get freed.
---
 src/raw_client.rs | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/raw_client.rs b/src/raw_client.rs
index 4e649c2..da54f64 100644
--- a/src/raw_client.rs
+++ b/src/raw_client.rs
@@ -1131,6 +1131,40 @@ mod test {
         assert!(resp.confirmed >= 213091301265);
     }
 
+    #[test]
+    fn test_batch_script_get_history() {
+        use std::str::FromStr;
+
+        let client = RawClient::new(get_test_server(), None).unwrap();
+
+        // Using different address does not change the behavior
+        // new address: 38zTM2hA9Xg354HfhLGC5kDJfwv8mSU9Gc
+        // old address: 1FeexV6bAHb8ybZjqQMjJrcCrHGW9sb6uF
+        let script_1 = bitcoin::Address::from_str("1FeexV6bAHb8ybZjqQMjJrcCrHGW9sb6uF")
+            .unwrap()
+            .script_pubkey();
+
+        let mut list = Vec::new();
+
+        for _ in 1..7000 {
+            list.push(&script_1);
+        }
+
+        dbg!("batch_script_get_history started");
+        let _histories = client.batch_script_get_history(list.clone()).unwrap();
+        dbg!("batch_script_get_history finished");
+
+        let awaiting_map = client.waiting_map.lock().unwrap();
+        dbg!(awaiting_map.len());
+        // If we run the batching in a loop (i.e. a periodic sync) then this capacity never goes down.
+        // A lot of memory is getting allocated here
+        dbg!(awaiting_map.capacity());
+
+       let read_buffer = client.buf_reader.lock().unwrap();
+        dbg!(read_buffer.capacity());
+
+    }
+
     #[test]
     fn test_script_get_history() {
         use std::str::FromStr;