[WIP] Daemon & CLI

iota-daemon-lib/tests/command_router.rs

@@ -0,0 +1,52 @@
+use async_trait::async_trait;
+use iota_daemon_lib::{CommandRouter, DaemonRuntime, DaemonServices};
+use iota_ipc::{LocalRequest, ResponseResult};
+use mtp::codec::CommunicationValue;
+use omikron_connector::{OmikronClient, OmikronError};
+use std::sync::{
+    Arc,
+    atomic::{AtomicUsize, Ordering},
+};
+use std::time::Duration;
+
+struct FakeOmikron {
+    reconnects: AtomicUsize,
+}
+#[async_trait]
+impl OmikronClient for FakeOmikron {
+    async fn send_message(&self, _: &CommunicationValue) -> Result<(), OmikronError> {
+        Ok(())
+    }
+    async fn await_response(
+        &self,
+        _: &CommunicationValue,
+        _: Duration,
+    ) -> Result<CommunicationValue, OmikronError> {
+        Err(OmikronError::Disconnected("fake".into()))
+    }
+    async fn reconnect(&self) -> Result<(), OmikronError> {
+        self.reconnects.fetch_add(1, Ordering::SeqCst);
+        Ok(())
+    }
+    async fn is_connected(&self) -> bool {
+        true
+    }
+}
+
+#[tokio::test]
+async fn reconnect_uses_the_injected_client() {
+    let fake = Arc::new(FakeOmikron {
+        reconnects: AtomicUsize::new(0),
+    });
+    let services = Arc::new(DaemonServices {
+        omikron: fake.clone(),
+        users: Default::default(),
+        config: Default::default(),
+    });
+    let router = CommandRouter::new(Arc::new(DaemonRuntime::new()), services);
+    assert!(matches!(
+        router.route(1, LocalRequest::ReconnectOmikron).await.result,
+        ResponseResult::Ok(_)
+    ));
+    assert_eq!(fake.reconnects.load(Ordering::SeqCst), 1);
+}