mirror of https://github.com/rwf2/Rocket.git
102 lines
3.0 KiB
Rust
102 lines
3.0 KiB
Rust
|
use figment::{Figment, providers::Serialized};
|
||
|
|
use rocket::{Config, uri};
|
||
|
|
use rocket_http::uri::{Absolute, Asterisk, Authority, Origin, Reference};
|
||
|
|
use serde::{Serialize, Deserialize};
|
||
|
|
use pretty_assertions::assert_eq;
|
||
|
|
|
||
|
|
#[derive(PartialEq, Debug, Serialize, Deserialize)]
|
||
|
|
struct UriContainer<'a> {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
#[serde(borrow)]
|
||
|
|
origin: Origin<'a>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
authority: Authority<'a>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
absolute: Absolute<'a>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
reference: Reference<'a>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(PartialEq, Debug, Serialize, Deserialize)]
|
||
|
|
struct UriContainerOwned {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
#[serde(borrow)]
|
||
|
|
origin: Origin<'static>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
authority: Authority<'static>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
absolute: Absolute<'static>,
|
||
|
|
#[serde(borrow)]
|
||
|
|
reference: Reference<'static>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn uri_serde() {
|
||
|
|
figment::Jail::expect_with(|jail| {
|
||
|
|
jail.create_file("Rocket.toml", r#"
|
||
|
|
[default]
|
||
|
|
asterisk = "*"
|
||
|
|
origin = "/foo/bar?baz"
|
||
|
|
authority = "user:pass@rocket.rs:80"
|
||
|
|
absolute = "https://rocket.rs/foo/bar"
|
||
|
|
reference = "https://rocket.rs:8000/index.html"
|
||
|
|
"#)?;
|
||
|
|
|
||
|
|
let uris: UriContainer<'_> = Config::figment().extract()?;
|
||
|
|
assert_eq!(uris, UriContainer {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
origin: uri!("/foo/bar?baz"),
|
||
|
|
authority: uri!("user:pass@rocket.rs:80"),
|
||
|
|
absolute: uri!("https://rocket.rs/foo/bar"),
|
||
|
|
reference: uri!("https://rocket.rs:8000/index.html").into(),
|
||
|
|
});
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn uri_serde_owned() {
|
||
|
|
figment::Jail::expect_with(|jail| {
|
||
|
|
jail.create_file("Rocket.toml", r#"
|
||
|
|
[default]
|
||
|
|
asterisk = "*"
|
||
|
|
origin = "/foo/bar?baz"
|
||
|
|
authority = "user:pass@rocket.rs:80"
|
||
|
|
absolute = "https://rocket.rs/foo/bar"
|
||
|
|
reference = "https://rocket.rs:8000/index.html"
|
||
|
|
"#)?;
|
||
|
|
|
||
|
|
let uris: UriContainerOwned = Config::figment().extract()?;
|
||
|
|
assert_eq!(uris, UriContainerOwned {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
origin: uri!("/foo/bar?baz"),
|
||
|
|
authority: uri!("user:pass@rocket.rs:80"),
|
||
|
|
absolute: uri!("https://rocket.rs/foo/bar"),
|
||
|
|
reference: uri!("https://rocket.rs:8000/index.html").into(),
|
||
|
|
});
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#[test]
|
||
|
|
fn uri_serde_round_trip() {
|
||
|
|
let tmp = Figment::from(Serialized::defaults(UriContainer {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
origin: uri!("/foo/bar?baz"),
|
||
|
|
authority: uri!("user:pass@rocket.rs:80"),
|
||
|
|
absolute: uri!("https://rocket.rs/foo/bar"),
|
||
|
|
reference: uri!("https://rocket.rs:8000/index.html").into(),
|
||
|
|
}));
|
||
|
|
|
||
|
|
let uris: UriContainer<'_> = tmp.extract().expect("Parsing failed");
|
||
|
|
assert_eq!(uris, UriContainer {
|
||
|
|
asterisk: Asterisk,
|
||
|
|
origin: uri!("/foo/bar?baz"),
|
||
|
|
authority: uri!("user:pass@rocket.rs:80"),
|
||
|
|
absolute: uri!("https://rocket.rs/foo/bar"),
|
||
|
|
reference: uri!("https://rocket.rs:8000/index.html").into(),
|
||
|
|
});
|
||
|
|
}
|