343 lines
11 KiB
Rust
343 lines
11 KiB
Rust
use icarus_auth::callers;
|
|
use icarus_auth::config;
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
// initialize tracing
|
|
tracing_subscriber::fmt::init();
|
|
|
|
let app = init::app().await;
|
|
|
|
// run our app with hyper, listening globally on port 3000
|
|
let url = config::get_full();
|
|
let listener = tokio::net::TcpListener::bind(url).await.unwrap();
|
|
axum::serve(listener, app).await.unwrap();
|
|
}
|
|
|
|
mod init {
|
|
use axum::{
|
|
Router,
|
|
routing::{get, post},
|
|
};
|
|
|
|
use crate::callers;
|
|
|
|
pub async fn routes() -> Router {
|
|
// build our application with a route
|
|
Router::new()
|
|
.route(
|
|
callers::endpoints::DBTEST,
|
|
get(callers::common::endpoint::db_ping),
|
|
)
|
|
.route(
|
|
callers::endpoints::ROOT,
|
|
get(callers::common::endpoint::root),
|
|
)
|
|
.route(
|
|
callers::endpoints::REGISTER,
|
|
post(callers::register::register_user),
|
|
)
|
|
.route(
|
|
callers::endpoints::LOGIN,
|
|
post(callers::login::endpoint::login),
|
|
)
|
|
.route(callers::endpoints::SERVICE_LOGIN, post(callers::login::endpoint::service_login))
|
|
}
|
|
|
|
pub async fn app() -> Router {
|
|
let pool = icarus_auth::db::create_pool()
|
|
.await
|
|
.expect("Failed to create pool");
|
|
|
|
icarus_auth::db::migrations(&pool).await;
|
|
|
|
routes().await.layer(axum::Extension(pool))
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
use axum::{
|
|
body::Body,
|
|
http::{Request, StatusCode},
|
|
};
|
|
use http_body_util::BodyExt;
|
|
use serde_json::json;
|
|
use tower::ServiceExt; // for `call`, `oneshot`, and `ready`
|
|
|
|
mod db_mgr {
|
|
use std::str::FromStr;
|
|
|
|
pub const LIMIT: usize = 6;
|
|
|
|
pub async fn get_pool() -> Result<sqlx::PgPool, sqlx::Error> {
|
|
let tm_db_url = icarus_envy::environment::get_db_url().await;
|
|
let tm_options = sqlx::postgres::PgConnectOptions::from_str(&tm_db_url).unwrap();
|
|
sqlx::PgPool::connect_with(tm_options).await
|
|
}
|
|
|
|
pub async fn generate_db_name() -> String {
|
|
let db_name = get_database_name().await.unwrap()
|
|
+ &"_"
|
|
+ &uuid::Uuid::new_v4().to_string()[..LIMIT];
|
|
db_name
|
|
}
|
|
|
|
pub async fn connect_to_db(db_name: &str) -> Result<sqlx::PgPool, sqlx::Error> {
|
|
let db_url = icarus_envy::environment::get_db_url().await;
|
|
let options = sqlx::postgres::PgConnectOptions::from_str(&db_url)?.database(db_name);
|
|
sqlx::PgPool::connect_with(options).await
|
|
}
|
|
|
|
pub async fn create_database(
|
|
template_pool: &sqlx::PgPool,
|
|
db_name: &str,
|
|
) -> Result<(), sqlx::Error> {
|
|
let create_query = format!("CREATE DATABASE {}", db_name);
|
|
match sqlx::query(&create_query).execute(template_pool).await {
|
|
Ok(_) => Ok(()),
|
|
Err(e) => Err(e),
|
|
}
|
|
}
|
|
|
|
// Function to drop a database
|
|
pub async fn drop_database(
|
|
template_pool: &sqlx::PgPool,
|
|
db_name: &str,
|
|
) -> Result<(), sqlx::Error> {
|
|
let drop_query = format!("DROP DATABASE IF EXISTS {} WITH (FORCE)", db_name);
|
|
sqlx::query(&drop_query).execute(template_pool).await?;
|
|
Ok(())
|
|
}
|
|
|
|
pub async fn get_database_name() -> Result<String, Box<dyn std::error::Error>> {
|
|
let database_url = icarus_envy::environment::get_db_url().await;
|
|
|
|
let parsed_url = url::Url::parse(&database_url)?;
|
|
if parsed_url.scheme() == "postgres" || parsed_url.scheme() == "postgresql" {
|
|
match parsed_url
|
|
.path_segments()
|
|
.and_then(|segments| segments.last().map(|s| s.to_string()))
|
|
{
|
|
Some(sss) => Ok(sss),
|
|
None => Err("Error parsing".into()),
|
|
}
|
|
} else {
|
|
// Handle other database types if needed
|
|
Err("Error parsing".into())
|
|
}
|
|
}
|
|
}
|
|
|
|
fn get_test_register_request() -> icarus_auth::callers::register::request::Request {
|
|
icarus_auth::callers::register::request::Request {
|
|
username: String::from("somethingsss"),
|
|
password: String::from("Raindown!"),
|
|
email: String::from("dev@null.com"),
|
|
phone: String::from("1234567890"),
|
|
firstname: String::from("Bob"),
|
|
lastname: String::from("Smith"),
|
|
}
|
|
}
|
|
|
|
fn get_test_register_payload(
|
|
usr: &icarus_auth::callers::register::request::Request,
|
|
) -> serde_json::Value {
|
|
json!({
|
|
"username": &usr.username,
|
|
"password": &usr.password,
|
|
"email": &usr.email,
|
|
"phone": &usr.phone,
|
|
"firstname": &usr.firstname,
|
|
"lastname": &usr.lastname,
|
|
})
|
|
}
|
|
|
|
pub mod requests {
|
|
use tower::ServiceExt; // for `call`, `oneshot`, and `ready`
|
|
|
|
pub async fn register(app: &axum::Router, usr: &icarus_auth::callers::register::request::Request) -> Result<axum::response::Response, std::convert::Infallible> {
|
|
// let usr = super::get_test_register_request();
|
|
let payload = super::get_test_register_payload(&usr);
|
|
let req = axum::http::Request::builder()
|
|
.method(axum::http::Method::POST)
|
|
.uri(crate::callers::endpoints::REGISTER)
|
|
.header(axum::http::header::CONTENT_TYPE, "application/json")
|
|
.body(axum::body::Body::from(payload.to_string()))
|
|
.unwrap();
|
|
|
|
app.clone().oneshot(req).await
|
|
}
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_hello_world() {
|
|
let app = init::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);
|
|
|
|
let body = response.into_body().collect().await.unwrap().to_bytes();
|
|
assert_eq!(&body[..], b"Hello, World!");
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_register_user() {
|
|
let tm_pool = db_mgr::get_pool().await.unwrap();
|
|
|
|
let db_name = db_mgr::generate_db_name().await;
|
|
|
|
match db_mgr::create_database(&tm_pool, &db_name).await {
|
|
Ok(_) => {
|
|
println!("Success");
|
|
}
|
|
Err(e) => {
|
|
assert!(false, "Error: {:?}", e.to_string());
|
|
}
|
|
}
|
|
|
|
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
|
|
|
|
icarus_auth::db::migrations(&pool).await;
|
|
|
|
let app = init::routes().await.layer(axum::Extension(pool));
|
|
|
|
let usr = get_test_register_request();
|
|
|
|
let response = requests::register(&app, &usr).await;
|
|
|
|
match response {
|
|
Ok(resp) => {
|
|
assert_eq!(
|
|
resp.status(),
|
|
StatusCode::CREATED,
|
|
"Message: {:?} {:?}",
|
|
resp,
|
|
usr.username
|
|
);
|
|
let body = axum::body::to_bytes(resp.into_body(), usize::MAX)
|
|
.await
|
|
.unwrap();
|
|
let parsed_body: callers::register::response::Response =
|
|
serde_json::from_slice(&body).unwrap();
|
|
let returned_usr = &parsed_body.data[0];
|
|
|
|
assert_eq!(false, returned_usr.id.is_nil(), "Id is not populated");
|
|
|
|
assert_eq!(
|
|
usr.username, returned_usr.username,
|
|
"Usernames do not match"
|
|
);
|
|
assert!(returned_usr.date_created.is_some(), "Date Created is empty");
|
|
}
|
|
Err(err) => {
|
|
assert!(false, "Error: {:?}", err.to_string());
|
|
}
|
|
};
|
|
|
|
let _ = db_mgr::drop_database(&tm_pool, &db_name).await;
|
|
}
|
|
|
|
|
|
#[tokio::test]
|
|
async fn test_login_user() {
|
|
let tm_pool = db_mgr::get_pool().await.unwrap();
|
|
|
|
let db_name = db_mgr::generate_db_name().await;
|
|
|
|
match db_mgr::create_database(&tm_pool, &db_name).await {
|
|
Ok(_) => {
|
|
println!("Success");
|
|
}
|
|
Err(e) => {
|
|
assert!(false, "Error: {:?}", e.to_string());
|
|
}
|
|
}
|
|
|
|
let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
|
|
|
|
icarus_auth::db::migrations(&pool).await;
|
|
|
|
let app = init::routes().await.layer(axum::Extension(pool));
|
|
|
|
let usr = get_test_register_request();
|
|
|
|
let response = requests::register(&app, &usr).await;
|
|
|
|
match response {
|
|
Ok(resp) => {
|
|
assert_eq!(
|
|
resp.status(),
|
|
StatusCode::CREATED,
|
|
"Message: {:?} {:?}",
|
|
resp,
|
|
usr.username
|
|
);
|
|
let body = axum::body::to_bytes(resp.into_body(), usize::MAX)
|
|
.await
|
|
.unwrap();
|
|
let parsed_body: callers::register::response::Response =
|
|
serde_json::from_slice(&body).unwrap();
|
|
let returned_usr = &parsed_body.data[0];
|
|
|
|
assert_eq!(false, returned_usr.id.is_nil(), "Id is not populated");
|
|
|
|
assert_eq!(
|
|
usr.username, returned_usr.username,
|
|
"Usernames do not match"
|
|
);
|
|
assert!(returned_usr.date_created.is_some(), "Date Created is empty");
|
|
|
|
let login_payload = json!({
|
|
"username": &usr.username,
|
|
"password": &usr.password,
|
|
});
|
|
|
|
match app
|
|
.oneshot(
|
|
Request::builder()
|
|
.method(axum::http::Method::POST)
|
|
.uri(callers::endpoints::LOGIN)
|
|
.header(axum::http::header::CONTENT_TYPE, "application/json")
|
|
.body(Body::from(login_payload.to_string()))
|
|
.unwrap(),
|
|
)
|
|
.await
|
|
{
|
|
Ok(resp) => {
|
|
assert_eq!(StatusCode::OK, resp.status(), "Status is not right");
|
|
let body = axum::body::to_bytes(resp.into_body(), usize::MAX)
|
|
.await
|
|
.unwrap();
|
|
let parsed_body: callers::login::response::Response =
|
|
serde_json::from_slice(&body).unwrap();
|
|
let login_result = &parsed_body.data[0];
|
|
assert!(!login_result.id.is_nil(), "Id is nil");
|
|
}
|
|
Err(err) => {
|
|
assert!(false, "Error: {:?}", err.to_string());
|
|
}
|
|
}
|
|
}
|
|
Err(err) => {
|
|
assert!(false, "Error: {:?}", err.to_string());
|
|
}
|
|
};
|
|
|
|
let _ = db_mgr::drop_database(&tm_pool, &db_name).await;
|
|
}
|
|
}
|