refactor: madsim

Signed-off-by: bsbds <[email protected]>
xline-kv · Apr 19, 2024 · 8d89b0e · 8d89b0e
1 parent afef71a
commit 8d89b0e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/crates/curp/src/server/mod.rs b/crates/curp/src/server/mod.rs
@@ -292,6 +292,8 @@ impl<C: Command, RC: RoleChange> Rpc<C, RC> {
         storage: Arc<DB<C>>,
         task_manager: Arc<TaskManager>,
         client_tls_config: Option<ClientTlsConfig>,
+        sps: Vec<SpObject<C>>,
+        ucps: Vec<UcpObject<C>>,
     ) -> Result<(), crate::error::ServerError>
     where
         CE: CommandExecutor<C>,
@@ -311,6 +313,8 @@ impl<C: Command, RC: RoleChange> Rpc<C, RC> {
             storage,
             task_manager,
             client_tls_config,
+            sps,
+            ucps,
         )
         .await;
 

diff --git a/crates/simulation/src/curp_group.rs b/crates/simulation/src/curp_group.rs
@@ -12,7 +12,10 @@ use curp::{
         ConfChange, FetchClusterRequest, FetchClusterResponse, Member, ProposeConfChangeRequest,
         ProposeConfChangeResponse, ReadState,
     },
-    server::{Rpc, StorageApi, DB},
+    server::{
+        conflict::test_pools::{TestSpecPool, TestUncomPool},
+        Rpc, StorageApi, DB,
+    },
     LogIndex,
 };
 use curp_test_utils::{
@@ -132,6 +135,8 @@ impl CurpGroup {
                             curp_storage,
                             task_manager,
                             None,
+                            vec![Box::<TestSpecPool>::default()],
+                            vec![Box::<TestUncomPool>::default()],
                         )
                     })
                     .build();