pharos-plugins 0.1.0

CoreDNS-compatible Plugins for the Pharos DNS server
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

#![forbid(unsafe_code)]
#![cfg(feature = "plugin-loadbalance")]
#![recursion_limit = "512"]

use pharos_corefile::{configure, parse};
use pharos_plugins::{Chain, directives};

fn accepts(directive: &str) -> bool {
    let text = format!(".:53 {{\n{directive}\n}}");
    parse("Corefile", &text)
        .and_then(|blocks| configure::<Chain>(&blocks, directives()))
        .is_ok()
}

#[test]
fn accepts_coredns_loadbalance_setup_cases() {
    for text in [
        "loadbalance",
        "loadbalance round_robin",
        "loadbalance weighted wfile",
        "loadbalance weighted wf {\n reload 10s\n}",
        "loadbalance weighted wf {\n reload 0s\n}",
        "loadbalance round_robin\n loadbalance weighted wf",
    ] {
        assert!(accepts(text), "rejected {text}");
    }
}

#[test]
fn rejects_coredns_invalid_loadbalance_setup_cases() {
    for text in [
        "loadbalance fleeb",
        "loadbalance round_robin a",
        "loadbalance weighted",
        "loadbalance weighted a b",
        "loadbalance weighted wfile {\n susu\n}",
        "loadbalance weighted wfile {\n reload a\n}",
        "loadbalance weighted wfile {\n reload 30s a\n}",
        "loadbalance\n loadbalance fleeb",
    ] {
        assert!(!accepts(text), "accepted {text}");
    }
}