fix tests

This commit is contained in:
Quentin 2023-11-29 13:06:32 +01:00
parent de72d6037f
commit f11592926b
Signed by untrusted user: quentin
GPG key ID: E9602264D639FF68

View file

@ -144,16 +144,10 @@ impl ProxyEntry {
let mut redirects = vec![]; let mut redirects = vec![];
for mid in middleware.into_iter() { for mid in middleware.into_iter() {
match mid { match mid {
ConfigTag::AddHeader(k, v) => add_headers.push((k.to_string(), v.to_string())), ConfigTag::AddHeader(k, v) => add_headers.push((k.to_string(), v.clone())),
ConfigTag::AddRedirect(m, r, c) => redirects.push(((*m).clone(), (*r).clone(), *c)), ConfigTag::AddRedirect(m, r, c) => redirects.push(((*m).clone(), (*r).clone(), *c)),
ConfigTag::LocalLb => ConfigTag::LocalLb | ConfigTag::GlobalLb => {
/* handled in parent fx */ /* handled in parent fx */
{
()
}
ConfigTag::GlobalLb =>
/* handled in parent fx */
{
() ()
} }
}; };
@ -251,7 +245,7 @@ enum MatchTag {
#[derive(Debug)] #[derive(Debug)]
enum ConfigTag<'a> { enum ConfigTag<'a> {
AddHeader(&'a str, &'a str), AddHeader(&'a str, String),
AddRedirect(UrlPrefix, UrlPrefix, u32), AddRedirect(UrlPrefix, UrlPrefix, u32),
GlobalLb, GlobalLb,
LocalLb, LocalLb,
@ -280,8 +274,8 @@ fn parse_tricot_tags(tag: &str) -> Option<ParsedTag> {
UrlPrefix::new(raw_prefix) UrlPrefix::new(raw_prefix)
.map(|prefix| ParsedTag::Frontend(MatchTag::HttpWithTls(prefix, priority))) .map(|prefix| ParsedTag::Frontend(MatchTag::HttpWithTls(prefix, priority)))
} }
["tricot-add-header", header_key, header_value] => Some(ParsedTag::Middleware( ["tricot-add-header", header_key, header_values @ ..] => Some(ParsedTag::Middleware(
ConfigTag::AddHeader(header_key, header_value), ConfigTag::AddHeader(header_key, header_values.join(" ")),
)), )),
["tricot-add-redirect", raw_match, raw_replace, maybe_raw_code @ ..] => { ["tricot-add-redirect", raw_match, raw_replace, maybe_raw_code @ ..] => {
let (p_match, p_replace) = let (p_match, p_replace) =
@ -525,8 +519,8 @@ mod tests {
#[test] #[test]
fn test_parse_tricot_add_header_tag() { fn test_parse_tricot_add_header_tag() {
match parse_tricot_add_header_tag("tricot-add-header Content-Security-Policy default-src 'none'; img-src 'self'; script-src 'self'; style-src 'self'") { match parse_tricot_tags("tricot-add-header Content-Security-Policy default-src 'none'; img-src 'self'; script-src 'self'; style-src 'self'") {
Some((name, value)) => { Some(ParsedTag::Middleware(ConfigTag::AddHeader(name, value))) => {
assert_eq!(name, "Content-Security-Policy"); assert_eq!(name, "Content-Security-Policy");
assert_eq!(value, "default-src 'none'; img-src 'self'; script-src 'self'; style-src 'self'"); assert_eq!(value, "default-src 'none'; img-src 'self'; script-src 'self'; style-src 'self'");
} }