salsa/tests/accumulate-no-duplicates.rs
Niko Matsakis daaa78056a switch to new database design
Under this design, *all* databases are a
`DatabaseImpl<U>`, where the `U` implements
`UserData` (you can use `()` if there is none).

Code would default to `&dyn salsa::Database` but
if you want to give access to the userdata, you
can define a custom database trait
`MyDatabase: salsa::Databse` so long as you

* annotate `MyDatabase` trait definition of
  impls of `MyDatabase` with `#[salsa::db]`
* implement `MyDatabase` for `DatabaseImpl<U>`
  where `U` is your userdata (this could be a
  blanket impl, if you don't know the precise
  userdata type).

The `tests/common/mod.rs` shows the pattern.
2024-07-28 12:47:50 +00:00

104 lines
2.1 KiB
Rust

//! 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::DatabaseImpl::new().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));
})
}