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
43
44
45
46
47
48
49
50
51
52
|
mod logger_db;
use expect_test::expect;
use logger_db::LoggerDb;
use query_group_macro::query_group;
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct Error;
#[query_group]
pub trait ResultDatabase: salsa::Database {
#[salsa::input]
fn input_string(&self) -> String;
#[salsa::invoke_interned(length)]
fn length(&self, key: ()) -> Result<usize, Error>;
#[salsa::invoke_interned(length2)]
fn length2(&self, key: ()) -> Result<usize, Error>;
}
fn length(db: &dyn ResultDatabase, _key: ()) -> Result<usize, Error> {
Ok(db.input_string().len())
}
fn length2(db: &dyn ResultDatabase, _key: ()) -> Result<usize, Error> {
Ok(db.input_string().len())
}
#[test]
fn test_queries_with_results() {
let mut db = LoggerDb::default();
let input = "hello";
db.set_input_string(input.to_owned());
assert_eq!(db.length(()), Ok(input.len()));
assert_eq!(db.length2(()), Ok(input.len()));
db.assert_logs(expect![[r#"
[
"salsa_event(WillCheckCancellation)",
"salsa_event(WillExecute { database_key: create_data_ResultDatabase(Id(0)) })",
"salsa_event(WillCheckCancellation)",
"salsa_event(DidValidateMemoizedValue { database_key: create_data_ResultDatabase(Id(0)) })",
"salsa_event(WillCheckCancellation)",
"salsa_event(WillExecute { database_key: length_shim(Id(800)) })",
"salsa_event(WillCheckCancellation)",
"salsa_event(WillCheckCancellation)",
"salsa_event(WillCheckCancellation)",
"salsa_event(WillExecute { database_key: length2_shim(Id(c00)) })",
"salsa_event(WillCheckCancellation)",
]"#]]);
}
|
