#![cfg(all(test, feature = "test_e2e"))]
use azure_sdk_cosmos::prelude::*;

mod setup;

#[tokio::test]
async fn create_and_delete_database() {
    const DATABASE_NAME: &str = "cosmos-test-db-create-and-delete-database";

    let client = setup::initialize().unwrap();

    // list existing databases and remember their number
    let databases = client.list_databases().execute().await.unwrap();
    let database_count_before = databases.databases.len();

    // create a new database and check if the number of DBs increased
    let database = client
        .create_database()
        .with_database_name(&DATABASE_NAME)
        .execute()
        .await
        .unwrap();
    let databases = client.list_databases().execute().await.unwrap();
    assert!(databases.databases.len() == database_count_before + 1);

    // get the previously created database
    let database_after_get = client
        .with_database_client(DATABASE_NAME)
        .get_database()
        .execute()
        .await
        .unwrap();
    assert!(database.database.rid == database_after_get.database.rid);

    // delete the database
    client
        .with_database_client(DATABASE_NAME)
        .delete_database()
        .execute()
        .await
        .unwrap();
    let databases = client.list_databases().execute().await.unwrap();
    assert!(databases.databases.len() == database_count_before);
}