From a6a867aae4c21c3f31961f8a78b3c87e4f36021f Mon Sep 17 00:00:00 2001
From: max-dfinity <100170574+max-dfinity@users.noreply.github.com>
Date: Thu, 5 Dec 2024 15:42:24 -0800
Subject: [PATCH] feat: switch to composite queries for benchmarks

This allows benchmarks to include query calls.
---
 canbench-rs-macros/src/lib.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/canbench-rs-macros/src/lib.rs b/canbench-rs-macros/src/lib.rs
index 2b829d43..8e19a46e 100644
--- a/canbench-rs-macros/src/lib.rs
+++ b/canbench-rs-macros/src/lib.rs
@@ -54,7 +54,7 @@ pub fn bench(arg_tokens: TokenStream, item: TokenStream) -> TokenStream {
             quote! {
                 #input
 
-                #[ic_cdk::query]
+                #[ic_cdk::query(composite = true)]
                 #[allow(non_snake_case)]
                 fn #renamed_func_name() -> canbench_rs::BenchResult {
                     #func_name()
@@ -73,7 +73,7 @@ pub fn bench(arg_tokens: TokenStream, item: TokenStream) -> TokenStream {
             quote! {
                 #input
 
-                #[ic_cdk::query]
+                #[ic_cdk::query(composite = true)]
                 #[allow(non_snake_case)]
                 fn #renamed_func_name() -> canbench_rs::BenchResult {
                     canbench_rs::bench_fn(|| {