Unnamed repository; edit this file 'description' to name the repository.
Diffstat (limited to 'crates/query-group-macro/tests/multiple_dbs.rs')
-rw-r--r--crates/query-group-macro/tests/multiple_dbs.rs25
1 files changed, 25 insertions, 0 deletions
diff --git a/crates/query-group-macro/tests/multiple_dbs.rs b/crates/query-group-macro/tests/multiple_dbs.rs
new file mode 100644
index 0000000000..f36e7fdbeb
--- /dev/null
+++ b/crates/query-group-macro/tests/multiple_dbs.rs
@@ -0,0 +1,25 @@
+use query_group_macro::query_group;
+
+#[query_group]
+pub trait DatabaseOne: salsa::Database {
+ #[salsa::input]
+ fn input_string(&self) -> String;
+
+ // unadorned query
+ #[salsa::invoke_interned(length)]
+ fn length(&self, key: ()) -> usize;
+}
+
+#[query_group]
+pub trait DatabaseTwo: DatabaseOne {
+ #[salsa::invoke_interned(second_length)]
+ fn second_length(&self, key: ()) -> usize;
+}
+
+fn length(db: &dyn DatabaseOne, _key: ()) -> usize {
+ db.input_string().len()
+}
+
+fn second_length(db: &dyn DatabaseTwo, _key: ()) -> usize {
+ db.input_string().len()
+}