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

extern crate rand;
extern crate spectral;

use rand::{thread_rng, sample};
use spectral::prelude::*;

mod names;

/// Get all character names
///
/// # Example
///
/// ```rust,ignore
/// let all_names = starwars-names::all();
///
/// assert_that(&all_names).has_length(93);
/// ```
pub fn all() -> Vec<&'static str> {
    names::get_names()
}

/// Get radom names
///
/// # Example
///
/// ```rust,ignore
/// let num = 3 as usize;
/// let random_names = starwars-names::random(num);
///
/// assert_that(&random_names).has_length(num);
/// ```
pub fn random(n: usize) -> Vec<&'static str> {
    let mut rng = thread_rng();
    let all_names = names::get_names();

    sample(&mut rng, all_names, n)
}

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

    #[test]
    fn get_all() {
        let all_names = all();

        assert_that(&all_names).has_length(93);
    }

    #[test]
    fn get_random() {
        let num = 3 as usize;
        let random_names = random(num);

        assert_that(&random_names).has_length(num);
        
        let all_names = all();

        for name in &random_names {
            assert_that(&all_names).contains(name);
        }
    }
}