Add test

Results on current `v0.41-dev`

---- test_load_headers_from_storage::genesis_1_block stdout ----

thread 'test_load_headers_from_storage::genesis_1_block' (8727544) panicked at dash-spv/tests/header_sync_test.rs:380:5:
assertion `left == right` failed: Loaded count mismatch
  left: 0
 right: 1

---- test_load_headers_from_storage::checkpoint_1_block stdout ----

thread 'test_load_headers_from_storage::checkpoint_1_block' (8727541) panicked at dash-spv/tests/header_sync_test.rs:381:5:
assertion `left == right` failed: Chain state count mismatch
  left: 1
 right: 2
note: run with `RUST_BACKTRACE=1` environment variable to display a backtrace

---- test_load_headers_from_storage::checkpoint_60000_blocks stdout ----

thread 'test_load_headers_from_storage::checkpoint_60000_blocks' (8727542) panicked at dash-spv/tests/header_sync_test.rs:381:5:
assertion `left == right` failed: Chain state count mismatch
  left: 60000
 right: 120000

---- test_load_headers_from_storage::genesis_60000_blocks stdout ----

thread 'test_load_headers_from_storage::genesis_60000_blocks' (8727545) panicked at dash-spv/tests/header_sync_test.rs:380:5:
assertion `left == right` failed: Loaded count mismatch
  left: 59999
 right: 60000

Author: xdustinface

dash-spv/Cargo.toml

@@ -60,6 +60,7 @@ tempfile = "3.0"
 tokio-test = "0.4"
 env_logger = "0.10"
 hex = "0.4"
+test-case = "3.3"
 
 [[bin]]
 name = "dash-spv"

dash-spv/tests/header_sync_test.rs

@@ -5,15 +5,18 @@ use std::time::Duration;
 use dash_spv::{
     client::{ClientConfig, DashSpvClient},
     network::PeerNetworkManager,
-    storage::{MemoryStorageManager, StorageManager},
+    storage::{DiskStorageManager, MemoryStorageManager, StorageManager},
+    sync::{HeaderSyncManager, ReorgConfig},
     types::{ChainState, ValidationMode},
 };
 use dashcore::{block::Header as BlockHeader, block::Version, Network};
 use dashcore_hashes::Hash;
 use key_wallet::wallet::managed_wallet_info::ManagedWalletInfo;
 use key_wallet_manager::wallet_manager::WalletManager;
 use log::{debug, info};
+use test_case::test_case;
 use std::sync::Arc;
+use tempfile::TempDir;
 use tokio::sync::RwLock;
 
 #[tokio::test]
@@ -332,3 +335,48 @@ async fn test_header_storage_consistency() {
 
     info!("Header storage consistency test completed");
 }
+
+#[test_case(0, 0 ; "genesis_0_blocks")]
+#[test_case(0, 1 ; "genesis_1_block")]
+#[test_case(0, 60000 ; "genesis_60000_blocks")]
+#[test_case(0, 0 ; "checkpoint_0_blocks")]
+#[test_case(170000, 1 ; "checkpoint_1_block")]
+#[test_case(12345, 60000 ; "checkpoint_60000_blocks")]
+#[tokio::test]
+async fn test_load_headers_from_storage(
+    sync_base_height: u32,
+    header_count: usize,
+) {
+    // Setup: Create storage with 100 headers
+    let temp_dir = TempDir::new().expect("Failed to create temp dir");
+    let mut storage = DiskStorageManager::new(temp_dir.path().to_path_buf())
+        .await
+        .expect("Failed to create storage");
+
+    let test_headers = create_test_header_chain(header_count);
+
+    // Store chain state
+    let mut chain_state = ChainState::new_for_network(Network::Dash);
+    chain_state.sync_base_height = sync_base_height;
+    chain_state.headers = test_headers.clone();
+    storage.store_chain_state(&chain_state).await.expect("Failed to store chain state");
+
+    // Create HeaderSyncManager and load headers
+    let config = ClientConfig::new(Network::Dash);
+    let chain_state = Arc::new(RwLock::new(ChainState::new_for_network(Network::Dash)));
+    let mut header_sync = HeaderSyncManager::<DiskStorageManager, PeerNetworkManager>::new(
+        &config,
+        ReorgConfig::default(),
+        chain_state.clone(),
+    )
+    .expect("Failed to create HeaderSyncManager");
+
+    // Load headers from storage
+    let loaded_count =
+        header_sync.load_headers_from_storage(&storage).await.expect("Failed to load headers");
+
+    let cs = chain_state.read().await;
+
+    assert_eq!(loaded_count as usize, header_count, "Loaded count mismatch");
+    assert_eq!(header_count, cs.headers.len(), "Chain state count mismatch");
+}