Moved cache testing to the general tests

This commit is contained in:
Syrus 2020-04-16 12:14:12 -07:00
parent 636b0291af
commit 595f5bb472
2 changed files with 56 additions and 50 deletions

View File

@ -145,53 +145,3 @@ impl Cache for FileSystemCache {
Ok(()) Ok(())
} }
} }
#[cfg(test)]
mod tests {
use super::*;
use std::env;
#[test]
fn test_file_system_cache_run() {
use crate::{compile, imports, Func};
use wabt::wat2wasm;
static WAT: &'static str = r#"
(module
(type $t0 (func (param i32) (result i32)))
(func $add_one (export "add_one") (type $t0) (param $p0 i32) (result i32)
get_local $p0
i32.const 1
i32.add))
"#;
let wasm = wat2wasm(WAT).unwrap();
let module = compile(&wasm).unwrap();
let cache_dir = env::temp_dir();
println!("test temp_dir {:?}", cache_dir);
let mut fs_cache = unsafe {
FileSystemCache::new(cache_dir)
.map_err(|e| format!("Cache error: {:?}", e))
.unwrap()
};
// store module
let key = WasmHash::generate(&wasm);
fs_cache.store(key, module.clone()).unwrap();
// load module
let cached_module = fs_cache.load(key).unwrap();
let import_object = imports! {};
let instance = cached_module.instantiate(&import_object).unwrap();
let add_one: Func<i32, i32> = instance.exports.get("add_one").unwrap();
let value = add_one.call(42).unwrap();
// verify it works
assert_eq!(value, 43);
}
}

56
tests/cache.rs Normal file
View File

@ -0,0 +1,56 @@
#[macro_use]
mod utils;
wasmer_backends! {
use std::env;
use wasmer::{compiler::compile_with, imports, Func};
use wasmer_runtime::Backend as BaseBackend;
use wasmer_runtime::cache::{FileSystemCache, Cache, WasmHash};
use wabt::wat2wasm;
use std::str::FromStr;
#[test]
fn test_file_system_cache_run() {
static WAT: &'static str = r#"
(module
(type $t0 (func (param i32) (result i32)))
(func $add_one (export "add_one") (type $t0) (param $p0 i32) (result i32)
get_local $p0
i32.const 1
i32.add))
"#;
let wasm = wat2wasm(WAT).unwrap();
let module = compile_with(&wasm, &*get_compiler()).unwrap();
let cache_dir = env::temp_dir();
println!("test temp_dir {:?}", cache_dir);
let mut fs_cache = unsafe {
FileSystemCache::new(cache_dir)
.map_err(|e| format!("Cache error: {:?}", e))
.unwrap()
};
// store module
let key = WasmHash::generate(&wasm);
fs_cache.store(key, module.clone()).unwrap();
// We need to transform the `wasmer::Backend` into
// `wasmer_runtime::Backend` as that's something that the
// cache loader can understand.
let backend = BaseBackend::from_str(get_backend().to_string()).expect("Can't transform wasmer backend into wasmer_runtime backend");
// load module
let cached_module = fs_cache.load_with_backend(key, backend).unwrap();
let import_object = imports! {};
let instance = cached_module.instantiate(&import_object).unwrap();
let add_one: Func<i32, i32> = instance.exports.get("add_one").unwrap();
let value = add_one.call(42).unwrap();
// verify it works
assert_eq!(value, 43);
}
}