lavina/crates/projection-xmpp/src/testkit.rs

use prometheus::Registry as MetricsRegistry;

use crate::{Authenticated, XmppConnection};
use lavina_core::clustering::{ClusterConfig, ClusterMetadata};
use lavina_core::player::PlayerConnection;
use lavina_core::repo::{Storage, StorageConfig};
use lavina_core::LavinaCore;
use proto_xmpp::bind::{BindRequest, BindResponse, Jid, Name, Resource, Server};

pub(crate) struct TestServer {
    pub core: LavinaCore,
}

impl TestServer {
    pub async fn start() -> anyhow::Result<TestServer> {
        let _ = tracing_subscriber::fmt::try_init();
        let mut metrics = MetricsRegistry::new();
        let storage = Storage::open(StorageConfig {
            db_path: ":memory:".into(),
        })
        .await?;
        let cluster_config = ClusterConfig {
            metadata: ClusterMetadata {
                node_id: 0,
                main_owner: 0,
                rooms: Default::default(),
            },
            addresses: vec![],
        };
        let core = LavinaCore::new(&mut metrics, cluster_config, storage).await?;
        Ok(TestServer { core })
    }

    pub async fn reboot(self) -> anyhow::Result<TestServer> {
        let storage = self.core.shutdown().await;

        let mut metrics = MetricsRegistry::new();
        let cluster_config = ClusterConfig {
            metadata: ClusterMetadata {
                node_id: 0,
                main_owner: 0,
                rooms: Default::default(),
            },
            addresses: vec![],
        };
        let core = LavinaCore::new(&mut metrics, cluster_config, storage).await?;

        Ok(TestServer { core })
    }

    pub async fn shutdown(self) -> anyhow::Result<()> {
        let storage = self.core.shutdown().await;
        storage.close().await;
        Ok(())
    }
}

pub async fn expect_user_authenticated<'a>(
    server: &'a TestServer,
    user: &'a Authenticated,
    conn: &'a mut PlayerConnection,
) -> anyhow::Result<XmppConnection<'a>> {
    let conn = XmppConnection {
        user: &user,
        user_handle: conn,
        core: &server.core,
        hostname: "localhost".into(),
        hostname_rooms: "rooms.localhost".into(),
    };
    let result = conn.bind(&BindRequest(Resource("whatever".into()))).await;
    let expected = BindResponse(Jid {
        name: Some(Name("tester".into())),
        server: Server("localhost".into()),
        resource: Some(Resource("tester".into())),
    });
    assert_eq!(expected, result);
    Ok(conn)
}