salsa/tests/cycles.rs
Niko Matsakis 238be96432 introduce group storage structs to use from database_storage macro
This also detected a bunch of crate visibility mismatches in the
tests.
2019-01-23 05:35:07 -05:00

63 lines
1.3 KiB
Rust

#[derive(Default)]
struct DatabaseImpl {
runtime: salsa::Runtime<DatabaseImpl>,
}
impl salsa::Database for DatabaseImpl {
fn salsa_runtime(&self) -> &salsa::Runtime<DatabaseImpl> {
&self.runtime
}
}
salsa::database_storage! {
struct DatabaseImplStorage for DatabaseImpl {
impl Database {
fn memoized_a() for MemoizedAQuery;
fn memoized_b() for MemoizedBQuery;
fn volatile_a() for VolatileAQuery;
fn volatile_b() for VolatileBQuery;
}
}
}
#[salsa::query_group]
trait Database: salsa::Database {
// `a` and `b` depend on each other and form a cycle
fn memoized_a(&self) -> ();
fn memoized_b(&self) -> ();
#[salsa::volatile]
fn volatile_a(&self) -> ();
#[salsa::volatile]
fn volatile_b(&self) -> ();
}
fn memoized_a(db: &impl Database) -> () {
db.memoized_b()
}
fn memoized_b(db: &impl Database) -> () {
db.memoized_a()
}
fn volatile_a(db: &impl Database) -> () {
db.volatile_b()
}
fn volatile_b(db: &impl Database) -> () {
db.volatile_a()
}
#[test]
#[should_panic(expected = "cycle detected")]
fn cycle_memoized() {
let query = DatabaseImpl::default();
query.memoized_a();
}
#[test]
#[should_panic(expected = "cycle detected")]
fn cycle_volatile() {
let query = DatabaseImpl::default();
query.volatile_a();
}