[Fix] Stability

iota-daemon-lib/tests/command_router.rs

@@ -29,8 +29,12 @@ impl OmikronClient for FakeOmikron {
         self.reconnects.fetch_add(1, Ordering::SeqCst);
         Ok(())
     }
+    async fn rotate_identity(&self) -> Result<(), OmikronError> {
+        self.reconnects.fetch_add(1, Ordering::SeqCst);
+        Ok(())
+    }
     async fn is_connected(&self) -> bool {
-        true
+        false
     }
 }
 
@@ -43,6 +47,7 @@ async fn reconnect_uses_the_injected_client() {
         omikron: fake.clone(),
         users: Default::default(),
         config: Default::default(),
+        active: true,
     });
     let router = CommandRouter::new(
         Arc::new(DaemonRuntime::new()),
@@ -55,3 +60,29 @@ async fn reconnect_uses_the_injected_client() {
     ));
     assert_eq!(fake.reconnects.load(Ordering::SeqCst), 1);
 }
+
+#[tokio::test]
+async fn identity_rotation_is_available_while_omikron_is_offline() {
+    let fake = Arc::new(FakeOmikron {
+        reconnects: AtomicUsize::new(0),
+    });
+    let services = Arc::new(DaemonServices {
+        omikron: fake.clone(),
+        users: Default::default(),
+        config: Default::default(),
+        active: true,
+    });
+    let router = CommandRouter::new(
+        Arc::new(DaemonRuntime::new()),
+        services,
+        Arc::new(Mutex::new(LogBuffer::new(100))),
+    );
+    assert!(matches!(
+        router
+            .route(1, LocalRequest::RotateIotaIdentity)
+            .await
+            .result,
+        ResponseResult::Ok(_)
+    ));
+    assert_eq!(fake.reconnects.load(Ordering::SeqCst), 1);
+}