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);
}