mirror of
https://github.com/salsa-rs/salsa.git
synced 2024-10-24 05:09:52 +00:00
49 lines
1 KiB
Rust
49 lines
1 KiB
Rust
//! Test that transparent (uncached) queries work
|
|
|
|
#[salsa::query_group(QueryGroupStorage)]
|
|
trait QueryGroup {
|
|
#[salsa::input]
|
|
fn input(&self, c: char) -> u32;
|
|
|
|
fn increment(&self, c: char) -> u32;
|
|
}
|
|
|
|
fn increment(db: &dyn QueryGroup, c: char) -> u32 {
|
|
db.input(c) + 1
|
|
}
|
|
|
|
#[salsa::database(QueryGroupStorage)]
|
|
#[derive(Default)]
|
|
struct Database {
|
|
storage: salsa::Storage<Self>,
|
|
}
|
|
|
|
impl salsa::Database for Database {}
|
|
|
|
#[test]
|
|
fn remove_input_from_cached_query() {
|
|
let mut db = Database::default();
|
|
|
|
db.set_input('a', 22);
|
|
db.set_input('b', 44);
|
|
assert_eq!(db.increment('a'), 23);
|
|
assert_eq!(db.increment('b'), 45);
|
|
|
|
db.remove_input('a');
|
|
assert_eq!(db.increment('b'), 45);
|
|
}
|
|
|
|
#[test]
|
|
fn remove_and_restore_input_from_cached_query() {
|
|
let mut db = Database::default();
|
|
|
|
db.set_input('a', 22);
|
|
db.set_input('b', 44);
|
|
assert_eq!(db.increment('a'), 23);
|
|
assert_eq!(db.increment('b'), 45);
|
|
|
|
db.remove_input('a');
|
|
db.set_input('a', 66);
|
|
assert_eq!(db.increment('a'), 67);
|
|
}
|