feat: allow expecting status codes

This commit is contained in:
əlemi 2024-10-19 22:02:05 +02:00
parent 147186f8dc
commit 23abb7706e
Signed by: alemi
GPG key ID: A4895B84D311642C
3 changed files with 14 additions and 2 deletions

View file

@ -11,6 +11,10 @@ url = "https://api.alemi.dev/"
url = "https://api.alemi.dev/look/into/the/void" url = "https://api.alemi.dev/look/into/the/void"
extract = { type = "discard" } extract = { type = "discard" }
[route.notfound]
url = "https://api.alemi.dev/not-found"
expect = 404
[route.debug] [route.debug]
url = "https://api.alemi.dev/debug" url = "https://api.alemi.dev/debug"
method = "PUT" method = "PUT"

View file

@ -37,6 +37,9 @@ pub enum PostWomanError {
#[error("invalid regex: {0:?}")] #[error("invalid regex: {0:?}")]
InvalidRegex(#[from] regex::Error), InvalidRegex(#[from] regex::Error),
#[error("request didn't match expected status code: {0:?}")]
UnexpectedStatusCode(reqwest::Response),
#[error("invalid Json Query: {0}")] #[error("invalid Json Query: {0}")]
JQError(String), JQError(String),

View file

@ -21,6 +21,8 @@ pub struct Endpoint {
pub headers: Option<Vec<String>>, pub headers: Option<Vec<String>>,
/// body, optional string /// body, optional string
pub body: Option<StringOr<toml::Table>>, pub body: Option<StringOr<toml::Table>>,
/// expected error code, will fail if different
pub expect: Option<u16>,
/// response extractor /// response extractor
pub extract: Option<StringOr<Extractor>>, pub extract: Option<StringOr<Extractor>>,
} }
@ -135,8 +137,11 @@ impl Endpoint {
.headers(headers) .headers(headers)
.body(body) .body(body)
.send() .send()
.await? .await?;
.error_for_status()?;
if res.status().as_u16() != self.expect.unwrap_or(200) {
return Err(PostWomanError::UnexpectedStatusCode(res));
}
Ok(match self.extract.unwrap_or_default() { Ok(match self.extract.unwrap_or_default() {
StringOr::T(Extractor::Discard) => "".to_string(), StringOr::T(Extractor::Discard) => "".to_string(),