salsa/tests/specify-only-works-if-the-key-is-created-in-the-current-query.rs

58 lines
1.3 KiB
Rust
Raw Normal View History

//! Test that `specify` only works if the key is a tracked struct created in the current query.
//! compilation succeeds but execution panics
#![allow(warnings)]
2024-07-17 10:54:24 +00:00
#[salsa::db]
trait Db: salsa::Database {}
2024-07-16 10:04:01 +00:00
#[salsa::input]
struct MyInput {
field: u32,
}
2024-07-16 10:04:01 +00:00
#[salsa::tracked]
2024-05-24 01:16:30 +00:00
struct MyTracked<'db> {
field: u32,
}
2024-07-16 10:04:01 +00:00
#[salsa::tracked]
2024-05-24 01:16:30 +00:00
fn tracked_struct_created_in_another_query<'db>(db: &'db dyn Db, input: MyInput) -> MyTracked<'db> {
MyTracked::new(db, input.field(db) * 2)
}
2024-07-16 10:04:01 +00:00
#[salsa::tracked]
2024-05-24 01:16:30 +00:00
fn tracked_fn<'db>(db: &'db dyn Db, input: MyInput) -> MyTracked<'db> {
let t = tracked_struct_created_in_another_query(db, input);
if input.field(db) != 0 {
tracked_fn_extra::specify(db, t, 2222);
}
t
}
2024-07-17 10:54:24 +00:00
#[salsa::tracked(specify)]
2024-05-24 01:16:30 +00:00
fn tracked_fn_extra<'db>(_db: &'db dyn Db, _input: MyTracked<'db>) -> u32 {
0
}
2024-07-17 10:54:24 +00:00
#[salsa::db]
#[derive(Default)]
struct Database {
storage: salsa::Storage<Self>,
}
2024-07-17 10:54:24 +00:00
#[salsa::db]
impl salsa::Database for Database {}
2024-07-17 10:54:24 +00:00
#[salsa::db]
impl Db for Database {}
#[test]
2024-07-17 10:54:24 +00:00
#[should_panic(
expected = "can only use `specify` on salsa structs created during the current tracked fn"
)]
fn execute_when_specified() {
let mut db = Database::default();
let input = MyInput::new(&db, 22);
let tracked = tracked_fn(&db, input);
}