add integration test stub

This commit is contained in:
Ilion Beyst 2022-09-30 17:28:24 +02:00
parent 8178148996
commit 8f6b34db10
3 changed files with 75 additions and 9 deletions

View file

@ -49,4 +49,6 @@ shlex = "1.1"
tonic-build = "0.7.2" tonic-build = "0.7.2"
[dev-dependencies] [dev-dependencies]
parking_lot = "0.11" tower = { version = "0.4", features = ["util"] }
parking_lot = "0.12"
tempfile = "3"

View file

@ -112,7 +112,8 @@ fn init_directories(config: &GlobalConfig) -> std::io::Result<()> {
Ok(()) Ok(())
} }
pub fn api() -> Router { // just the routes, without connecting state
fn api() -> Router {
Router::new() Router::new()
.route("/register", post(routes::users::register)) .route("/register", post(routes::users::register))
.route("/login", post(routes::users::login)) .route("/login", post(routes::users::login))
@ -140,6 +141,14 @@ pub fn api() -> Router {
.route("/save_bot", post(routes::bots::save_bot)) .route("/save_bot", post(routes::bots::save_bot))
} }
// full service
pub fn create_pw_api(global_config: Arc<GlobalConfig>, db_pool: DbPool) -> Router {
Router::new()
.nest("/api", api())
.layer(Extension(db_pool))
.layer(Extension(global_config))
}
pub fn get_config() -> Result<GlobalConfig, ConfigError> { pub fn get_config() -> Result<GlobalConfig, ConfigError> {
config::Config::builder() config::Config::builder()
.add_source(config::File::with_name("configuration.toml")) .add_source(config::File::with_name("configuration.toml"))
@ -175,16 +184,14 @@ pub async fn run_app() {
tokio::spawn(run_registry(global_config.clone(), db_pool.clone())); tokio::spawn(run_registry(global_config.clone(), db_pool.clone()));
tokio::spawn(run_client_api(global_config.clone(), db_pool.clone())); tokio::spawn(run_client_api(global_config.clone(), db_pool.clone()));
let api_service = Router::new()
.nest("/api", api())
.layer(Extension(db_pool))
.layer(Extension(global_config))
.into_make_service();
// TODO: put in config // TODO: put in config
let addr = SocketAddr::from(([127, 0, 0, 1], 9000)); let addr = SocketAddr::from(([127, 0, 0, 1], 9000));
axum::Server::bind(&addr).serve(api_service).await.unwrap(); let pw_api_service = create_pw_api(global_config, db_pool).into_make_service();
axum::Server::bind(&addr)
.serve(pw_api_service)
.await
.unwrap();
} }
// we can also write a custom extractor that grabs a connection from the pool // we can also write a custom extractor that grabs a connection from the pool

View file

@ -0,0 +1,57 @@
use axum::{
body::Body,
http::{self, Request, StatusCode},
};
use planetwars_server::{create_db_pool, create_pw_api, GlobalConfig};
use serde_json::{self, json, Value as JsonValue};
use std::{io, path::Path, sync::Arc};
use tempfile::TempDir;
use tower::ServiceExt;
// Used to serialize tests that access the database.
// TODO: see to what degree we could support transactional testing.
static DB_LOCK: parking_lot::Mutex<()> = parking_lot::Mutex::new(());
fn create_subdir<P: AsRef<Path>>(base_path: &Path, p: P) -> io::Result<String> {
let dir_path = base_path.join(p);
std::fs::create_dir(&dir_path)?;
let dir_path_string = dir_path.into_os_string().into_string().unwrap();
Ok(dir_path_string)
}
#[tokio::test]
async fn test_application() -> io::Result<()> {
let _db_guard = DB_LOCK.lock();
let data_dir = TempDir::new().expect("failed to create temp dir");
let config = Arc::new(GlobalConfig {
database_url: "postgresql://planetwars:planetwars@localhost/planetwars-test".to_string(),
python_runner_image: "python:3.10-slim-buster".to_string(),
container_registry_url: "localhost:9001".to_string(),
root_url: "localhost:3000".to_string(),
bots_directory: create_subdir(data_dir.path(), "bots")?,
match_logs_directory: create_subdir(data_dir.path(), "matches")?,
maps_directory: create_subdir(data_dir.path(), "maps")?,
registry_directory: create_subdir(data_dir.path(), "registry")?,
registry_admin_password: "secret_admin_password".to_string(),
ranker_enabled: false,
});
let db_pool = create_db_pool(&config).await;
let app = create_pw_api(config, db_pool);
let response = app
.oneshot(
Request::builder()
.method(http::Method::GET)
.uri("/api/bots")
.body(Body::empty())
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let resp: JsonValue = serde_json::from_slice(&body).unwrap();
assert_eq!(resp, json!([]));
Ok(())
}