shadow-crypt-core 1.0.7

Core types and deterministic operations for shadow-crypt
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

use std::sync::Mutex;

pub struct ProgressCounter {
    current: Mutex<u64>,
    total: u64,
}
impl ProgressCounter {
    pub fn new(total: u64) -> Self {
        Self {
            current: Mutex::new(0),
            total,
        }
    }

    pub fn increment(&self) {
        let mut current = self.current.lock().unwrap_or_else(|e| e.into_inner());
        *current += 1;
    }
    pub fn get_current(&self) -> u64 {
        let current = self.current.lock().unwrap_or_else(|e| e.into_inner());
        *current
    }
    pub fn get_total(&self) -> u64 {
        self.total
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_new() {
        let counter = ProgressCounter::new(100);
        assert_eq!(counter.get_current(), 0);
        assert_eq!(counter.get_total(), 100);
    }

    #[test]
    fn test_increment() {
        let counter = ProgressCounter::new(10);
        assert_eq!(counter.get_current(), 0);
        counter.increment();
        assert_eq!(counter.get_current(), 1);
        counter.increment();
        assert_eq!(counter.get_current(), 2);
    }

    #[test]
    fn test_get_current_and_total() {
        let counter = ProgressCounter::new(50);
        assert_eq!(counter.get_current(), 0);
        assert_eq!(counter.get_total(), 50);
        counter.increment();
        assert_eq!(counter.get_current(), 1);
        assert_eq!(counter.get_total(), 50);
    }
}