mirror of
https://github.com/salsa-rs/salsa.git
synced 2025-01-23 05:07:27 +00:00
2df88d2c33
This changes tracked methods from being annotated with `#[tracked]` to `#[salsa::tracked]`. This improves consistency and allows outputting a nicer error message if someone forgets to put the attribute on the impl block.
44 lines
926 B
Rust
44 lines
926 B
Rust
//! Test that a `tracked` fn on a `salsa::input`
|
|
//! compiles and executes successfully.
|
|
#![allow(warnings)]
|
|
|
|
#[salsa::jar(db = Db)]
|
|
struct Jar(MyInput, MyInput_tracked_fn, MyInput_tracked_fn_ref);
|
|
|
|
trait Db: salsa::DbWithJar<Jar> {}
|
|
|
|
#[salsa::input]
|
|
struct MyInput {
|
|
field: u32,
|
|
}
|
|
|
|
#[salsa::tracked]
|
|
impl MyInput {
|
|
#[salsa::tracked]
|
|
fn tracked_fn(self, db: &dyn Db) -> u32 {
|
|
self.field(db) * 2
|
|
}
|
|
|
|
#[salsa::tracked(return_ref)]
|
|
fn tracked_fn_ref(self, db: &dyn Db) -> u32 {
|
|
self.field(db) * 3
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn execute() {
|
|
#[salsa::db(Jar)]
|
|
#[derive(Default)]
|
|
struct Database {
|
|
storage: salsa::Storage<Self>,
|
|
}
|
|
|
|
impl salsa::Database for Database {}
|
|
|
|
impl Db for Database {}
|
|
|
|
let mut db = Database::default();
|
|
let object = MyInput::new(&mut db, 22);
|
|
assert_eq!(object.tracked_fn(&db), 44);
|
|
assert_eq!(*object.tracked_fn_ref(&db), 66);
|
|
}
|