fix bench

holochain · Nov 6, 2023 · 346abcd · 346abcd
1 parent bf3b660
commit 346abcd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 16 deletions.
diff --git a/test/benches/bench.rs b/test/benches/bench.rs
@@ -116,7 +116,7 @@ pub fn wasm_call(c: &mut Criterion) {
 pub fn wasm_call_n(c: &mut Criterion) {
     let mut group = c.benchmark_group("wasm_call_n");
 
-    let instance_with_store = TestWasm::Io.instance();
+    let instance_with_store = TestWasm::Io.unmetered_instance();
 
     macro_rules! bench_n {
         ( $fs:expr; $t:ty; ) => {
@@ -165,7 +165,7 @@ pub fn wasm_call_n(c: &mut Criterion) {
 pub fn test_process_string(c: &mut Criterion) {
     let mut group = c.benchmark_group("test_process_string");
 
-    let instance_with_store = TestWasm::Test.instance();
+    let instance_with_store = TestWasm::Test.unmetered_instance();
 
     for n in vec![0, 1, 1_000, 1_000_000] {
         group.throughput(Throughput::Bytes(n));

diff --git a/test/src/wasms.rs b/test/src/wasms.rs
@@ -130,20 +130,22 @@ impl TestWasm {
                     .get_typed_function(&store_mut, "__hc__allocate_1")
                     .unwrap(),
             );
-            data_mut.wasmer_metering_points_exhausted = Some(
-                instance
-                    .exports
-                    .get_global("wasmer_metering_points_exhausted")
-                    .unwrap()
-                    .clone(),
-            );
-            data_mut.wasmer_metering_remaining_points = Some(
-                instance
-                    .exports
-                    .get_global("wasmer_metering_remaining_points")
-                    .unwrap()
-                    .clone(),
-            );
+            if metered {
+                data_mut.wasmer_metering_points_exhausted = Some(
+                    instance
+                        .exports
+                        .get_global("wasmer_metering_points_exhausted")
+                        .unwrap()
+                        .clone(),
+                );
+                data_mut.wasmer_metering_remaining_points = Some(
+                    instance
+                        .exports
+                        .get_global("wasmer_metering_remaining_points")
+                        .unwrap()
+                        .clone(),
+                );
+            }
         }
 
         InstanceWithStore {