use crate::{
prelude::*,
tests::{creators::*, data::*, types::*},
};
#[test]
fn remove_multiple_keys() {
let map = map_alice_bob_chuck(1, 2, 3);
let tx = map
.prepared_tx(&RemoveMultiple::SCHEMA)
.remove(RemoveMultiple::a, |entry, _params, state| {
state.user.push(entry.map(|e| e.0))
})
.remove(RemoveMultiple::b, |entry, _params, state| {
state.user.push(entry.map(|e| e.0))
})
.into_transaction();
assert_eq!(
tx.execute(
RemoveMultipleKeys {
a: ALICE.into(),
b: BOB.into(),
},
RemoveMultipleParams {},
),
TxResult::Completed(RemoveMultipleState {
user: vec![Some(ALICE.into()), Some(BOB.into())],
})
);
assert_eq!(map.len(), 1);
}
#[test]
fn remove_if() {
let map = map_alice_bob(1, 2);
let tx = map
.prepared_tx(&RemoveMultiple::SCHEMA)
.remove_if(RemoveMultiple::a, |k, v, _p, s| {
let cond = *v >= 2;
if cond {
s.user.push(Some(k.clone()));
} else {
s.user.push(None);
}
cond
})
.remove_if(RemoveMultiple::b, |k, v, _p, s| {
let cond = *v >= 2;
if cond {
s.user.push(Some(k.clone()));
} else {
s.user.push(None);
}
cond
})
.into_transaction();
assert_eq!(
tx.execute(
RemoveMultipleKeys {
a: ALICE.into(),
b: BOB.into()
},
RemoveMultipleParams {}
),
TxResult::Completed(RemoveMultipleState {
user: vec![None, Some(BOB.into())]
})
);
assert_eq!(map.len(), 1);
}