macltok

#![cfg(feature = "sqlite")]

mod support;

use std::time::SystemTime;

use macltok::{EngineConfig, Event, PasswordLogin, Session, SessionRecord, StoreConfig};

use matrix_sdk::{ruma::api::client::sync::sync_events::v5, test_utils::mocks::MatrixMockServer};

#[tokio::test]

async fn password_session_round_trips_through_an_encrypted_store() {

    let server = MatrixMockServer::new().await;

    server

        .mock_versions()

        .with_simplified_sliding_sync()

        .ok()

        .mount()

        .await;

    server.mock_login().ok().mount().await;

    server

        .mock_sliding_sync()

        .ok(v5::Response::new("1".to_owned()))

        .mount()

        .await;

    let store_path = temporary_store_path();

    std::fs::create_dir_all(&store_path).unwrap();

    let config = EngineConfig::default().with_initial_history_limit(0);

    let session = Session::login_password(

        PasswordLogin::new(server.uri(), "example", "wordpass")

            .with_device_display_name("macltok test"),

        StoreConfig::new(&store_path, "test store passphrase"),

    )

    .await

    .unwrap();

    let encoded = serde_json::to_vec(&session.record()).unwrap();

    let record: SessionRecord = serde_json::from_slice(&encoded).unwrap();

    assert_eq!(

        record.homeserver().trim_end_matches('/'),

        server.uri().trim_end_matches('/')

    );

    let (engine, handle, mut events) = session.build(config.clone()).await.unwrap();

    let running = tokio::spawn(engine.run());

    support::receive_until(&mut events, |event| {

        matches!(event, Event::Authenticated { .. })

    })

    .await;

    support::shut_down(&handle, &mut events, running).await;

    let session = Session::restore(

        record,

        StoreConfig::new(&store_path, "test store passphrase"),

    )

    .await

    .unwrap();

    let (engine, handle, mut events) = session.build(config).await.unwrap();

    let running = tokio::spawn(engine.run());

    support::receive_until(&mut events, |event| {

        matches!(event, Event::Authenticated { .. })

    })

    .await;

    support::shut_down(&handle, &mut events, running).await;

    std::fs::remove_dir_all(store_path).unwrap();

}

fn temporary_store_path() -> std::path::PathBuf {

    let nonce = SystemTime::now()

        .duration_since(SystemTime::UNIX_EPOCH)

        .unwrap()

        .as_nanos();

    std::env::temp_dir().join(format!("macltok-account-{nonce}"))

}