The start of unit testing and cleaner code for driver fetching :D
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user