Skip to content

Commit

Permalink
Add a more complex case
Browse files Browse the repository at this point in the history
  • Loading branch information
PhoebeSzmucer committed Jul 22, 2024
1 parent 49a147a commit a85ac26
Show file tree
Hide file tree
Showing 3 changed files with 111 additions and 1 deletion.
3 changes: 3 additions & 0 deletions tests/accumulate-chain.rs
Original file line number Diff line number Diff line change
@@ -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;
Expand Down
5 changes: 4 additions & 1 deletion tests/accumulate-execution-order.rs
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
//! Demonstrates that accumulation is done in the order
//! in which things were originally executed.
mod common;

use expect_test::expect;
Expand Down Expand Up @@ -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
Expand Down
104 changes: 104 additions & 0 deletions tests/accumulate-no-duplicates.rs
Original file line number Diff line number Diff line change
@@ -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));
})
}

0 comments on commit a85ac26

Please sign in to comment.