rustdns 0.4.0

A DNS parsing library
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

// TODO Switch this to use datatest after 0.6.3 (which is broken):
// https://github.com/commure/datatest/pull/30
// and custom_test_frameworks is supported https://github.com/rust-lang/rust/issues/50297
use pretty_assertions::assert_eq;
use regex::Regex;
use rustdns::Message;
use serde::Deserialize;
use std::fs;

const TEST_DATA_FILENAME: &str = "tests/test_data.yaml";

#[derive(Deserialize)]
struct TestCase {
    // Name of the test case.
    name: String,

    // Hex encoded binary string.
    // TODO Change this to a binary type, when serde_yaml supports it: https://github.com/dtolnay/serde-yaml/issues/91
    binary: String,

    // Dig-ish formatted output.
    // TODO Change this to a multi-line string type, for easier viewing in the generated YAML.
    string: String,
}

#[test]
fn tests() {
    let s = fs::read(TEST_DATA_FILENAME).expect("failed read test input");
    let tests: Vec<TestCase> =
        serde_yaml::from_slice(&s).expect("failed to deserialise test input");

    for case in tests {
        test_from_slice(case);
    }
}

fn normalise_whitespace(s: &str) -> String {
    let re = Regex::new(r"[ ]+").unwrap();
    return re.replace_all(s, " ").to_string();
}

fn test_from_slice(case: TestCase) {
    let input = match hex::decode(case.binary) {
        Err(e) => panic!("{}: Invalid test case input: {}", case.name, e),
        Ok(i) => i,
    };
    let m = match Message::from_slice(&input) {
        Err(e) => panic!("{}: Unable to parse: {}", case.name, e),
        Ok(p) => p,
    };

    // TODO Split this into a few tests. from_slice(), fmt(), to_vec()

    // Normalise the formatted output a little (to allow little whitespace changes).
    let got = normalise_whitespace(&format!("{}", m));
    let want = normalise_whitespace(&case.string);

    assert_eq!(got, want, "{}: Formatted string doesn't match", case.name);

    // TODO Test writing the result back out.
}