The start of unit testing and cleaner code for driver fetching :D

This commit is contained in:
LinlyBoi
2022-12-31 19:07:09 +02:00
parent ebe885e02d
commit 042867e0db
3 changed files with 49 additions and 9 deletions

View File

@@ -1,7 +1,11 @@
use actix_web::{web, App, HttpServer};
use dotenv::dotenv;
use sqlx::postgres::PgPoolOptions;
use std::env;
#[cfg(test)]
mod tests;
pub mod admin_data;
pub mod driver_data;
pub mod pog;
@@ -10,18 +14,25 @@ pub mod ticket_data;
pub mod vehicle_data;
#[actix_web::main]
async fn main() -> Result<(), sqlx::Error> {
async fn main() -> std::io::Result<()> {
dotenv().ok();
let port: u16 = env::var("ACTIX_PORT")
.expect("SET ACTIX_PORT PLOX")
.parse()
.expect("NOPE CANT PARSE THIS WHAT DID YOU PUT IN?!!!");
let address = env::var("ACTIX_IP").expect("SET ACTIX_IP PLOX");
let database_url = env::var("DATABASE_URL").expect("Put a DB url in the .env file dumbass");
let pool = PgPoolOptions::new()
.max_connections(10)
.connect(database_url.as_str())
.await?;
let row: (i64,) = sqlx::query_as("SELECT $1 FROM drivers")
.bind(150_i64)
.fetch_one(&pool)
.await?;
assert_eq!(row.0, 150);
println!("my nuts {}", row.0);
Ok(())
.await
.expect("No pool connection man :(");
HttpServer::new(move || {
App::new()
.app_data(web::Data::new(pool.clone()))
.service(driver_data::list_all)
})
.bind((address.as_str(), port))?
.run()
.await
}