From 51fb7e7f513b66045a45426eb93b803c364b9f25 Mon Sep 17 00:00:00 2001 From: Dimitris Sarlis Date: Thu, 5 Feb 2026 09:34:52 +0000 Subject: [PATCH] perf: Speed up parameterized upgrade tests --- rs/nns/test_utils_macros/src/lib.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/rs/nns/test_utils_macros/src/lib.rs b/rs/nns/test_utils_macros/src/lib.rs index c98d563ebe74..c53abe7f702e 100644 --- a/rs/nns/test_utils_macros/src/lib.rs +++ b/rs/nns/test_utils_macros/src/lib.rs @@ -47,7 +47,7 @@ pub fn parameterized_upgrades(_attr: TokenStream, item: TokenStream) -> TokenStr // Export a #[test] with the `Always` upgrade scenario #[test] fn #always_testname() { - ic_nns_test_utils::itest_helpers::local_test_on_nns_subnet(|runtime| async move { + ic_nns_test_utils::itest_helpers::state_machine_test_on_nns_subnet(|runtime| async move { #original_fn_ident( &runtime, ic_nns_test_utils::itest_helpers::UpgradeTestingScenario::Always).await; @@ -58,7 +58,7 @@ pub fn parameterized_upgrades(_attr: TokenStream, item: TokenStream) -> TokenStr // Export a #[test] with the `Never` upgrade scenario #[test] fn #never_testname() { - ic_nns_test_utils::itest_helpers::local_test_on_nns_subnet(|runtime| async move { + ic_nns_test_utils::itest_helpers::state_machine_test_on_nns_subnet(|runtime| async move { #original_fn_ident( &runtime, ic_nns_test_utils::itest_helpers::UpgradeTestingScenario::Never).await;