Add a more complex case

This commit is contained in:
Phoebe Szmucer 2024-07-22 20:12:54 +01:00
parent 49a147a627
commit a85ac260d3
3 changed files with 111 additions and 1 deletions

View file

@ -1,3 +1,6 @@
//! Test that when having nested tracked functions
//! we don't drop any values when accumulating.
mod common;
use expect_test::expect;

View file

@ -1,3 +1,6 @@
//! Demonstrates that accumulation is done in the order
//! in which things were originally executed.
mod common;
use expect_test::expect;
@ -37,7 +40,7 @@ fn push_d_logs(db: &dyn Database) {
}
#[test]
fn accumulate_chain() {
fn accumulate_execution_order() {
salsa::default_database().attach(|db| {
let logs = push_logs::accumulated::<Log>(db);
// Check that we get logs in execution order

View file

@ -0,0 +1,104 @@
//! Test that we don't get duplicate accumulated values
mod common;
use expect_test::expect;
use salsa::{Accumulator, Database};
use test_log::test;
// A(1) {
// B
// B
// C {
// D {
// A(2) {
// B
// }
// B
// }
// E
// }
// B
// }
#[salsa::accumulator]
struct Log(#[allow(dead_code)] String);
#[salsa::input]
struct MyInput {
n: u32,
}
#[salsa::tracked]
fn push_logs(db: &dyn Database) {
push_a_logs(db, MyInput::new(db, 1));
}
#[salsa::tracked]
fn push_a_logs(db: &dyn Database, input: MyInput) {
Log("log a".to_string()).accumulate(db);
if input.n(db) == 1 {
push_b_logs(db);
push_b_logs(db);
push_c_logs(db);
push_b_logs(db);
} else {
push_b_logs(db);
}
}
#[salsa::tracked]
fn push_b_logs(db: &dyn Database) {
Log("log b".to_string()).accumulate(db);
}
#[salsa::tracked]
fn push_c_logs(db: &dyn Database) {
Log("log c".to_string()).accumulate(db);
push_d_logs(db);
push_e_logs(db);
}
// Note this isn't tracked
fn push_d_logs(db: &dyn Database) {
Log("log d".to_string()).accumulate(db);
push_a_logs(db, MyInput::new(db, 2));
push_b_logs(db);
}
#[salsa::tracked]
fn push_e_logs(db: &dyn Database) {
Log("log e".to_string()).accumulate(db);
}
#[test]
fn accumulate_no_duplicates() {
salsa::default_database().attach(|db| {
let logs = push_logs::accumulated::<Log>(db);
// Test that there aren't duplicate B logs.
// Note that log A appears twice, because they both come
// from different inputs.
expect![[r#"
[
Log(
"log a",
),
Log(
"log b",
),
Log(
"log c",
),
Log(
"log d",
),
Log(
"log a",
),
Log(
"log e",
),
]"#]]
.assert_eq(&format!("{:#?}", logs));
})
}