matrix-rust-sdk/testing/matrix-sdk-integration-testing/src/tests.rs

use std::{collections::HashMap, option_env};

use anyhow::Result;
use assign::assign;
use lazy_static::lazy_static;
use matrix_sdk::{
    ruma::api::client::{account::register::v3::Request as RegistrationRequest, uiaa},
    store::make_store_config,
    Client,
};
use tempfile::{tempdir, TempDir};
use tokio::sync::Mutex;

lazy_static! {
    static ref USERS: Mutex<HashMap<String, (Client, TempDir)>> = Mutex::new(HashMap::new());
}

/// read the test configuration from the environment
pub fn test_server_conf() -> (String, String) {
    (
        option_env!("HOMSERVER_URL").unwrap_or("http://localhost:8228").to_owned(),
        option_env!("HOMSERVER_DOMAIN").unwrap_or("matrix-sdk.rs").to_owned(),
    )
}

pub async fn get_client_for_user(username: String) -> Result<Client> {
    let mut users = USERS.lock().await;
    if let Some((client, _)) = users.get(&username) {
        return Ok(client.clone());
    }

    let (homeserver_url, _domain_name) = test_server_conf();

    let tmp_dir = tempdir()?;

    let client = Client::builder()
        .user_agent("matrix-sdk-integation-tests")
        .store_config(make_store_config(tmp_dir.path(), None)?)
        .homeserver_url(homeserver_url)
        .build()
        .await?;
    // safe to assume we have not registered this user yet, but ignore if we did

    if let Err(resp) = client.register(RegistrationRequest::new()).await {
        // FIXME: do actually check the registration types...
        if let Some(_response) = resp.uiaa_response() {
            let request = assign!(RegistrationRequest::new(), {
                username: Some(username.as_ref()),
                password: Some(username.as_ref()),

                auth: Some(uiaa::AuthData::Dummy(uiaa::Dummy::new())),
            });
            // we don't care if this failed, then we just try to login anyways
            let _ = client.register(request).await;
        }
    }
    client.login_username(&username, &username).send().await?;
    users.insert(username, (client.clone(), tmp_dir)); // keeping temp dir around so it doesn't get destroyed yet

    Ok(client)
}

mod invitations;