diff --git a/Cargo.lock b/Cargo.lock
index cac932007f..bda12e5620 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -10803,6 +10803,7 @@ dependencies = [
  "cairo-lang-starknet-classes",
  "futures",
  "indexmap 2.6.0",
+ "libp2p",
  "papyrus_config",
  "papyrus_network",
  "papyrus_p2p_sync",
diff --git a/crates/starknet_state_sync/Cargo.toml b/crates/starknet_state_sync/Cargo.toml
index 76518d1771..57d5f17ae4 100644
--- a/crates/starknet_state_sync/Cargo.toml
+++ b/crates/starknet_state_sync/Cargo.toml
@@ -26,6 +26,7 @@ validator.workspace = true
 [dev-dependencies]
 cairo-lang-starknet-classes.workspace = true
 indexmap = { workspace = true, features = ["serde"] }
+libp2p.workspace = true
 papyrus_storage = { workspace = true, features = ["testing"] }
 papyrus_test_utils.workspace = true
 rand_chacha.workspace = true
diff --git a/crates/starknet_state_sync/src/runner/test.rs b/crates/starknet_state_sync/src/runner/test.rs
index 6939227547..55d1d16c8a 100644
--- a/crates/starknet_state_sync/src/runner/test.rs
+++ b/crates/starknet_state_sync/src/runner/test.rs
@@ -1,30 +1,38 @@
-// TODO: Refactor these to suit the change to state sync now using p2p sync.
+use futures::future::{pending, ready};
+use futures::FutureExt;
+use papyrus_network::network_manager::NetworkError;
+use papyrus_p2p_sync::client::P2PSyncClientError;
+use starknet_sequencer_infra::component_definitions::ComponentStarter;
 
-// use futures::channel::mpsc;
-// use futures::future::ready;
-// use futures::FutureExt;
-// use papyrus_storage::test_utils::get_test_storage;
-// use papyrus_sync::StateSyncError as PapyrusStateSyncError;
-// use starknet_sequencer_infra::component_definitions::ComponentStarter;
+use super::StateSyncRunner;
 
-// use super::StateSyncRunner;
+#[test]
+fn run_returns_when_network_future_returns() {
+    let network_future = ready(Ok(())).boxed();
+    let p2p_sync_client_future = pending().boxed();
+    let p2p_sync_server_future = pending().boxed();
+    let mut state_sync_runner =
+        StateSyncRunner { network_future, p2p_sync_client_future, p2p_sync_server_future };
+    state_sync_runner.start().now_or_never().unwrap().unwrap();
+}
 
-// const BUFFER_SIZE: usize = 1000;
+#[test]
+fn run_returns_error_when_network_future_returns_error() {
+    let network_future =
+        ready(Err(NetworkError::DialError(libp2p::swarm::DialError::Aborted))).boxed();
+    let p2p_sync_client_future = pending().boxed();
+    let p2p_sync_server_future = pending().boxed();
+    let mut state_sync_runner =
+        StateSyncRunner { network_future, p2p_sync_client_future, p2p_sync_server_future };
+    state_sync_runner.start().now_or_never().unwrap().unwrap_err();
+}
 
-// #[test]
-// fn run_returns_when_sync_future_returns() {
-//     let (_request_sender, request_receiver) = mpsc::channel(BUFFER_SIZE);
-//     let (storage_reader, _storage_writer) = get_test_storage().0;
-//     let sync_future = ready(Ok(())).boxed();
-//     let mut state_sync_runner = StateSyncRunner { request_receiver, storage_reader, sync_future
-// };     state_sync_runner.start().now_or_never().unwrap().unwrap();
-// }
-
-// #[test]
-// fn run_returns_error_when_sync_future_returns_error() {
-//     let (_request_sender, request_receiver) = mpsc::channel(BUFFER_SIZE);
-//     let (storage_reader, _storage_writer) = get_test_storage().0;
-//     let sync_future = ready(Err(PapyrusStateSyncError::NoProgress)).boxed();
-//     let mut state_sync_runner = StateSyncRunner { request_receiver, storage_reader, sync_future
-// };     state_sync_runner.start().now_or_never().unwrap().unwrap_err();
-// }
+#[test]
+fn run_returns_error_when_sync_client_future_returns_error() {
+    let network_future = pending().boxed();
+    let p2p_sync_client_future = ready(Err(P2PSyncClientError::TooManyResponses)).boxed();
+    let p2p_sync_server_future = pending().boxed();
+    let mut state_sync_runner =
+        StateSyncRunner { network_future, p2p_sync_client_future, p2p_sync_server_future };
+    state_sync_runner.start().now_or_never().unwrap().unwrap_err();
+}