use std::{ convert::TryFrom, sync::{ Arc, }, time::Duration, }; use tokio::{ runtime::Runtime, spawn, sync::oneshot, time::delay_for, }; use super::{ server, }; #[test] fn end_to_end () { use maplit::*; use reqwest::Client; use tracing::{debug, info}; // Prefer this form for tests, since all tests share one process // and we don't care if another test already installed a subscriber. tracing_subscriber::fmt ().try_init ().ok (); let mut rt = Runtime::new ().unwrap (); // Spawn the root task rt.block_on (async { let server_name = "aliens_wildland"; let api_key = "AnacondaHardcoverGrannyUnlatchLankinessMutate"; let tripcode = base64::encode (blake3::hash (api_key.as_bytes ()).as_bytes ()); debug! ("Relay is expecting tripcode {}", tripcode); let config_file = ptth_relay::config::file::Config { port: None, servers: hashmap! { server_name.into () => ptth_relay::config::file::Server { tripcode, display_name: None, }, }, }; let relay_state = Arc::new (ptth_relay::RelayState::from (ptth_relay::config::Config::try_from (config_file).unwrap ())); let relay_state_2 = relay_state.clone (); let (stop_relay_tx, stop_relay_rx) = oneshot::channel (); let task_relay = spawn (async move { ptth_relay::run_relay (relay_state_2, stop_relay_rx, None).await.unwrap (); }); assert! (relay_state.list_servers ().await.is_empty ()); let relay_url = "http://127.0.0.1:4000"; let config_file = server::ConfigFile { name: server_name.into (), api_key: api_key.into (), relay_url: "http://127.0.0.1:4000/7ZSFUKGV".into (), file_server_root: None, }; let (stop_server_tx, stop_server_rx) = oneshot::channel (); let task_server = { spawn (async move { server::run_server (config_file, stop_server_rx, None, None).await.unwrap (); }) }; delay_for (Duration::from_millis (500)).await; assert_eq! (relay_state.list_servers ().await, vec! [ server_name.to_string (), ]); let client = Client::builder () .timeout (Duration::from_secs (2)) .build ().unwrap (); let resp = client.get (&format! ("{}/frontend/relay_up_check", relay_url)) .send ().await.unwrap ().bytes ().await.unwrap (); assert_eq! (resp, "Relay is up\n"); let resp = client.get (&format! ("{}/frontend/servers/{}/files/COPYING", relay_url, server_name)) .send ().await.unwrap ().bytes ().await.unwrap (); if blake3::hash (&resp) != blake3::Hash::from ([ 0xca, 0x02, 0x92, 0x78, 0x9c, 0x0a, 0x0e, 0xcb, 0xa7, 0x06, 0xf4, 0xb3, 0xf3, 0x49, 0x30, 0x07, 0xa9, 0x95, 0x17, 0x31, 0xc1, 0xd4, 0x32, 0xc5, 0x2c, 0x4a, 0xac, 0x1f, 0x1a, 0xbb, 0xa8, 0xef, ]) { panic! ("{}", String::from_utf8 (resp.to_vec ()).unwrap ()); } // Requesting a file from a server that isn't registered // will error out let resp = client.get (&format! ("{}/frontend/servers/obviously_this_server_does_not_exist/files/COPYING", relay_url)) .send ().await.unwrap (); assert_eq! (resp.status (), reqwest::StatusCode::NOT_FOUND); info! ("Shutting down end-to-end test"); stop_server_tx.send (()).unwrap (); stop_relay_tx.send (()).unwrap (); info! ("Sent stop messages"); task_relay.await.unwrap (); info! ("Relay stopped"); task_server.await.unwrap (); info! ("Server stopped"); }); }