#![cfg(feature = "sqlite")]
use autumn_web::config::DatabaseConfig;
use autumn_web::db::{RuntimeConnection, create_pool};
use autumn_web::reexports::{axum, diesel, diesel_async};
use axum::Router;
use axum::body::Body;
use axum::extract::State;
use axum::http::{Request, StatusCode};
use axum::routing::get;
use diesel_async::RunQueryDsl as _;
use diesel_async::pooled_connection::deadpool::Pool;
use tower::ServiceExt as _;
type SqlitePool = Pool<RuntimeConnection>;
#[derive(diesel::QueryableByName)]
struct Greeting {
#[diesel(sql_type = diesel::sql_types::Text)]
message: String,
}
async fn greet(State(pool): State<SqlitePool>) -> Result<String, StatusCode> {
let mut conn = pool
.get()
.await
.map_err(|_| StatusCode::SERVICE_UNAVAILABLE)?;
let rows: Vec<Greeting> = diesel::sql_query("SELECT message FROM greetings WHERE id = 1")
.load(&mut *conn)
.await
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
rows.into_iter()
.next()
.map(|g| g.message)
.ok_or(StatusCode::NOT_FOUND)
}
#[tokio::test]
async fn sqlite_pool_boots_and_serves_a_db_backed_request() {
let tmp = tempfile::TempDir::new().expect("temp dir");
let db_path = tmp.path().join("boot.db");
let url = format!("sqlite://{}", db_path.display());
let config = DatabaseConfig {
url: Some(url),
..Default::default()
};
let pool: SqlitePool = create_pool(&config)
.expect("sqlite pool builds via the new build_sqlite_pool path")
.expect("a url is configured");
{
let mut conn = pool.get().await.expect("checkout a sqlite connection");
diesel::sql_query("CREATE TABLE greetings (id INTEGER PRIMARY KEY, message TEXT NOT NULL)")
.execute(&mut *conn)
.await
.expect("create table on sqlite");
diesel::sql_query("INSERT INTO greetings (id, message) VALUES (1, 'hello from sqlite')")
.execute(&mut *conn)
.await
.expect("seed row on sqlite");
}
let app: Router = Router::new().route("/greet", get(greet)).with_state(pool);
let response = app
.oneshot(
Request::builder()
.uri("/greet")
.body(Body::empty())
.expect("build request"),
)
.await
.expect("router serves the request");
assert_eq!(response.status(), StatusCode::OK, "DB-backed route is 200");
let body = axum::body::to_bytes(response.into_body(), usize::MAX)
.await
.expect("read body");
assert_eq!(
&body[..],
b"hello from sqlite",
"response body is the row SELECTed from SQLite"
);
}