hanger 0.1.0

Attribute macro to put hooks around a function.
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

use hanger::hook;
use std::sync::atomic::{AtomicUsize, Ordering};

#[test]
fn test_hooks() {
    static COUNTER: AtomicUsize = AtomicUsize::new(0);
    fn add_one() {
        COUNTER.fetch_add(1, Ordering::Relaxed);
    }
    fn sub_one() {
        COUNTER.fetch_sub(1, Ordering::Relaxed);
    }

    #[hook(add_one)]
    fn entry_add_one() {
        COUNTER.fetch_add(1, Ordering::Relaxed);
    }
    entry_add_one();
    assert_eq!(COUNTER.load(Ordering::Relaxed), 2);

    #[hook(before = add_one, after = sub_one)]
    fn entry_add_sub() {
        COUNTER.fetch_add(1, Ordering::Relaxed);
    }
    entry_add_sub();
    assert_eq!(COUNTER.load(Ordering::Relaxed), 3);
}

#[test]
fn test_path_hooks() {
    mod path {
        use super::*;
        pub static COUNTER: AtomicUsize = AtomicUsize::new(0);

        pub fn add_one() {
            COUNTER.fetch_add(1, Ordering::Relaxed);
        }
        pub fn sub_one() {
            COUNTER.fetch_sub(1, Ordering::Relaxed);
        }
    }

    #[hook(path::add_one)]
    fn entry_add_one() {
        path::COUNTER.fetch_add(1, Ordering::Relaxed);
    }
    entry_add_one();
    assert_eq!(path::COUNTER.load(Ordering::Relaxed), 2);

    #[hook(before = path::add_one, after = path::sub_one)]
    fn entry_add_sub() {
        path::COUNTER.fetch_add(1, Ordering::Relaxed);
    }
    entry_add_sub();
    assert_eq!(path::COUNTER.load(Ordering::Relaxed), 3);
}

mod tests {
    use super::*;
    static G_COUNTER: AtomicUsize = AtomicUsize::new(0);
    fn add_g() {
        G_COUNTER.fetch_add(1, Ordering::Relaxed);
    }

    #[test]
    #[hook(add_g)]
    fn test_hook_tests() {
        assert_eq!(G_COUNTER.load(Ordering::Relaxed), 1);
    }
}