What is proper way to reset db after each test?

[Veeksi]

Hey,

I am quite new in rust & axum world and now I am writing integration tests that use postgresql and redis containers. I am trying to figure out what is the optimal way to reset the db after each test? During each test I have setup script that initializes db and redis + migrates the db but I have no clue what could be proper way to reset the db? (like best practice)

I have also unit tests for db itself where I use sqlx::test macro which handles that problem perfectly but that macro cannot be used because I need tokio runtime in integration tests.

Setup script

pub async fn setup_test_app() -> Router {
    dotenvy::from_filename(".env.test").ok();

    let config = Config::init();

    let mut env = minijinja::Environment::new();
    env.set_loader(minijinja::path_loader("templates"));

    let manager = bb8_redis::RedisConnectionManager::new(config.redis_url.to_string())
        .expect("Failed to create redis manager");

    let redis_store = bb8_redis::bb8::Pool::builder()
        .max_size(15)
        .build(manager)
        .await
        .expect("Failed to create redis pool.");

    let store = Store::new(&config.database_url).await;

    let _ = sqlx::migrate!()
        .run(&store.clone().pool)
        .await
        .map_err(|e| {
            panic!("Failed to run migrations: {}", e);
        });

    let app_state = AppState {
        db_store: Store { pool: store.pool },
        config,
        env,
        redis_store,
    };

    let _ = create_test_user_in_db(&app_state.db_store.pool)
        .await
        .map_err(|e| {
            panic!("Failed to create test user in database: {}", e);
        });

    axum::Router::new()
        .route("/api/health", axum::routing::get(routes::health))
        .nest("/api/users", routes::users_routes(app_state.clone()))
        .with_state(app_state.clone())
        .layer(tower_http::trace::TraceLayer::new_for_http())
}

Example test

#[tokio::test]
async fn test_health_route() {
    let app = helpers::setup_test_app().await;
    let server = TestServer::new(app).expect("Failed to create test server");

    let response = get_health_check(&server).await;
    assert_eq!(response.status_code, StatusCode::OK);
}

[Veeksi]

When using sqlx::test macro, I get:

thread 'test_health_route' panicked at /home/veikko/.cargo/registry/src/index.crates.io-1949cf8c6b5b557f/tokio-1.44.0/src/time/interval.rs:138:26:
there is no reactor running, must be called from the context of a Tokio 1.x runtime

Which gets triggered when initializing redis client here:

let redis_store = bb8_redis::bb8::Pool::builder()
        .max_size(15)
        .build(manager)
        .await
        .expect("Failed to create redis pool.");

Otherwise problem would be solved with that macro