test upgrade

dfinity · Sep 23, 2023 · b4ba793 · b4ba793
1 parent d3822af
commit b4ba793
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/instrumentation.rs b/src/instrumentation.rs
@@ -613,7 +613,7 @@ fn inject_pre_upgrade(m: &mut Module, vars: &Variables, config: &Config) {
     let mut builder = get_or_create_export_func(m, "canister_pre_upgrade");
     #[rustfmt::skip]
     builder
-        // backup memory
+        // backup memory. This is not restricted needed, since the end of pre-upgrade.
         .i32_const(0)
         .load(memory, LoadKind::I64 { atomic: false }, MemArg { offset: 0, align: 8})
         .local_set(a)

diff --git a/tests/deployable.ic-repl.sh b/tests/deployable.ic-repl.sh
@@ -14,6 +14,16 @@ function install(wasm) {
   );
   S
 };
+function upgrade(S, wasm) {
+  call ic.install_code(
+    record {
+      arg = encode ();
+      wasm_module = wasm;
+      mode = variant { upgrade };
+      canister_id = S;
+    }
+  );
+};
 
 function motoko(wasm) {
   let S = install(wasm);
@@ -94,14 +104,20 @@ function check_profiling(S, cycles, len) {
   assert _ == (cycles : int64);
   call S.__get_profiling();
   assert _.size() == (len : nat);
+  null
 };
 
 let S = motoko(file("ok/motoko-instrument.wasm"));
 check_profiling(S, 9397, 78);
 let S = motoko(file("ok/motoko-gc-instrument.wasm"));
 check_profiling(S, 250, 4);
-let S = motoko(file("ok/motoko-region-instrument.wasm"));
+let wasm = file("ok/motoko-region-instrument.wasm");
+let S = motoko(wasm);
 check_profiling(S, 463589, 78);
+upgrade(S, wasm);
+call S.get();
+assert _ == (45 : nat);
+check_profiling(S, 472223, 374);
 motoko(file("ok/motoko-shrink.wasm"));
 motoko(file("ok/motoko-limit.wasm"));