diff --git a/rs/execution_environment/src/canister_manager.rs b/rs/execution_environment/src/canister_manager.rs index 7e5760a29b58..2d2afe897f6e 100644 --- a/rs/execution_environment/src/canister_manager.rs +++ b/rs/execution_environment/src/canister_manager.rs @@ -1686,11 +1686,10 @@ impl CanisterManager { if self.config.rate_limiting_of_heap_delta == FlagStatus::Enabled && canister.scheduler_state.heap_delta_debit >= self.config.heap_delta_rate_limit { - return Err(CanisterManagerError::WasmChunkStoreError { - message: format!( - "Canister is heap delta rate limited. Current delta debit: {}, limit: {}", - canister.scheduler_state.heap_delta_debit, self.config.heap_delta_rate_limit - ), + return Err(CanisterManagerError::CanisterHeapDeltaRateLimited { + canister_id: canister.canister_id(), + value: canister.scheduler_state.heap_delta_debit, + limit: self.config.heap_delta_rate_limit, }); } @@ -2851,11 +2850,10 @@ impl CanisterManager { if self.config.rate_limiting_of_heap_delta == FlagStatus::Enabled && canister.scheduler_state.heap_delta_debit >= self.config.heap_delta_rate_limit { - return Err(CanisterManagerError::WasmChunkStoreError { - message: format!( - "Canister is heap delta rate limited. Current delta debit: {}, limit: {}", - canister.scheduler_state.heap_delta_debit, self.config.heap_delta_rate_limit - ), + return Err(CanisterManagerError::CanisterHeapDeltaRateLimited { + canister_id: canister.canister_id(), + value: canister.scheduler_state.heap_delta_debit, + limit: self.config.heap_delta_rate_limit, }); } diff --git a/rs/execution_environment/src/canister_manager/tests.rs b/rs/execution_environment/src/canister_manager/tests.rs index e57f6e45ea8c..a22e0c592605 100644 --- a/rs/execution_environment/src/canister_manager/tests.rs +++ b/rs/execution_environment/src/canister_manager/tests.rs @@ -5341,9 +5341,8 @@ fn upload_chunk_fails_when_heap_delta_rate_limited() { test.subnet_message("upload_chunk", upload_args.encode()) .unwrap_err() .assert_contains( - ErrorCode::CanisterContractViolation, - "Error from Wasm chunk store: Canister is heap delta rate limited. \ - Current delta debit: 1048576, limit: 1048576.", + ErrorCode::CanisterHeapDeltaRateLimited, + &format!("Canister {canister_id} is heap delta rate limited: current delta debit is 1048576, but limit is 1048576") ); assert_eq!(