aboutsummaryrefslogtreecommitdiff
path: root/rand/rand_os/tests/mod.rs
blob: 2130e169bbfc17db269339cd6d353e65e1488270 (plain)
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
71
72
73
74
75
76
77
78
79
80
extern crate rand_os;

use rand_os::rand_core::RngCore;
use rand_os::OsRng;

#[test]
fn test_os_rng() {
    let mut r = OsRng::new().unwrap();

    r.next_u32();
    r.next_u64();

    let mut v1 = [0u8; 1000];
    r.fill_bytes(&mut v1);

    let mut v2 = [0u8; 1000];
    r.fill_bytes(&mut v2);

    let mut n_diff_bits = 0;
    for i in 0..v1.len() {
        n_diff_bits += (v1[i] ^ v2[i]).count_ones();
    }

    // Check at least 1 bit per byte differs. p(failure) < 1e-1000 with random input.
    assert!(n_diff_bits >= v1.len() as u32);
}

#[test]
fn test_os_rng_empty() {
    let mut r = OsRng::new().unwrap();

    let mut empty = [0u8; 0];
    r.fill_bytes(&mut empty);
}

#[test]
fn test_os_rng_huge() {
    let mut r = OsRng::new().unwrap();

    let mut huge = [0u8; 100_000];
    r.fill_bytes(&mut huge);
}

#[cfg(not(any(target_arch = "wasm32", target_arch = "asmjs")))]
#[test]
fn test_os_rng_tasks() {
    use std::sync::mpsc::channel;
    use std::thread;

    let mut txs = vec!();
    for _ in 0..20 {
        let (tx, rx) = channel();
        txs.push(tx);

        thread::spawn(move|| {
            // wait until all the tasks are ready to go.
            rx.recv().unwrap();

            // deschedule to attempt to interleave things as much
            // as possible (XXX: is this a good test?)
            let mut r = OsRng::new().unwrap();
            thread::yield_now();
            let mut v = [0u8; 1000];

            for _ in 0..100 {
                r.next_u32();
                thread::yield_now();
                r.next_u64();
                thread::yield_now();
                r.fill_bytes(&mut v);
                thread::yield_now();
            }
        });
    }

    // start all the tasks
    for tx in txs.iter() {
        tx.send(()).unwrap();
    }
}