schedtxt_api/tests/test.rs

// use std::io::Write;

// use common_multipart_rfc7578::client::multipart::{Body as MultipartBody, Form as MultipartForm};
// use tower::ServiceExt;

use textsender_api::db;

mod db_mgr {
    use std::str::FromStr;

    pub const LIMIT: usize = 6;

    pub async fn get_pool() -> Result<sqlx::PgPool, sqlx::Error> {
        let tm_db_url = textsender_models::envy::environment::get_db_url()
            .await
            .value;
        let tm_options = sqlx::postgres::PgConnectOptions::from_str(&tm_db_url).unwrap();
        sqlx::PgPool::connect_with(tm_options).await
    }

    pub async fn generate_db_name() -> String {
        let db_name =
            get_database_name().await.unwrap() + &"_" + &uuid::Uuid::new_v4().to_string()[..LIMIT];
        db_name
    }

    pub async fn connect_to_db(db_name: &str) -> Result<sqlx::PgPool, sqlx::Error> {
        let db_url = textsender_models::envy::environment::get_db_url()
            .await
            .value;
        let options = sqlx::postgres::PgConnectOptions::from_str(&db_url)?.database(db_name);
        sqlx::PgPool::connect_with(options).await
    }

    pub async fn create_database(
        template_pool: &sqlx::PgPool,
        db_name: &str,
    ) -> Result<(), sqlx::Error> {
        let create_query = format!("CREATE DATABASE {}", db_name);
        match sqlx::query(&create_query).execute(template_pool).await {
            Ok(_) => Ok(()),
            Err(e) => Err(e),
        }
    }

    // Function to drop a database
    pub async fn drop_database(
        template_pool: &sqlx::PgPool,
        db_name: &str,
    ) -> Result<(), sqlx::Error> {
        let drop_query = format!("DROP DATABASE IF EXISTS {} WITH (FORCE)", db_name);
        sqlx::query(&drop_query).execute(template_pool).await?;
        Ok(())
    }

    pub async fn get_database_name() -> Result<String, Box<dyn std::error::Error>> {
        let database_url = textsender_models::envy::environment::get_db_url()
            .await
            .value;
        let parsed_url = url::Url::parse(&database_url)?;

        if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" {
            match parsed_url
                .path_segments()
                .and_then(|segments| segments.last().map(|s| s.to_string()))
            {
                Some(sss) => Ok(sss),
                None => Err("Error parsing".into()),
            }
        } else {
            // Handle other database types if needed
            Err("Error parsing".into())
        }
    }

    /*
    pub async fn migrations(pool: &sqlx::PgPool) {
        // Run migrations using the sqlx::migrate! macro
        // Assumes your test migrations are in a ./test_migrations folder relative to Cargo.toml
        sqlx::migrate!("./migrations")
            .run(pool)
            .await
            .expect("Failed to run migrations");
    }
    */
}

mod init {
    pub async fn app(pool: sqlx::PgPool) -> axum::Router {
        textsender_api::config::init::routes()
            .await
            .layer(axum::Extension(pool))
            .layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024))
            .layer(tower_http::timeout::TimeoutLayer::with_status_code(
                axum::http::StatusCode::OK,
                std::time::Duration::from_secs(300),
            ))
    }
}

mod util {
    pub async fn resp_to_bytes(
        response: axum::response::Response,
    ) -> Result<axum::body::Bytes, axum::Error> {
        axum::body::to_bytes(response.into_body(), std::usize::MAX).await
    }

    pub async fn get_resp_data<Data>(response: axum::response::Response) -> Data
    where
        Data: for<'a> serde::Deserialize<'a>,
    {
        let body = resp_to_bytes(response).await.unwrap();
        serde_json::from_slice(&body).unwrap()
    }

    /*
    pub async fn format_url_with_value(endpoint: &str, value: &uuid::Uuid) -> String {
        let last = endpoint.len() - 5;
        format!("{}/{value}", &endpoint[0..last])
    }
    */
}

pub fn token_fields() -> (String, String, String) {
    (
        String::from("What a twist!"),
        String::from("textsender_test"),
        String::from("textsender"),
    )
}

pub const TEST_USER_ID: uuid::Uuid = uuid::uuid!("cc938368-615a-4694-b2ca-6e122fa31c52");

pub async fn test_token() -> Result<String, josekit::JoseError> {
    let key: String = textsender_models::envy::environment::get_secret_main_key()
        .await
        .value;
    let (message, issuer, audience) = token_fields();

    let token_resource = textsender_models::token::TokenResource {
        message: message,
        issuer: issuer,
        audiences: vec![audience],
        user_id: TEST_USER_ID,
    };

    match textsender_models::token::create_token(&key, &token_resource, time::Duration::hours(1)) {
        Ok((access_token, _some_time)) => Ok(access_token),
        Err(err) => Err(err),
    }
}

pub async fn bearer_auth() -> String {
    let token = match test_token().await {
        Ok(access_token) => access_token,
        Err(err) => {
            assert!(false, "Error: {err:?}");
            String::new()
        }
    };

    format!("Bearer {token}")
}

mod request {
    // use common_multipart_rfc7578::client::multipart::{
    //     Body as MultipartBody, Form as MultipartForm,
    // };
    use tower::ServiceExt;

    pub async fn create_contact(
        app: &axum::Router,
    ) -> Result<axum::response::Response, std::convert::Infallible> {
        let payload = serde_json::json!({
            "phone_number": super::TEST_CONTACT_PHONE_NUMBER,
            "user_id": super::TEST_USER_ID,
        });
        // Create multipart form
        // let mut form = MultipartForm::default();
        // let _ = form.add_file("flac", "tests/I/track01.flac");

        // Create request
        // let content_type = form.content_type();
        // let body = MultipartBody::from(form);
        let req = axum::http::Request::builder()
            .method(axum::http::Method::POST)
            .uri(textsender_api::caller::endpoints::ADD_CONTACT)
            .header(axum::http::header::CONTENT_TYPE, "application/json")
            .header(
                axum::http::header::AUTHORIZATION,
                super::bearer_auth().await,
            )
            .body(axum::body::Body::from(payload.to_string()))
            // .body(axum::body::Body::from_stream(body))
            .unwrap();
        app.clone().oneshot(req).await
    }
}

/// Test contact phone number
pub const TEST_CONTACT_PHONE_NUMBER: &str = "+10123456789";

#[tokio::test]
async fn test_create_contact() {
    let tm_pool = db_mgr::get_pool().await.unwrap();
    let db_name = db_mgr::generate_db_name().await;

    match db_mgr::create_database(&tm_pool, &db_name).await {
        Ok(_) => {
            println!("Success");
        }
        Err(err) => {
            assert!(false, "Error: {:?}", err);
        }
    }

    let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
    db::migrations(&pool).await;

    let app = init::app(pool).await;

    // Send request
    match request::create_contact(&app).await {
        Ok(response) => {
            let resp = util::get_resp_data::<
                textsender_api::caller::contact::response::AddContactResponse,
            >(response)
            .await;
            assert_eq!(false, resp.data.is_empty(), "Should not be empty");
        }
        Err(err) => {
            assert!(false, "Error: {:?}", err);
        }
    };

    let _ = db_mgr::drop_database(&tm_pool, &db_name).await;
}