test: Implement MemoryUsage on test middlewares.

zeroctr · Mar 22, 2021 · c212f74 · c212f74
1 parent 2afed4d
commit c212f74
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -69,6 +69,7 @@ criterion = "0.3"
 lazy_static = "1.4"
 wasmer-engine-dummy = { path = "tests/lib/engine-dummy" }
 tempfile = "3.1"
+loupe = { git = "https://github.com/wasmerio/loupe", branch = "master" }
 
 [features]
 # Don't add the compiler features in default, please add them on the Makefile

diff --git a/tests/compilers/middlewares.rs b/tests/compilers/middlewares.rs
@@ -1,11 +1,12 @@
 use crate::utils::get_store_with_middlewares;
 use anyhow::Result;
 
+use loupe::MemoryUsage;
 use std::sync::Arc;
 use wasmer::wasmparser::Operator;
 use wasmer::*;
 
-#[derive(Debug)]
+#[derive(Debug, MemoryUsage)]
 struct Add2MulGen {
     value_off: i32,
 }
@@ -47,7 +48,7 @@ impl FunctionMiddleware for Add2Mul {
     }
 }
 
-#[derive(Debug)]
+#[derive(Debug, MemoryUsage)]
 struct FusionGen;
 
 #[derive(Debug)]