bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/query-group-macro/tests/lru.rs')
-rw-r--r--crates/query-group-macro/tests/lru.rs68
1 files changed, 68 insertions, 0 deletions
diff --git a/crates/query-group-macro/tests/lru.rs b/crates/query-group-macro/tests/lru.rs
new file mode 100644
index 0000000000..f56dd5c2f9
--- /dev/null
+++ b/crates/query-group-macro/tests/lru.rs
@@ -0,0 +1,68 @@
+use expect_test::expect;
+
+mod logger_db;
+use logger_db::LoggerDb;
+use query_group_macro::query_group;
+
+#[query_group]
+pub trait LruDB: salsa::Database {
+ // // input with no params
+ #[salsa::input]
+ fn input_string(&self) -> String;
+
+ #[salsa::lru(16)]
+ #[salsa::invoke_interned(length_query)]
+ fn length_query(&self, key: ()) -> usize;
+
+ #[salsa::lru(16)]
+ #[salsa::invoke_interned(invoked_query)]
+ fn length_query_invoke(&self, key: ()) -> usize;
+}
+
+fn length_query(db: &dyn LruDB, _key: ()) -> usize {
+ db.input_string().len()
+}
+
+fn invoked_query(db: &dyn LruDB, _key: ()) -> usize {
+ db.input_string().len()
+}
+
+#[test]
+fn plain_lru() {
+ let mut db = LoggerDb::default();
+
+ db.set_input_string(String::from("Hello, world!"));
+ let len = db.length_query(());
+
+ assert_eq!(len, 13);
+ db.assert_logs(expect![[r#"
+ [
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(WillExecute { database_key: create_data_LruDB(Id(0)) })",
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(DidValidateMemoizedValue { database_key: create_data_LruDB(Id(0)) })",
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(WillExecute { database_key: length_query_shim(Id(800)) })",
+ "salsa_event(WillCheckCancellation)",
+ ]"#]]);
+}
+
+#[test]
+fn invoke_lru() {
+ let mut db = LoggerDb::default();
+
+ db.set_input_string(String::from("Hello, world!"));
+ let len = db.length_query_invoke(());
+
+ assert_eq!(len, 13);
+ db.assert_logs(expect![[r#"
+ [
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(WillExecute { database_key: create_data_LruDB(Id(0)) })",
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(DidValidateMemoizedValue { database_key: create_data_LruDB(Id(0)) })",
+ "salsa_event(WillCheckCancellation)",
+ "salsa_event(WillExecute { database_key: length_query_invoke_shim(Id(800)) })",
+ "salsa_event(WillCheckCancellation)",
+ ]"#]]);
+}