add a test

Cuprate · Jun 13, 2024 · dd9e63d · dd9e63d
1 parent cefd8a7
commit dd9e63d
Show file tree

Hide file tree

Showing 5 changed files with 368 additions and 54 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/p2p/cuprate-p2p/Cargo.toml b/p2p/cuprate-p2p/Cargo.toml
@@ -27,13 +27,12 @@ dashmap = { workspace = true }
 
 thiserror = { workspace = true }
 bytes = { workspace = true, features = ["std"] }
-indexmap = { workspace = true, features = ["std"] }
 rand = { workspace = true, features = ["std", "std_rng"] }
 rand_distr = { workspace = true, features = ["std"] }
 hex = { workspace = true, features = ["std"] }
 tracing = { workspace = true, features = ["std", "attributes"] }
 
-tracing-subscriber = "0.3.18"
-
 [dev-dependencies]
 cuprate-test-utils = { path = "../../test-utils" }
+indexmap = { workspace = true }
+proptest = { workspace = true }
diff --git a/p2p/cuprate-p2p/src/block_downloader.rs b/p2p/cuprate-p2p/src/block_downloader.rs
@@ -4,6 +4,7 @@
 //! and downloads it.
 //!
 //! The block downloader is started by [`download_blocks`].
+use futures::FutureExt;
 use std::{
     cmp::{max, min, Ordering, Reverse},
     collections::{BTreeMap, BinaryHeap, HashSet},
@@ -46,6 +47,9 @@ use crate::{
 mod chain_tracker;
 use chain_tracker::{BlocksToRetrieve, ChainEntry, ChainTracker};
 
+#[cfg(test)]
+mod tests;
+
 /// A downloaded batch of blocks.
 #[derive(Debug)]
 pub struct BlockBatch {
@@ -154,7 +158,12 @@ where
         config,
     );
 
-    tokio::spawn(block_downloader.run().instrument(Span::current()));
+    tokio::spawn(
+        block_downloader
+            .run()
+            .instrument(Span::current())
+            .map(|res| panic!("{res:?}")),
+    );
 
     buffer_stream
 }