anchor-coverage 0.3.1

A wrapper around `anchor test` for computing test coverage
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

// smoelius: `VarGuard` is based on a struct of the same name from Clippy's source code. The `set`
// method was modified to accept an `Option` so that variables could be cleared. Also, `unsafe`
// annotations were added to make the struct compile with Rust Edition 2024. See:
// https://github.com/rust-lang/rust-clippy/blob/c51472b4b09d22bdbb46027f08be54c4b285a725/tests/compile-test.rs#L267-L289

use std::{
    env::{remove_var, set_var, var_os},
    ffi::{OsStr, OsString},
    marker::PhantomData,
};

/// Restores an env var on drop
#[must_use]
pub struct VarGuard<T = OsString> {
    key: &'static str,
    value: Option<OsString>,
    _phantom: PhantomData<T>,
}

impl<T: AsRef<OsStr>> VarGuard<T> {
    pub fn set(key: &'static str, val: Option<T>) -> Self {
        let value = var_os(key);
        if let Some(val) = val {
            unsafe { set_var(key, val) };
        } else {
            unsafe { remove_var(key) };
        }
        Self {
            key,
            value,
            _phantom: PhantomData,
        }
    }
}

impl<T> Drop for VarGuard<T> {
    fn drop(&mut self) {
        match self.value.as_deref() {
            None => unsafe { remove_var(self.key) },
            Some(value) => unsafe { set_var(self.key, value) },
        }
    }
}