forked from salsa-rs/salsa
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tracked_method.rs
42 lines (36 loc) · 967 Bytes
/
tracked_method.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
//! Test that a `tracked` fn on a `salsa::input`
//! compiles and executes successfully.
#![allow(warnings)]
trait TrackedTrait {
fn tracked_trait_fn(self, db: &dyn salsa::Database) -> u32;
}
#[salsa::input]
struct MyInput {
field: u32,
}
#[salsa::tracked]
impl MyInput {
#[salsa::tracked]
fn tracked_fn(self, db: &dyn salsa::Database) -> u32 {
self.field(db) * 2
}
#[salsa::tracked(return_ref)]
fn tracked_fn_ref(self, db: &dyn salsa::Database) -> u32 {
self.field(db) * 3
}
}
#[salsa::tracked]
impl TrackedTrait for MyInput {
#[salsa::tracked]
fn tracked_trait_fn(self, db: &dyn salsa::Database) -> u32 {
self.field(db) * 4
}
}
#[test]
fn execute() {
let mut db = salsa::DatabaseImpl::new();
let object = MyInput::new(&mut db, 22);
// assert_eq!(object.tracked_fn(&db), 44);
// assert_eq!(*object.tracked_fn_ref(&db), 66);
assert_eq!(object.tracked_trait_fn(&db), 88);
}