mirror of
https://github.com/fluencelabs/wasm-bindgen
synced 2025-03-24 22:11:06 +00:00
* Tweak the implementation of heap closures This commit updates the implementation of the `Closure` type to internally store an `Rc` and be suitable for dropping a `Closure` during the execution of the closure. This is currently needed for promises but may be generally useful as well! * Support asynchronous tests This commit adds support for executing tests asynchronously. This is modeled by tests returning a `Future` instead of simply executing inline, and is signified with `#[wasm_bindgen_test(async)]`. Support for this is added through a new `wasm-bindgen-futures` crate which is a binding between the `futures` crate and JS `Promise` objects. Lots more details can be found in the details of the commit, but one of the end results is that the `web-sys` tests are now entirely contained in the same test suite and don't need `npm install` to be run to execute them! * Review tweaks * Add some bindings for `Function.call` to `js_sys` Name them `call0`, `call1`, `call2`, ... for the number of arguments being passed. * Use oneshots channels with `JsFuture` It did indeed clean up the implementation!
73 lines
2.3 KiB
Rust
73 lines
2.3 KiB
Rust
//! See the README for `wasm-bindgen-test` for a bit more info about what's
|
|
//! going on here.
|
|
|
|
extern crate proc_macro;
|
|
extern crate proc_macro2;
|
|
#[macro_use]
|
|
extern crate quote;
|
|
|
|
use std::sync::atomic::*;
|
|
use proc_macro2::*;
|
|
|
|
static CNT: AtomicUsize = ATOMIC_USIZE_INIT;
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn wasm_bindgen_test(
|
|
attr: proc_macro::TokenStream,
|
|
body: proc_macro::TokenStream,
|
|
) -> proc_macro::TokenStream {
|
|
let mut attr = attr.into_iter();
|
|
let mut async = false;
|
|
while let Some(token) = attr.next() {
|
|
match &token {
|
|
proc_macro::TokenTree::Ident(i) if i.to_string() == "async" => async = true,
|
|
_ => panic!("malformed `#[wasm_bindgen_test]` attribute"),
|
|
}
|
|
match &attr.next() {
|
|
Some(proc_macro::TokenTree::Punct(op)) if op.as_char() == ',' => {}
|
|
Some(_) => panic!("malformed `#[wasm_bindgen_test]` attribute"),
|
|
None => break,
|
|
}
|
|
}
|
|
|
|
let mut body = TokenStream::from(body).into_iter();
|
|
|
|
// Assume the input item is of the form `fn #ident ...`, and extract
|
|
// `#ident`
|
|
let fn_tok = body.next();
|
|
let ident = match body.next() {
|
|
Some(TokenTree::Ident(token)) => token,
|
|
_ => panic!("expected a function name"),
|
|
};
|
|
|
|
let mut tokens = Vec::<TokenTree>::new();
|
|
|
|
let test_body = if async {
|
|
quote! { cx.execute_async(test_name, #ident); }
|
|
} else {
|
|
quote! { cx.execute_sync(test_name, #ident); }
|
|
};
|
|
|
|
// We generate a `#[no_mangle]` with a known prefix so the test harness can
|
|
// later slurp up all of these functions and pass them as arguments to the
|
|
// main test harness. This is the entry point for all tests.
|
|
let name = format!("__wbg_test_{}_{}", ident, CNT.fetch_add(1, Ordering::SeqCst));
|
|
let name = Ident::new(&name, Span::call_site());
|
|
tokens.extend((quote! {
|
|
#[no_mangle]
|
|
pub extern fn #name(cx: *const ::wasm_bindgen_test::__rt::Context) {
|
|
unsafe {
|
|
let cx = &*cx;
|
|
let test_name = concat!(module_path!(), "::", stringify!(#ident));
|
|
#test_body
|
|
}
|
|
}
|
|
}).into_iter());
|
|
|
|
tokens.extend(fn_tok);
|
|
tokens.push(ident.into());
|
|
tokens.extend(body);
|
|
|
|
tokens.into_iter().collect::<TokenStream>().into()
|
|
}
|