bendns' repos / rust-analyzer
Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/ra-salsa/tests/incremental/memoized_dep_inputs.rs')
-rw-r--r--crates/ra-salsa/tests/incremental/memoized_dep_inputs.rs60
1 files changed, 60 insertions, 0 deletions
diff --git a/crates/ra-salsa/tests/incremental/memoized_dep_inputs.rs b/crates/ra-salsa/tests/incremental/memoized_dep_inputs.rs
new file mode 100644
index 0000000000..0043bb4574
--- /dev/null
+++ b/crates/ra-salsa/tests/incremental/memoized_dep_inputs.rs
@@ -0,0 +1,60 @@
+use crate::implementation::{TestContext, TestContextImpl};
+
+#[ra_salsa::query_group(MemoizedDepInputs)]
+pub(crate) trait MemoizedDepInputsContext: TestContext {
+ fn dep_memoized2(&self) -> usize;
+ fn dep_memoized1(&self) -> usize;
+ #[ra_salsa::dependencies]
+ fn dep_derived1(&self) -> usize;
+ #[ra_salsa::input]
+ fn dep_input1(&self) -> usize;
+ #[ra_salsa::input]
+ fn dep_input2(&self) -> usize;
+}
+
+fn dep_memoized2(db: &dyn MemoizedDepInputsContext) -> usize {
+ db.log().add("Memoized2 invoked");
+ db.dep_memoized1()
+}
+
+fn dep_memoized1(db: &dyn MemoizedDepInputsContext) -> usize {
+ db.log().add("Memoized1 invoked");
+ db.dep_derived1() * 2
+}
+
+fn dep_derived1(db: &dyn MemoizedDepInputsContext) -> usize {
+ db.log().add("Derived1 invoked");
+ db.dep_input1() / 2
+}
+
+#[test]
+fn revalidate() {
+ let db = &mut TestContextImpl::default();
+
+ db.set_dep_input1(0);
+
+ // Initial run starts from Memoized2:
+ let v = db.dep_memoized2();
+ assert_eq!(v, 0);
+ db.assert_log(&["Memoized2 invoked", "Memoized1 invoked", "Derived1 invoked"]);
+
+ // After that, we first try to validate Memoized1 but wind up
+ // running Memoized2. Note that we don't try to validate
+ // Derived1, so it is invoked by Memoized1.
+ db.set_dep_input1(44);
+ let v = db.dep_memoized2();
+ assert_eq!(v, 44);
+ db.assert_log(&["Memoized1 invoked", "Derived1 invoked", "Memoized2 invoked"]);
+
+ // Here validation of Memoized1 succeeds so Memoized2 never runs.
+ db.set_dep_input1(45);
+ let v = db.dep_memoized2();
+ assert_eq!(v, 44);
+ db.assert_log(&["Memoized1 invoked", "Derived1 invoked"]);
+
+ // Here, a change to input2 doesn't affect us, so nothing runs.
+ db.set_dep_input2(45);
+ let v = db.dep_memoized2();
+ assert_eq!(v, 44);
+ db.assert_log(&[]);
+}