deblan-mirror/sero
1
0
Fork 0
mirror of https://github.com/clowzed/sero synced 2024-05-23 01:32:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'master' into feature/cors

Browse source
This commit is contained in:
Dmitry Miasnenko 2023-11-12 18:52:16 +03:00 committed by GitHub
parent f16432cb2a df82dbce28
commit 4b2bf61274
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 1086 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
.github/workflows/tests.yml vendored Normal file
View file

@ -0,0 +1,36 @@
name: Running Tests
on: [push]
jobs:
test:
runs-on: ubuntu-latest
services:
postgres:
image: postgres
env:
POSTGRES_PASSWORD: postgres
POSTGRES_USER: postgres
POSTGRES_DB: postgres
options: >-
--health-cmd pg_isready
--health-interval 10s
--health-timeout 5s
--health-retries 5
ports:
- 5432:5432
steps:
- uses: actions/checkout@v2
- name: Build
run: cargo build --verbose
- name: Run tests
run: cargo test --verbose -- --test-threads=1
env:
DATABASE_URL: "postgres://postgres:postgres@localhost:5432/postgres"
PORT: 8080

313
Cargo.lock generated
View file

@ -425,6 +425,24 @@ dependencies = [
"syn 2.0.37",
]
[[package]]
name = "axum-test-helper"
version = "0.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "298f62fa902c2515c169ab0bfb56c593229f33faa01131215d58e3d4898e3aa9"
dependencies = [
"axum",
"bytes",
"http",
"http-body",
"hyper",
"reqwest",
"serde",
"tokio",
"tower",
"tower-service",
]
[[package]]
name = "axum_typed_multipart"
version = "0.10.0"
@ -762,6 +780,16 @@ dependencies = [
"version_check",
]
[[package]]
name = "core-foundation"
version = "0.9.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "194a7a9e6de53fa55116934067c844d9d749312f75c6f6d0980e8c252f8c2146"
dependencies = [
"core-foundation-sys",
"libc",
]
[[package]]
name = "core-foundation-sys"
version = "0.8.4"
@ -1052,6 +1080,21 @@ version = "1.0.7"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3f9eec918d3f24069decb9af1554cad7c880e2da24a9afd88aca000531ab82c1"
[[package]]
name = "foreign-types"
version = "0.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f6f339eb8adc052cd2ca78910fda869aefa38d22d5cb648e6485e4d3fc06f3b1"
dependencies = [
"foreign-types-shared",
]
[[package]]
name = "foreign-types-shared"
version = "0.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "00b0228411908ca8685dba7fc2cdd70ec9990a6e753e89b6ac91a84c40fbaf4b"
[[package]]
name = "form_urlencoded"
version = "1.2.0"
@ -1227,6 +1270,25 @@ dependencies = [
"wasm-bindgen",
]
[[package]]
name = "h2"
version = "0.3.21"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "91fc23aa11be92976ef4729127f1a74adf36d8436f7816b185d18df956790833"
dependencies = [
"bytes",
"fnv",
"futures-core",
"futures-sink",
"futures-util",
"http",
"indexmap 1.9.3",
"slab",
"tokio",
"tokio-util",
"tracing",
]
[[package]]
name = "hashbrown"
version = "0.12.3"
@ -1386,6 +1448,7 @@ dependencies = [
"futures-channel",
"futures-core",
"futures-util",
"h2",
"http",
"http-body",
"httparse",
@ -1399,6 +1462,19 @@ dependencies = [
"want",
]
[[package]]
name = "hyper-tls"
version = "0.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d6183ddfa99b85da61a140bea0efc93fdf56ceaa041b37d553518030827f9905"
dependencies = [
"bytes",
"hyper",
"native-tls",
"tokio",
"tokio-native-tls",
]
[[package]]
name = "iana-time-zone"
version = "0.1.57"
@ -1438,6 +1514,16 @@ dependencies = [
"unicode-normalization",
]
[[package]]
name = "indexmap"
version = "1.9.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bd070e393353796e801d209ad339e89596eb4c8d430d18ede6a1cced8fafbd99"
dependencies = [
"autocfg",
"hashbrown 0.12.3",
]
[[package]]
name = "indexmap"
version = "2.0.0"
@ -1479,6 +1565,12 @@ dependencies = [
"windows-sys",
]
[[package]]
name = "ipnet"
version = "2.9.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8f518f335dce6725a761382244631d86cf0ccb2863413590b31338feb467f9c3"
[[package]]
name = "itertools"
version = "0.11.0"
@ -1654,6 +1746,16 @@ version = "0.3.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6877bb514081ee2a7ff5ef9de3281f14a4dd4bceac4c09388074a6b5df8a139a"
[[package]]
name = "mime_guess"
version = "2.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4192263c238a5f0d0c6bfd21f336a313a4ce1c450542449ca191bb657b4642ef"
dependencies = [
"mime",
"unicase",
]
[[package]]
name = "minimal-lexical"
version = "0.2.1"
@ -1698,6 +1800,24 @@ dependencies = [
"version_check",
]
[[package]]
name = "native-tls"
version = "0.2.11"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "07226173c32f2926027b63cce4bcd8076c3552846cbe7925f3aaffeac0a3b92e"
dependencies = [
"lazy_static",
"libc",
"log",
"openssl",
"openssl-probe",
"openssl-sys",
"schannel",
"security-framework",
"security-framework-sys",
"tempfile",
]
[[package]]
name = "nom"
version = "7.1.3"
@ -1802,6 +1922,50 @@ version = "1.18.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dd8b5dd2ae5ed71462c540258bedcb51965123ad7e7ccf4b9a8cafaa4a63576d"
[[package]]
name = "openssl"
version = "0.10.57"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bac25ee399abb46215765b1cb35bc0212377e58a061560d8b29b024fd0430e7c"
dependencies = [
"bitflags 2.4.0",
"cfg-if",
"foreign-types",
"libc",
"once_cell",
"openssl-macros",
"openssl-sys",
]
[[package]]
name = "openssl-macros"
version = "0.1.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a948666b637a0f465e8564c73e89d4dde00d72d4d473cc972f390fc3dcee7d9c"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.37",
]
[[package]]
name = "openssl-probe"
version = "0.1.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ff011a302c396a5197692431fc1948019154afc178baf7d8e37367442a4601cf"
[[package]]
name = "openssl-sys"
version = "0.9.93"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "db4d56a4c0478783083cfafcc42493dd4a981d41669da64b4572a2a089b51b1d"
dependencies = [
"cc",
"libc",
"pkg-config",
"vcpkg",
]
[[package]]
name = "ordered-float"
version = "3.9.1"
@ -2208,6 +2372,47 @@ dependencies = [
"bytecheck",
]
[[package]]
name = "reqwest"
version = "0.11.22"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "046cd98826c46c2ac8ddecae268eb5c2e58628688a5fc7a2643704a73faba95b"
dependencies = [
"base64 0.21.4",
"bytes",
"encoding_rs",
"futures-core",
"futures-util",
"h2",
"http",
"http-body",
"hyper",
"hyper-tls",
"ipnet",
"js-sys",
"log",
"mime",
"mime_guess",
"native-tls",
"once_cell",
"percent-encoding",
"pin-project-lite",
"serde",
"serde_json",
"serde_urlencoded",
"system-configuration",
"tokio",
"tokio-native-tls",
"tokio-util",
"tower-service",
"url",
"wasm-bindgen",
"wasm-bindgen-futures",
"wasm-streams",
"web-sys",
"winreg",
]
[[package]]
name = "ring"
version = "0.16.20"
@ -2374,6 +2579,15 @@ version = "1.0.15"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1ad4cc8da4ef723ed60bced201181d83791ad433213d8c24efffda1eec85d741"
[[package]]
name = "schannel"
version = "0.1.22"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0c3733bf4cf7ea0880754e19cb5a462007c4a8c1914bff372ccc95b464f1df88"
dependencies = [
"windows-sys",
]
[[package]]
name = "scopeguard"
version = "1.2.0"
@ -2555,6 +2769,29 @@ version = "4.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1c107b6f4780854c8b126e228ea8869f4d7b71260f962fefb57b996b8959ba6b"
[[package]]
name = "security-framework"
version = "2.9.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "05b64fb303737d99b81884b2c63433e9ae28abebe5eb5045dcdd175dc2ecf4de"
dependencies = [
"bitflags 1.3.2",
"core-foundation",
"core-foundation-sys",
"libc",
"security-framework-sys",
]
[[package]]
name = "security-framework-sys"
version = "2.9.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e932934257d3b408ed8f30db49d85ea163bfe74961f017f405b025af298f0c7a"
dependencies = [
"core-foundation-sys",
"libc",
]
[[package]]
name = "serde"
version = "1.0.188"
@ -2577,9 +2814,9 @@ dependencies = [
[[package]]
name = "serde_json"
version = "1.0.107"
version = "1.0.108"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "6b420ce6e3d8bd882e9b243c6eed35dbc9a6110c9769e74b584e0d68d1f20c65"
checksum = "3d1c7e3eac408d115102c4c24ad393e0821bb3a5df4d506a80f85f7a742a526b"
dependencies = [
"itoa",
"ryu",
@ -2616,6 +2853,7 @@ dependencies = [
"async_zip",
"axum",
"axum-extra",
"axum-test-helper",
"axum_typed_multipart",
"bytes",
"chrono",
@ -2624,16 +2862,22 @@ dependencies = [
"envy",
"futures",
"futures-util",
"hyper",
"jsonwebtoken",
"migration",
"mime",
"reqwest",
"sea-orm",
"serde",
"serde_json",
"serde_urlencoded",
"sha256",
"thiserror",
"tokio",
"tokio-postgres",
"tokio-util",
"tower-http",
"tower",
"tracing",
"tracing-subscriber",
"uuid",
@ -2834,7 +3078,7 @@ dependencies = [
"futures-util",
"hashlink",
"hex",
"indexmap",
"indexmap 2.0.0",
"log",
"memchr",
"once_cell",
@ -3077,6 +3321,27 @@ version = "0.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2047c6ded9c721764247e62cd3b03c09ffc529b2ba5b10ec482ae507a4a70160"
[[package]]
name = "system-configuration"
version = "0.5.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ba3a3adc5c275d719af8cb4272ea1c4a6d668a777f37e115f6d11ddbc1c8e0e7"
dependencies = [
"bitflags 1.3.2",
"core-foundation",
"system-configuration-sys",
]
[[package]]
name = "system-configuration-sys"
version = "0.5.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a75fb188eb626b924683e3b95e3a48e63551fcfb51949de2f06a9d91dbee93c9"
dependencies = [
"core-foundation-sys",
"libc",
]
[[package]]
name = "tap"
version = "1.0.1"
@ -3199,6 +3464,16 @@ dependencies = [
"syn 2.0.37",
]
[[package]]
name = "tokio-native-tls"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bbae76ab933c85776efabc971569dd6119c580d8f5d448769dec1764bf796ef2"
dependencies = [
"native-tls",
"tokio",
]
[[package]]
name = "tokio-postgres"
version = "0.7.10"
@ -3387,6 +3662,15 @@ version = "0.10.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f720def6ce1ee2fc44d40ac9ed6d3a59c361c80a75a7aa8e75bb9baed31cf2ea"
[[package]]
name = "unicase"
version = "2.7.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f7d2d4dafb69621809a81864c9c1b864479e1235c0dd4e199924b9742439ed89"
dependencies = [
"version_check",
]
[[package]]
name = "unicode-bidi"
version = "0.3.13"
@ -3564,6 +3848,19 @@ version = "0.2.87"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ca6ad05a4870b2bf5fe995117d3728437bd27d7cd5f06f13c17443ef369775a1"
[[package]]
name = "wasm-streams"
version = "0.3.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b4609d447824375f43e1ffbc051b50ad8f4b3ae8219680c94452ea05eb240ac7"
dependencies = [
"futures-util",
"js-sys",
"wasm-bindgen",
"wasm-bindgen-futures",
"web-sys",
]
[[package]]
name = "web-sys"
version = "0.3.64"
@ -3690,6 +3987,16 @@ version = "0.48.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ed94fce61571a4006852b7389a063ab983c02eb1bb37b47f8272ce92d06d9538"
[[package]]
name = "winreg"
version = "0.50.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "524e57b2c537c0f9b1e69f1965311ec12182b4122e45035b1508cd24d2adadb1"
dependencies = [
"cfg-if",
"windows-sys",
]
[[package]]
name = "wyz"
version = "0.5.1"

7
Cargo.toml
View file

@ -34,6 +34,13 @@ tokio-util = { version = "0.7.9", features = ["io"] }
dotenv = "0.15.0"
tower-http = { version = "0.4.4", features = ["cors"] }
futures = "0.3.29"
tower = { version = "0.4.13", features = ["util"] }
serde_urlencoded = "0.7.1"
serde_json = "1.0.108"
mime = "0.3.17"
axum-test-helper = "0.3.0"
hyper = "0.14.27"
reqwest = { version = "0.11.22", features = ["json", "multipart"] }
[workspace]
members = [".", "entity", "migration"]

BIN
assets/a.zip Normal file
View file

Binary file not shown.

2
src/handlers/auth.rs
View file

@ -15,7 +15,7 @@ use std::sync::Arc;
#[derive(serde::Serialize, serde::Deserialize)]
pub struct AuthToken {
token: Jwt,
pub token: Jwt,
}
#[tracing::instrument(skip(state))]

742
src/main.rs
View file

@ -26,9 +26,7 @@ pub struct AppState {
config: config::Config,
}
#[tokio::main]
async fn main() {
dotenv::dotenv().ok();
async fn app() -> Router {
let config = config::Config::default();
let mut opt = ConnectOptions::new(&config.database_url);
@ -38,12 +36,6 @@ async fn main() {
Migrator::up(&connection, None).await.unwrap();
let addr = SocketAddr::from(([0, 0, 0, 0], config.port));
tracing_subscriber::fmt()
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
.init();
let api_router = Router::new()
.route("/health", get(|| async { StatusCode::OK }))
.route("/login", post(handlers::auth::login))
@ -118,8 +110,738 @@ async fn main() {
app = app.layer(DefaultBodyLimit::max(config.max_body_limit_size.unwrap()));
}
app
}
#[tokio::main]
async fn main() {
dotenv::dotenv().ok();
let config = config::Config::default();
let addr = SocketAddr::from(([0, 0, 0, 0], config.port));
tracing_subscriber::fmt()
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
.init();
axum::Server::bind(&addr)
.serve(app.into_make_service())
.serve(app().await.into_make_service())
.await
.unwrap();
}
#[cfg(test)]
mod tests {
use crate::{handlers::auth::AuthToken, services::auth::AuthCredentials};
use super::*;
use axum::{
body::Body,
http::{header, HeaderName, HeaderValue, Method, Request, StatusCode},
};
use axum_test_helper::TestClient;
use std::io::Read;
use tower::ServiceExt; // for `oneshot` and `ready`
async fn prepare() -> Router {
dotenv::dotenv().ok();
//tracing_subscriber::fmt()
// .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
// .try_init()
// .ok();
app().await
}
#[tokio::test]
async fn health_check() {
let app = prepare().await;
let response = app
.oneshot(
Request::builder()
.uri("/api/health")
.body(Body::empty())
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK)
}
#[tokio::test]
async fn registration_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::CONFLICT);
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: String::default(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::BAD_REQUEST);
}
#[tokio::test]
async fn login_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
assert!(serde_json::from_slice::<AuthToken>(&body).is_ok());
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: String::from("wrong"),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::UNAUTHORIZED);
}
#[tokio::test]
async fn upload_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let first_user = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let second_user = serde_urlencoded::to_string(AuthCredentials {
username: uuid::Uuid::new_v4().to_string(),
password: uuid::Uuid::new_v4().to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(second_user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(first_user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(first_user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let token = serde_json::from_slice::<AuthToken>(&body).unwrap().token;
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(second_user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let second_token = serde_json::from_slice::<AuthToken>(&body).unwrap().token;
let client = TestClient::new(app);
let test_zip = std::path::PathBuf::from("./assets/a.zip");
assert!(test_zip.exists());
let mut f = std::fs::File::open(&test_zip).expect("no file found");
let metadata = std::fs::metadata(&test_zip).expect("unable to read metadata");
let mut buffer = vec![0; metadata.len() as usize];
f.read_exact(&mut buffer).expect("buffer overflow");
let form = reqwest::multipart::Form::new()
.part("archive", reqwest::multipart::Part::bytes(buffer.clone()));
let random_subdoamain = uuid::Uuid::new_v4().to_string();
let bearer = format!("Bearer {token}");
let response = client
.post("/api/upload")
.multipart(form)
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let bearer = format!("Bearer {second_token}");
let form = reqwest::multipart::Form::new()
.part("archive", reqwest::multipart::Part::bytes(buffer));
let response = client
.post("/api/upload")
.multipart(form)
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::FORBIDDEN);
}
#[tokio::test]
async fn teardown_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let token = serde_json::from_slice::<AuthToken>(&body).unwrap().token;
let client = TestClient::new(app);
let test_zip = std::path::PathBuf::from("./assets/a.zip");
assert!(test_zip.exists());
let mut f = std::fs::File::open(&test_zip).expect("no file found");
let metadata = std::fs::metadata(&test_zip).expect("unable to read metadata");
let mut buffer = vec![0; metadata.len() as usize];
f.read_exact(&mut buffer).expect("buffer overflow");
let form = reqwest::multipart::Form::new()
.part("archive", reqwest::multipart::Part::bytes(buffer));
let random_subdoamain = uuid::Uuid::new_v4().to_string();
let bearer = format!("Bearer {token}");
let response = client
.post("/api/upload")
.multipart(form)
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.get("/a/index.html")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.post("/api/teardown")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.get("/a/index")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::NOT_FOUND);
}
#[tokio::test]
async fn disable_enable_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let body = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let token = serde_json::from_slice::<AuthToken>(&body).unwrap().token;
let client = TestClient::new(app);
let test_zip = std::path::PathBuf::from("./assets/a.zip");
assert!(test_zip.exists());
let mut f = std::fs::File::open(&test_zip).expect("no file found");
let metadata = std::fs::metadata(&test_zip).expect("unable to read metadata");
let mut buffer = vec![0; metadata.len() as usize];
f.read_exact(&mut buffer).expect("buffer overflow");
let form = reqwest::multipart::Form::new()
.part("archive", reqwest::multipart::Part::bytes(buffer));
let random_subdoamain = uuid::Uuid::new_v4().to_string();
let bearer = format!("Bearer {token}");
let response = client
.post("/api/upload")
.multipart(form)
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.get("/a/index.html")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.post("/api/disable")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.get("/a/index")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::SERVICE_UNAVAILABLE);
let response = client
.post("/api/enable")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
}
#[tokio::test]
async fn download_check() {
let app = prepare().await;
let random_username = uuid::Uuid::new_v4();
let random_password = uuid::Uuid::new_v4();
let user = serde_urlencoded::to_string(AuthCredentials {
username: random_username.to_string(),
password: random_password.to_string(),
})
.unwrap();
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/registration")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let response = app
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/login")
.header(
header::CONTENT_TYPE,
mime::APPLICATION_WWW_FORM_URLENCODED.as_ref(),
)
.body(Body::from(user.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
let token = serde_json::from_slice::<AuthToken>(&body).unwrap().token;
let client = TestClient::new(app.clone());
let test_zip = std::path::PathBuf::from("./assets/a.zip");
assert!(test_zip.exists());
let mut f = std::fs::File::open(&test_zip).expect("no file found");
let metadata = std::fs::metadata(&test_zip).expect("unable to read metadata");
let mut buffer = vec![0; metadata.len() as usize];
f.read_exact(&mut buffer).expect("buffer overflow");
let form = reqwest::multipart::Form::new()
.part("archive", reqwest::multipart::Part::bytes(buffer.clone()));
let random_subdoamain = uuid::Uuid::new_v4().to_string();
let bearer = format!("Bearer {token}");
let response = client
.post("/api/upload")
.multipart(form)
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let response = client
.get("/a/index.html")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.send()
.await;
assert_eq!(response.status(), StatusCode::OK);
let bearer = format!("Bearer {token}");
let response = app
.clone()
.clone()
.oneshot(
Request::builder()
.method(Method::POST)
.uri("/api/download")
.header(
HeaderName::from_static("x-subdomain"),
HeaderValue::from_str(&random_subdoamain).unwrap(),
)
.header(
header::AUTHORIZATION,
HeaderValue::from_str(&bearer).unwrap(),
)
.body(Body::from(body.clone()))
.unwrap(),
)
.await
.unwrap();
assert_eq!(response.status(), StatusCode::OK);
let body = hyper::body::to_bytes(response.into_body()).await.unwrap();
assert_eq!(body, buffer)
}
}