Add simple test with sqlx integration

.github/workflows/ci.yml

@@ -7,6 +7,19 @@ jobs:
     name: CI
     runs-on: ubuntu-latest
 
+    services:
+      postgres:
+        image: postgres:16
+        env:
+          POSTGRES_PASSWORD: postgres
+        options: >-
+          --health-cmd pg_isready
+          --health-interval 10s
+          --health-timeout 5s
+          --health-retries 5
+        ports:
+          - 5432:5432
+
     steps:
       - name: Checkout code
         uses: actions/checkout@v3
@@ -19,12 +32,16 @@ jobs:
 
       - name: cargo build
         run: cargo build
-
-      - name: cargo test
-        run: cargo test
+        env:
+          SQLX_OFFLINE: true
 
       - name: cargo fmt
         run: cargo fmt --all -- --check
 
       - name: cargo clippy
         run: cargo clippy -- -D warnings
+
+      - name: cargo test
+        run: cargo test
+        env:
+          DATABASE_URL: postgres://postgres:postgres@localhost:5432/linkblocks

Cargo.toml

@@ -23,3 +23,6 @@ tracing-subscriber = { version = "0.3.18", features = ["env-filter"] }
 [build-dependencies]
 railwind = "0.1.5"
 walkdir = "2"
+
+[dev-dependencies]
+tower = { version = "0.4.13", features = ["util"] }

Justfile

@@ -19,18 +19,25 @@ start-database:
         exit
     fi
 
-    podman run \
-        --name linkblocks_postgres --detach \
-        --health-cmd pg_isready --health-interval 10s \
-        -e POSTGRES_HOST_AUTH_METHOD=trust -e POSTGRES_DB=linkblocks \
-        -p ${DATABASE_PORT}:5432 docker.io/postgres:16 \
-        postgres
+    if ! podman inspect linkblocks_postgres &> /dev/null; then
+        podman create \
+            --name linkblocks_postgres --detach \
+            --health-cmd pg_isready --health-interval 10s \
+            -e POSTGRES_HOST_AUTH_METHOD=trust -e POSTGRES_DB=linkblocks \
+            -p ${DATABASE_PORT}:5432 docker.io/postgres:16 \
+            postgres
+    fi
+
+    podman start linkblocks_postgres
 
     for i in {1..20}; do 
         pg_isready -h localhost -p $DATABASE_PORT && break
         sleep 1
     done
 
+stop-database:
+    podman stop linkblocks_postgres
+
 test:
     cargo test
 

src/cli.rs

@@ -51,7 +51,8 @@ pub async fn run() -> Result<()> {
         } => {
             let pool = db::pool(&database_url).await?;
             db::migrate(&pool).await?;
-            server::start(listen_address, pool).await
+            let app = server::app(pool);
+            server::start(listen_address, app).await
         }
     }
 }

src/lib.rs

@@ -2,4 +2,4 @@ mod app_error;
 pub mod cli;
 mod db;
 mod routes;
-mod server;
+pub mod server;

src/server.rs

@@ -6,8 +6,8 @@ use axum::{debug_handler, routing::get, Router};
 use listenfd::ListenFd;
 use tower_http::trace::TraceLayer;
 
-pub async fn start(listen: ListenArgs, pool: sqlx::PgPool) -> anyhow::Result<()> {
-    let app = Router::new()
+pub fn app(pool: sqlx::PgPool) -> Router {
+    Router::new()
         .route("/", get(hello))
         .route("/htmx-fragment", get(htmx_fragment))
         .route(
@@ -16,8 +16,10 @@ pub async fn start(listen: ListenArgs, pool: sqlx::PgPool) -> anyhow::Result<()>
         )
         .route("/assets/*path", get(routes::assets::assets))
         .layer(TraceLayer::new_for_http())
-        .with_state(pool);
+        .with_state(pool)
+}
 
+pub async fn start(listen: ListenArgs, app: Router) -> anyhow::Result<()> {
     let listener = if let Some(listen_address) = listen.listen {
         tokio::net::TcpListener::bind(format!("{listen_address}")).await?
     } else {
@@ -31,7 +33,7 @@ pub async fn start(listen: ListenArgs, pool: sqlx::PgPool) -> anyhow::Result<()>
     let listening_on = listener.local_addr()?;
     tracing::info!("Listening on http://{listening_on}");
 
-    axum::serve(listener, app).await?;
+    axum::serve(listener, app);
 
     Ok(())
 }

tests/index.rs

@@ -0,0 +1,17 @@
+use axum::{
+    body::Body,
+    http::{Request, StatusCode},
+};
+use sqlx::{Pool, Postgres};
+use tower::ServiceExt; // for `call`, `oneshot`, and `ready`
+#[sqlx::test]
+async fn index(pool: Pool<Postgres>) {
+    let app = linkblocks::server::app(pool);
+
+    let response = app
+        .oneshot(Request::builder().uri("/").body(Body::empty()).unwrap())
+        .await
+        .unwrap();
+
+    assert_eq!(response.status(), StatusCode::OK);
+}