summaryrefslogtreecommitdiff
path: root/schist_queries/src/utils/find_by_id_or.rs
blob: 8335e5c1aeffa184590593556e822abb03ffffd9 (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
pub fn find_by_id_or<Id: PartialEq, Val: Copy>(
    arr: &[(Id, Val)],
    id: Id,
    default: Val,
) -> Val {
    arr.iter()
        .find(|(elem_id, _)| *elem_id == id)
        .map_or(default, |row| row.1)
}

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

    #[test]
    fn when_no_elements_then_return_default() {
        let arr = [];

        let result = find_by_id_or(&arr, 1, 42);

        assert_eq!(result, 42);
    }

    #[test]
    fn when_id_not_in_elements_then_return_default() {
        let arr = [(0, 100), (2, 200)];

        let result = find_by_id_or(&arr, 1, 42);

        assert_eq!(result, 42);
    }

    #[test]
    fn when_id_in_elements_then_return_value() {
        let arr = [(1, 67)];

        let result = find_by_id_or(&arr, 1, 42);

        assert_eq!(result, 67);
    }
}