From d49f4ea60d251ecb6ce2db911462330e8c287216 Mon Sep 17 00:00:00 2001 From: Thomas Eizinger Date: Tue, 6 Jul 2021 15:53:14 +1000 Subject: [PATCH] Test saving and loading of addresses in isolation --- swap/src/database.rs | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/swap/src/database.rs b/swap/src/database.rs index 77ae6cb4..34225c2f 100644 --- a/swap/src/database.rs +++ b/swap/src/database.rs @@ -405,4 +405,24 @@ mod tests { Ok(()) } + + #[tokio::test] + async fn save_and_load_addresses() -> Result<()> { + let db_dir = tempfile::tempdir()?; + let peer_id = PeerId::random(); + let home1 = "/ip4/127.0.0.1/tcp/1".parse::()?; + let home2 = "/ip4/127.0.0.1/tcp/2".parse::()?; + + { + let db = Database::open(db_dir.path())?; + db.insert_address(peer_id, home1.clone()).await?; + db.insert_address(peer_id, home2.clone()).await?; + } + + let addresses = Database::open(db_dir.path())?.get_addresses(peer_id)?; + + assert_eq!(addresses, vec![home1, home2]); + + Ok(()) + } }