Merge two HashMaps adding the values implemented as a function

• HashMap

use std::collections::HashMap;
fn main() {
    let a = HashMap::from([("apple", 1), ("banana", 1)]);
    let b = HashMap::from([("apple", 2), ("peach", 2), ("grape", 2)]);

    let mut total: HashMap<&str, i32> = HashMap::new();

    add(&mut total, &a);
    println!("{:#?}", total);
    assert_eq!(total, a);

    add(&mut total, &b);
    println!("{:#?}", total);

    let expected_total = HashMap::from([("apple", 3), ("peach", 2), ("grape", 2), ("banana", 1)]);
    assert_eq!(total, expected_total);
}

fn add<'a>(total: &mut HashMap<&'a str, i32>, data: &HashMap<&'a str, i32>) {
    for (key, value) in data.iter() {
        *total.entry(key).or_insert(0) += value;
    }
}

{
    "banana": 1,
    "apple": 1,
}
{
    "banana": 1,
    "peach": 2,
    "grape": 2,
    "apple": 3,
}