mirror of https://github.com/rwf2/Rocket.git
139 lines
4.6 KiB
Rust
139 lines
4.6 KiB
Rust
use std::fmt::Debug;
|
|
|
|
use rocket::form::{Form, FromForm};
|
|
use rocket::form::error::{Error, Errors, ErrorKind};
|
|
|
|
#[derive(Debug, FromForm)]
|
|
struct Cat<'v> {
|
|
#[field(validate = len(5..))]
|
|
name: &'v str,
|
|
#[field(validate = starts_with("kitty"))]
|
|
nick: &'v str,
|
|
}
|
|
|
|
#[derive(Debug, FromForm)]
|
|
struct Dog<'v> {
|
|
#[field(validate = len(5..))]
|
|
name: &'v str,
|
|
}
|
|
|
|
#[derive(Debug, FromForm)]
|
|
struct Person<'v> {
|
|
kitty: Cat<'v>,
|
|
#[field(validate = len(1..))]
|
|
cats: Vec<Cat<'v>>,
|
|
dog: Dog<'v>,
|
|
}
|
|
|
|
fn starts_with<'v, S: AsRef<str>>(string: S, prefix: &str) -> Result<(), Errors<'v>> {
|
|
if !string.as_ref().starts_with(prefix) {
|
|
Err(Error::validation(format!("must start with {:?}", prefix)))?
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[track_caller]
|
|
fn errors<'v, T: FromForm<'v> + Debug + 'v>(string: &'v str) -> Errors<'v> {
|
|
Form::<T>::parse(string).expect_err("expected an error")
|
|
}
|
|
|
|
#[test]
|
|
fn test_form_validation_context() {
|
|
use ErrorKind::*;
|
|
|
|
fn count<'a, K>(c: &Errors<'_>, n: &str, kind: K, fuzz: bool) -> usize
|
|
where K: Into<Option<ErrorKind<'a>>>
|
|
{
|
|
let kind = kind.into();
|
|
c.iter().filter(|e| {
|
|
let matches = (fuzz && e.is_for(n)) || (!fuzz && e.is_for_exactly(n));
|
|
let kinded = kind.as_ref().map(|k| k == &e.kind).unwrap_or(true);
|
|
matches && kinded
|
|
}).count()
|
|
}
|
|
|
|
fn fuzzy<'a, K>(c: &Errors<'_>, n: &str, kind: K) -> usize
|
|
where K: Into<Option<ErrorKind<'a>>>
|
|
{
|
|
count(c, n, kind, true)
|
|
}
|
|
|
|
fn exact<'a, K>(c: &Errors<'_>, n: &str, kind: K) -> usize
|
|
where K: Into<Option<ErrorKind<'a>>>
|
|
{
|
|
count(c, n, kind, false)
|
|
}
|
|
|
|
let c = errors::<Cat>("name=littlebobby");
|
|
assert_eq!(exact(&c, "nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "nick", None), 1);
|
|
|
|
let c = errors::<Person>("cats[0].name=Bob");
|
|
assert_eq!(exact(&c, "kitty", None), 1);
|
|
assert_eq!(exact(&c, "kitty", Missing), 1);
|
|
assert_eq!(exact(&c, "cats[0].nick", None), 1);
|
|
assert_eq!(exact(&c, "cats[0].nick", Missing), 1);
|
|
assert_eq!(exact(&c, "dog", None), 1);
|
|
assert_eq!(exact(&c, "dog", Missing), 1);
|
|
assert_eq!(exact(&c, "dog.name", None), 0);
|
|
assert_eq!(exact(&c, "kitty.name", None), 0);
|
|
assert_eq!(exact(&c, "kitty.nick", None), 0);
|
|
|
|
assert_eq!(fuzzy(&c, "kitty", None), 1);
|
|
assert_eq!(fuzzy(&c, "kitty.name", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "kitty.nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "cats[0].nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "dog.name", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "dog", None), 1);
|
|
|
|
let c = errors::<Person>("cats[0].name=Bob&cats[0].nick=kit&kitty.name=Hi");
|
|
assert_eq!(exact(&c, "kitty.nick", Missing), 1);
|
|
assert_eq!(exact(&c, "kitty", None), 0);
|
|
assert_eq!(exact(&c, "dog", Missing), 1);
|
|
assert_eq!(exact(&c, "dog", None), 1);
|
|
assert_eq!(exact(&c, "cats[0].name", None), 1);
|
|
assert_eq!(exact(&c, "cats[0].name", InvalidLength { min: Some(5), max: None }), 1);
|
|
assert_eq!(exact(&c, "cats[0].nick", None), 1);
|
|
assert_eq!(exact(&c, "cats[0].nick", Validation("must start with \"kitty\"".into())), 1);
|
|
|
|
assert_eq!(fuzzy(&c, "kitty.nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "kitty.nick", None), 1);
|
|
assert_eq!(fuzzy(&c, "kitty", None), 0);
|
|
assert_eq!(fuzzy(&c, "dog.name", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "dog", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "cats[0].nick", None), 1);
|
|
assert_eq!(exact(&c, "cats[0].name", None), 1);
|
|
|
|
let c = errors::<Person>("kitty.name=Michael");
|
|
assert_eq!(exact(&c, "kitty.nick", Missing), 1);
|
|
assert_eq!(exact(&c, "dog", Missing), 1);
|
|
assert_eq!(exact(&c, "cats[0].name", None), 0);
|
|
assert_eq!(exact(&c, "cats[0].nick", None), 0);
|
|
|
|
assert_eq!(exact(&c, "cats", None), 1);
|
|
assert_eq!(exact(&c, "cats", InvalidLength { min: Some(1), max: None }), 1);
|
|
|
|
assert_eq!(fuzzy(&c, "kitty.nick", Missing), 1);
|
|
assert_eq!(fuzzy(&c, "kitty.nick", None), 1);
|
|
assert_eq!(fuzzy(&c, "dog", None), 1);
|
|
assert_eq!(fuzzy(&c, "dog.name", Missing), 1);
|
|
assert_eq!(exact(&c, "cats[0].name", None), 0);
|
|
assert_eq!(exact(&c, "cats[0].nick", None), 0);
|
|
|
|
let c = errors::<Person>("kitty.name=Michael&kitty.nick=kittykat&dog.name=woofy");
|
|
assert_eq!(c.iter().count(), 1);
|
|
assert_eq!(exact(&c, "cats", None), 1);
|
|
assert_eq!(exact(&c, "cats", InvalidLength { min: Some(1), max: None }), 1);
|
|
assert_eq!(fuzzy(&c, "cats[0].name", None), 1);
|
|
}
|
|
|
|
// #[derive(Debug, FromForm)]
|
|
// struct Person<'v> {
|
|
// kitty: Cat<'v>,
|
|
// #[field(validate = len(1..))]
|
|
// cats: Vec<Cat<'v>>,
|
|
// dog: Dog<'v>,
|
|
// }
|