mirror of
https://github.com/salsa-rs/salsa.git
synced 2025-02-02 09:46:06 +00:00
add accumulation test
I can't seem to get it to perform incorrectly lol, what am I missing.
This commit is contained in:
parent
c24ab8ffc1
commit
2a9f0f7140
1 changed files with 180 additions and 0 deletions
180
salsa-2022-tests/tests/accumulate-and-reuse.rs
Normal file
180
salsa-2022-tests/tests/accumulate-and-reuse.rs
Normal file
|
@ -0,0 +1,180 @@
|
|||
//! Basic deletion test:
|
||||
//!
|
||||
//! * entities not created in a revision are deleted, as is any memoized data keyed on them.
|
||||
|
||||
use salsa::DebugWithDb;
|
||||
use salsa_2022_tests::{HasLogger, Logger};
|
||||
|
||||
use expect_test::expect;
|
||||
use test_log::test;
|
||||
|
||||
#[salsa::jar(db = Db)]
|
||||
struct Jar(MyInput, Logs, push_logs, push_a_logs, push_b_logs);
|
||||
|
||||
trait Db: salsa::DbWithJar<Jar> + HasLogger {}
|
||||
|
||||
#[salsa::input]
|
||||
struct MyInput {
|
||||
field_a: u32,
|
||||
field_b: u32,
|
||||
}
|
||||
|
||||
#[salsa::accumulator]
|
||||
struct Logs(String);
|
||||
|
||||
#[salsa::tracked]
|
||||
#[allow(dead_code)]
|
||||
fn push_logs(db: &dyn Db, input: MyInput) {
|
||||
db.push_log(format!(
|
||||
"push_logs(a = {}, b = {})",
|
||||
input.field_a(db),
|
||||
input.field_b(db)
|
||||
));
|
||||
|
||||
// We don't invoke `push_a_logs` (or `push_b_logs`) with a value of 1 or less.
|
||||
// This allows us to test what happens a change in inputs causes a function not to be called at all.
|
||||
if input.field_a(db) > 1 {
|
||||
push_a_logs(db, input);
|
||||
}
|
||||
|
||||
if input.field_b(db) > 1 {
|
||||
push_b_logs(db, input);
|
||||
}
|
||||
}
|
||||
|
||||
#[salsa::tracked]
|
||||
fn push_a_logs(db: &dyn Db, input: MyInput) {
|
||||
let field_a = input.field_a(db);
|
||||
db.push_log(format!("push_a_logs({})", field_a));
|
||||
|
||||
for i in 0..field_a {
|
||||
Logs::push(db, format!("log_a({} of {})", i, field_a));
|
||||
}
|
||||
}
|
||||
|
||||
#[salsa::tracked]
|
||||
fn push_b_logs(db: &dyn Db, input: MyInput) {
|
||||
let field_a = input.field_b(db);
|
||||
db.push_log(format!("push_b_logs({})", field_a));
|
||||
|
||||
for i in 0..field_a {
|
||||
Logs::push(db, format!("log_b({} of {})", i, field_a));
|
||||
}
|
||||
}
|
||||
|
||||
#[salsa::db(Jar)]
|
||||
#[derive(Default)]
|
||||
struct Database {
|
||||
storage: salsa::Storage<Self>,
|
||||
logger: Logger,
|
||||
}
|
||||
|
||||
impl salsa::Database for Database {
|
||||
fn salsa_event(&self, event: salsa::Event) {}
|
||||
|
||||
fn salsa_runtime(&self) -> &salsa::Runtime {
|
||||
self.storage.runtime()
|
||||
}
|
||||
}
|
||||
|
||||
impl Db for Database {}
|
||||
|
||||
impl HasLogger for Database {
|
||||
fn logger(&self) -> &Logger {
|
||||
&self.logger
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn accumulate_once() {
|
||||
let mut db = Database::default();
|
||||
|
||||
// Just call accumulate on a base input to see what happens.
|
||||
let input = MyInput::new(&mut db, 2, 3);
|
||||
let logs = push_logs::accumulated::<Logs>(&db, input);
|
||||
expect![[r#"
|
||||
[
|
||||
"log_b(0 of 3)",
|
||||
"log_b(1 of 3)",
|
||||
"log_b(2 of 3)",
|
||||
"log_a(0 of 2)",
|
||||
"log_a(1 of 2)",
|
||||
]"#]]
|
||||
.assert_eq(&format!("{:#?}", logs));
|
||||
db.assert_logs(expect![[r#"
|
||||
[
|
||||
"push_logs(a = 2, b = 3)",
|
||||
"push_a_logs(2)",
|
||||
"push_b_logs(3)",
|
||||
]"#]])
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn change_a_and_reaccumulate() {
|
||||
let mut db = Database::default();
|
||||
|
||||
// Accumulate logs for `a = 2` and `b = 3`
|
||||
let input = MyInput::new(&mut db, 2, 3);
|
||||
let logs = push_logs::accumulated::<Logs>(&db, input);
|
||||
expect![[r#"
|
||||
[
|
||||
"log_b(0 of 3)",
|
||||
"log_b(1 of 3)",
|
||||
"log_b(2 of 3)",
|
||||
"log_a(0 of 2)",
|
||||
"log_a(1 of 2)",
|
||||
]"#]]
|
||||
.assert_eq(&format!("{:#?}", logs));
|
||||
db.assert_logs(expect![[r#"
|
||||
[
|
||||
"push_logs(a = 2, b = 3)",
|
||||
"push_a_logs(2)",
|
||||
"push_b_logs(3)",
|
||||
]"#]]);
|
||||
|
||||
// Change to `a = 1`, which means `push_logs` does not call `push_a_logs` at all
|
||||
input.set_field_a(&mut db, 1);
|
||||
let logs = push_logs::accumulated::<Logs>(&db, input);
|
||||
expect![[r#"
|
||||
[
|
||||
"log_b(0 of 3)",
|
||||
"log_b(1 of 3)",
|
||||
"log_b(2 of 3)",
|
||||
]"#]]
|
||||
.assert_eq(&format!("{:#?}", logs));
|
||||
db.assert_logs(expect![[r#"
|
||||
[
|
||||
"push_logs(a = 1, b = 3)",
|
||||
]"#]]);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn get_a_logs_after_changing_b() {
|
||||
let mut db = Database::default();
|
||||
|
||||
// Invoke `push_a_logs` with `a = 2` and `b = 3` (but `b` doesn't matter)
|
||||
let input = MyInput::new(&mut db, 2, 3);
|
||||
let logs = push_a_logs::accumulated::<Logs>(&db, input);
|
||||
expect![[r#"
|
||||
[
|
||||
"log_a(0 of 2)",
|
||||
"log_a(1 of 2)",
|
||||
]"#]]
|
||||
.assert_eq(&format!("{:#?}", logs));
|
||||
db.assert_logs(expect![[r#"
|
||||
[
|
||||
"push_a_logs(2)",
|
||||
]"#]]);
|
||||
|
||||
// Changing `b` does not cause `push_a_logs` to re-execute
|
||||
// and we still get the same result
|
||||
input.set_field_b(&mut db, 5);
|
||||
let logs = push_a_logs::accumulated::<Logs>(&db, input);
|
||||
expect![[r#"
|
||||
[
|
||||
"log_a(0 of 2)",
|
||||
"log_a(1 of 2)",
|
||||
]"#]]
|
||||
.assert_eq(&format!("{:#?}", logs));
|
||||
db.assert_logs(expect!["[]"]);
|
||||
}
|
Loading…
Reference in a new issue