Added test
This commit is contained in:
@@ -13,8 +13,11 @@ tower = { version = "0.5.2" }
|
|||||||
hyper = { version = "1.6.0" }
|
hyper = { version = "1.6.0" }
|
||||||
sqlx = { version = "0.8.3", features = ["postgres", "runtime-tokio-native-tls"] }
|
sqlx = { version = "0.8.3", features = ["postgres", "runtime-tokio-native-tls"] }
|
||||||
dotenv = { version = "0.15" }
|
dotenv = { version = "0.15" }
|
||||||
|
http-body-util = "0.1.3"
|
||||||
icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.2.0" }
|
icarus_models = { git = "ssh://git@git.kundeng.us/phoenix/icarus_models.git", tag = "v0.2.0" }
|
||||||
|
|
||||||
[dev-dependencies]
|
[dev-dependencies]
|
||||||
testcontainers = { version = "0.23.3" }
|
testcontainers = { version = "0.23.3" }
|
||||||
testcontainers-modules = { version = "0.11.6", features = ["blocking", "postgres"] }
|
testcontainers-modules = { version = "0.11.6", features = ["blocking", "postgres"] }
|
||||||
|
reqwest = { version = "0.12", features = ["json"] } # For making HTTP requests in tests
|
||||||
|
once_cell = "1.19" # Useful for lazy initialization in tests/app setup
|
85
src/main.rs
85
src/main.rs
@@ -6,28 +6,87 @@ use axum::{
|
|||||||
|
|
||||||
use icarus_auth::callers;
|
use icarus_auth::callers;
|
||||||
use icarus_auth::config;
|
use icarus_auth::config;
|
||||||
|
// use sqlx::Postgres;
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
// initialize tracing
|
// initialize tracing
|
||||||
tracing_subscriber::fmt::init();
|
tracing_subscriber::fmt::init();
|
||||||
|
|
||||||
let pool = icarus_auth::db_pool::create_pool()
|
let app = app().await;
|
||||||
.await
|
|
||||||
.expect("Failed to create pool");
|
|
||||||
|
|
||||||
// build our application with a route
|
|
||||||
let app = Router::new()
|
|
||||||
.route(callers::endpoints::DBTEST, get(callers::common::db_ping))
|
|
||||||
.route(callers::endpoints::ROOT, get(callers::common::root))
|
|
||||||
.route(
|
|
||||||
callers::endpoints::REGISTER,
|
|
||||||
post(callers::register::register_user),
|
|
||||||
)
|
|
||||||
.layer(axum::Extension(pool));
|
|
||||||
|
|
||||||
// run our app with hyper, listening globally on port 3000
|
// run our app with hyper, listening globally on port 3000
|
||||||
let url = config::get_full();
|
let url = config::get_full();
|
||||||
let listener = tokio::net::TcpListener::bind(url).await.unwrap();
|
let listener = tokio::net::TcpListener::bind(url).await.unwrap();
|
||||||
axum::serve(listener, app).await.unwrap();
|
axum::serve(listener, app).await.unwrap();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn app() -> Router {
|
||||||
|
let pool = icarus_auth::db_pool::create_pool()
|
||||||
|
.await
|
||||||
|
.expect("Failed to create pool");
|
||||||
|
|
||||||
|
// build our application with a route
|
||||||
|
Router::new()
|
||||||
|
.route(callers::endpoints::DBTEST, get(callers::common::db_ping))
|
||||||
|
.route(callers::endpoints::ROOT, get(callers::common::root))
|
||||||
|
.route(
|
||||||
|
callers::endpoints::REGISTER,
|
||||||
|
post(callers::register::register_user),
|
||||||
|
)
|
||||||
|
.layer(axum::Extension(pool))
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use axum::{
|
||||||
|
body::Body,
|
||||||
|
extract::connect_info::MockConnectInfo,
|
||||||
|
http::{self, Request, StatusCode},
|
||||||
|
};
|
||||||
|
use http_body_util::BodyExt;
|
||||||
|
// use http_body_util::BodyExt; // for `collect`
|
||||||
|
use serde_json::{Value, json};
|
||||||
|
use tokio::net::TcpListener;
|
||||||
|
use tower::{Service, ServiceExt}; // for `call`, `oneshot`, and `ready`
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn hello_world() {
|
||||||
|
let app = app().await;
|
||||||
|
|
||||||
|
// `Router` implements `tower::Service<Request<Body>>` so we can
|
||||||
|
// call it like any tower service, no need to run an HTTP server.
|
||||||
|
let response = app
|
||||||
|
.oneshot(
|
||||||
|
Request::builder()
|
||||||
|
.uri(callers::endpoints::ROOT)
|
||||||
|
.body(Body::empty())
|
||||||
|
.unwrap(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
assert_eq!(response.status(), StatusCode::OK);
|
||||||
|
|
||||||
|
/*
|
||||||
|
match response.into_body().collect().await {
|
||||||
|
Ok(o) => {
|
||||||
|
let parsed: String = match String::from_utf8(o.to_bytes()) {
|
||||||
|
Ok(s) => s,
|
||||||
|
Err(err) => {
|
||||||
|
String::new()
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
Err(err) => {
|
||||||
|
assert!(false,
|
||||||
|
"Error: {:?}", err.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
let body = response.into_body().collect().await.unwrap().to_bytes();
|
||||||
|
assert_eq!(&body[..], b"Hello, World!");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Reference in New Issue
Block a user