♻️ Extract src/tests.rs
_
parent
28ce6a32cd
commit
7c2ce65864
156
src/lib.rs
156
src/lib.rs
|
|
@ -41,158 +41,4 @@ pub fn password_is_bad (mut password: String) -> bool {
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg (test)]
|
#[cfg (test)]
|
||||||
mod tests {
|
mod tests;
|
||||||
use std::{
|
|
||||||
convert::TryFrom,
|
|
||||||
sync::{
|
|
||||||
Arc,
|
|
||||||
},
|
|
||||||
time::Duration,
|
|
||||||
};
|
|
||||||
|
|
||||||
use tokio::{
|
|
||||||
runtime::Runtime,
|
|
||||||
spawn,
|
|
||||||
sync::oneshot,
|
|
||||||
time::delay_for,
|
|
||||||
};
|
|
||||||
|
|
||||||
use super::{
|
|
||||||
relay,
|
|
||||||
server,
|
|
||||||
};
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn check_bad_passwords () {
|
|
||||||
use crate::password_is_bad;
|
|
||||||
|
|
||||||
for pw in &[
|
|
||||||
"",
|
|
||||||
" ",
|
|
||||||
"user",
|
|
||||||
"password",
|
|
||||||
"pAsSwOrD",
|
|
||||||
"secret",
|
|
||||||
"123123",
|
|
||||||
] {
|
|
||||||
assert! (password_is_bad (pw.to_string ()));
|
|
||||||
}
|
|
||||||
|
|
||||||
use rand::prelude::*;
|
|
||||||
|
|
||||||
let mut entropy = [0u8; 32];
|
|
||||||
thread_rng ().fill_bytes (&mut entropy);
|
|
||||||
let good_password = base64::encode (entropy);
|
|
||||||
|
|
||||||
assert! (! password_is_bad (good_password));
|
|
||||||
}
|
|
||||||
|
|
||||||
#[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 = relay::config::file::Config {
|
|
||||||
port: None,
|
|
||||||
servers: hashmap! {
|
|
||||||
server_name.into () => relay::config::file::Server {
|
|
||||||
tripcode,
|
|
||||||
display_name: None,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
let relay_state = Arc::new (relay::RelayState::from (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 {
|
|
||||||
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");
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,153 @@
|
||||||
|
use std::{
|
||||||
|
convert::TryFrom,
|
||||||
|
sync::{
|
||||||
|
Arc,
|
||||||
|
},
|
||||||
|
time::Duration,
|
||||||
|
};
|
||||||
|
|
||||||
|
use tokio::{
|
||||||
|
runtime::Runtime,
|
||||||
|
spawn,
|
||||||
|
sync::oneshot,
|
||||||
|
time::delay_for,
|
||||||
|
};
|
||||||
|
|
||||||
|
use super::{
|
||||||
|
relay,
|
||||||
|
server,
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn check_bad_passwords () {
|
||||||
|
use crate::password_is_bad;
|
||||||
|
|
||||||
|
for pw in &[
|
||||||
|
"",
|
||||||
|
" ",
|
||||||
|
"user",
|
||||||
|
"password",
|
||||||
|
"pAsSwOrD",
|
||||||
|
"secret",
|
||||||
|
"123123",
|
||||||
|
] {
|
||||||
|
assert! (password_is_bad (pw.to_string ()));
|
||||||
|
}
|
||||||
|
|
||||||
|
use rand::prelude::*;
|
||||||
|
|
||||||
|
let mut entropy = [0u8; 32];
|
||||||
|
thread_rng ().fill_bytes (&mut entropy);
|
||||||
|
let good_password = base64::encode (entropy);
|
||||||
|
|
||||||
|
assert! (! password_is_bad (good_password));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[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 = relay::config::file::Config {
|
||||||
|
port: None,
|
||||||
|
servers: hashmap! {
|
||||||
|
server_name.into () => relay::config::file::Server {
|
||||||
|
tripcode,
|
||||||
|
display_name: None,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
let relay_state = Arc::new (relay::RelayState::from (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 {
|
||||||
|
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");
|
||||||
|
});
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue