From 576afd574dc26aaaea1643f3bfb025cd48a08d1c Mon Sep 17 00:00:00 2001
From: lunar-mining <ihateface@autistici.org>
Date: Wed, 17 Jan 2024 16:57:46 +0100
Subject: [PATCH] store: create test_remove() unit test

---
 src/net/hosts/store.rs | 45 +++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 44 insertions(+), 1 deletion(-)

diff --git a/src/net/hosts/store.rs b/src/net/hosts/store.rs
index 622eda2f0..d77215209 100644
--- a/src/net/hosts/store.rs
+++ b/src/net/hosts/store.rs
@@ -1063,9 +1063,12 @@ impl Hosts {
 #[cfg(test)]
 mod tests {
     use super::{
-        super::super::{settings::Settings, P2p},
+        super::{
+            super::{settings::Settings, P2p},
+        },
         *,
     };
+    use crate::system::sleep;
     use smol::Executor;
     use std::{sync::Arc, time::UNIX_EPOCH};
 
@@ -1206,6 +1209,46 @@ mod tests {
         });
     }
 
+    #[test]
+    fn test_remove() {
+        smol::block_on(async {
+            let settings = Settings {
+                outbound_connections: 8,
+                allowed_transports: vec!["tcp".to_string()],
+                ..Default::default()
+            };
+            let hosts = Hosts::new(Arc::new(settings.clone()));
+
+            let url = Url::parse("tcp://dark.renaissance:333").unwrap();
+            let last_seen = UNIX_EPOCH.elapsed().unwrap().as_secs();
+
+            hosts.whitelist_store(url.clone(), last_seen).await;
+
+            sleep(1).await;
+
+            let url = Url::parse("tcp://milady:333").unwrap();
+            let last_seen = UNIX_EPOCH.elapsed().unwrap().as_secs();
+
+            hosts.whitelist_store(url.clone(), last_seen).await;
+
+            sleep(1).await;
+
+            let url = Url::parse("tcp://king-ted:333").unwrap();
+            let last_seen = UNIX_EPOCH.elapsed().unwrap().as_secs();
+
+            hosts.whitelist_store(url.clone(), last_seen).await;
+            for (url, last_seen) in hosts.whitelist.read().await.iter() {
+                println!("{}, {}", url, last_seen);
+            }
+
+            let position = hosts.get_whitelist_index_at_addr(url.clone()).await.unwrap();
+            hosts.whitelist_remove(&url, position).await;
+            for (url, last_seen) in hosts.whitelist.read().await.iter() {
+                println!("{}, {}", url, last_seen);
+            }
+        });
+    }
+
     #[test]
     fn test_fetch_address() {
         smol::block_on(async {