Add some tests for exceptions

This commit is contained in:
Alex Crichton 2017-12-18 14:44:09 -08:00
parent 6593b5ef69
commit 62808e8d99
5 changed files with 115 additions and 46 deletions

View File

@ -6,6 +6,7 @@ pub struct Js {
pub expose_global_exports: bool, pub expose_global_exports: bool,
pub expose_get_string_from_wasm: bool, pub expose_get_string_from_wasm: bool,
pub expose_pass_string_to_wasm: bool, pub expose_pass_string_to_wasm: bool,
pub expose_assert_num: bool,
pub expose_token: bool, pub expose_token: bool,
pub exports: Vec<(String, String)>, pub exports: Vec<(String, String)>,
pub classes: Vec<String>, pub classes: Vec<String>,
@ -24,17 +25,6 @@ impl Js {
} }
pub fn generate_free_function(&mut self, func: &shared::Function) { pub fn generate_free_function(&mut self, func: &shared::Function) {
let simple = func.arguments.iter().all(|t| t.is_number()) &&
func.ret.as_ref().map(|t| t.is_number()).unwrap_or(true);
if simple {
self.exports.push((
func.name.clone(),
format!("obj.instance.exports.{}", func.name),
));
return
}
let ret = self.generate_function(&format!("function {}", func.name), let ret = self.generate_function(&format!("function {}", func.name),
&func.name, &func.name,
false, false,
@ -118,7 +108,11 @@ impl Js {
passed_args.push_str(arg); passed_args.push_str(arg);
}; };
match *arg { match *arg {
shared::Type::Number => pass(&name), shared::Type::Number => {
self.expose_assert_num = true;
arg_conversions.push_str(&format!("_assertNum({});\n", name));
pass(&name)
}
shared::Type::BorrowedStr | shared::Type::BorrowedStr |
shared::Type::String => { shared::Type::String => {
self.expose_global_exports = true; self.expose_global_exports = true;
@ -207,15 +201,25 @@ impl Js {
globals.push_str("\ globals.push_str("\
const token = Symbol('foo'); const token = Symbol('foo');
function _checkToken(sym) { function _checkToken(sym) {
if (token != sym) if (token !== sym)
throw new Error('cannot invoke `new` directly'); throw new Error('cannot invoke `new` directly');
} }
"); ");
} }
if self.expose_assert_num {
globals.push_str("\
function _assertNum(n) {
if (typeof(n) !== 'number')
throw new Error('expected a number argument');
}
");
}
if self.expose_pass_string_to_wasm { if self.expose_pass_string_to_wasm {
if self.nodejs { if self.nodejs {
globals.push_str(" globals.push_str("
function passStringToWasm(arg) { function passStringToWasm(arg) {
if (typeof(n) !== 'string')
throw new Error('expected a string argument');
const buf = Buffer.from(arg); const buf = Buffer.from(arg);
const len = buf.length; const len = buf.length;
const ptr = exports.__wbindgen_malloc(len); const ptr = exports.__wbindgen_malloc(len);

View File

@ -10,6 +10,6 @@ proc-macro = true
syn = { git = 'https://github.com/dtolnay/syn', features = ['full'] } syn = { git = 'https://github.com/dtolnay/syn', features = ['full'] }
synom = { git = 'https://github.com/dtolnay/syn' } synom = { git = 'https://github.com/dtolnay/syn' }
quote = { git = 'https://github.com/dtolnay/quote' } quote = { git = 'https://github.com/dtolnay/quote' }
proc-macro2 = "0.1" proc-macro2 = { version = "0.1", features = ["unstable"] }
serde_json = "1" serde_json = "1"
wasm-bindgen-shared = { path = "../wasm-bindgen-shared" } wasm-bindgen-shared = { path = "../wasm-bindgen-shared" }

View File

@ -12,7 +12,7 @@ extern crate wasm_bindgen_shared;
use std::sync::atomic::*; use std::sync::atomic::*;
use proc_macro::TokenStream; use proc_macro::TokenStream;
use proc_macro2::Literal; use proc_macro2::{Literal, Span};
use quote::{Tokens, ToTokens}; use quote::{Tokens, ToTokens};
mod ast; mod ast;
@ -20,6 +20,10 @@ mod ast;
static MALLOC_GENERATED: AtomicBool = ATOMIC_BOOL_INIT; static MALLOC_GENERATED: AtomicBool = ATOMIC_BOOL_INIT;
static BOXED_STR_GENERATED: AtomicBool = ATOMIC_BOOL_INIT; static BOXED_STR_GENERATED: AtomicBool = ATOMIC_BOOL_INIT;
macro_rules! my_quote {
($($t:tt)*) => (quote_spanned!(Span::call_site(), $($t)*))
}
#[proc_macro] #[proc_macro]
pub fn wasm_bindgen(input: TokenStream) -> TokenStream { pub fn wasm_bindgen(input: TokenStream) -> TokenStream {
let file = syn::parse::<syn::File>(input) let file = syn::parse::<syn::File>(input)
@ -69,7 +73,7 @@ pub fn wasm_bindgen(input: TokenStream) -> TokenStream {
}; };
let generated_static_length = generated_static.len(); let generated_static_length = generated_static.len();
(quote! { (my_quote! {
#[no_mangle] #[no_mangle]
#[allow(non_upper_case_globals)] #[allow(non_upper_case_globals)]
pub static #generated_static_name: [u8; #generated_static_length] = pub static #generated_static_name: [u8; #generated_static_length] =
@ -100,7 +104,7 @@ fn bindgen_struct(s: &ast::Struct, into: &mut Tokens) {
let name = &s.name; let name = &s.name;
let free_fn = s.free_function(); let free_fn = s.free_function();
(quote! { (my_quote! {
#[no_mangle] #[no_mangle]
pub unsafe extern fn #free_fn(ptr: *mut ::std::cell::RefCell<#name>) { pub unsafe extern fn #free_fn(ptr: *mut ::std::cell::RefCell<#name>) {
assert!(!ptr.is_null()); assert!(!ptr.is_null());
@ -149,8 +153,8 @@ fn bindgen(export_name: &syn::Lit,
let mut offset = 0; let mut offset = 0;
if let Receiver::StructMethod(class, _, _) = receiver { if let Receiver::StructMethod(class, _, _) = receiver {
args.push(quote! { me: *mut ::std::cell::RefCell<#class> }); args.push(my_quote! { me: *mut ::std::cell::RefCell<#class> });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
assert!(!me.is_null()); assert!(!me.is_null());
let me = unsafe { &*me }; let me = unsafe { &*me };
}); });
@ -162,15 +166,15 @@ fn bindgen(export_name: &syn::Lit,
let ident = syn::Ident::from(format!("arg{}", i)); let ident = syn::Ident::from(format!("arg{}", i));
match *ty { match *ty {
ast::Type::Integer(i) => { ast::Type::Integer(i) => {
args.push(quote! { #ident: #i }); args.push(my_quote! { #ident: #i });
} }
ast::Type::BorrowedStr => { ast::Type::BorrowedStr => {
malloc = !MALLOC_GENERATED.swap(true, Ordering::SeqCst); malloc = !MALLOC_GENERATED.swap(true, Ordering::SeqCst);
let ptr = syn::Ident::from(format!("arg{}_ptr", i)); let ptr = syn::Ident::from(format!("arg{}_ptr", i));
let len = syn::Ident::from(format!("arg{}_len", i)); let len = syn::Ident::from(format!("arg{}_len", i));
args.push(quote! { #ptr: *const u8 }); args.push(my_quote! { #ptr: *const u8 });
args.push(quote! { #len: usize }); args.push(my_quote! { #len: usize });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
let #ident = unsafe { let #ident = unsafe {
let slice = ::std::slice::from_raw_parts(#ptr, #len); let slice = ::std::slice::from_raw_parts(#ptr, #len);
::std::str::from_utf8_unchecked(slice) ::std::str::from_utf8_unchecked(slice)
@ -181,9 +185,9 @@ fn bindgen(export_name: &syn::Lit,
malloc = !MALLOC_GENERATED.swap(true, Ordering::SeqCst); malloc = !MALLOC_GENERATED.swap(true, Ordering::SeqCst);
let ptr = syn::Ident::from(format!("arg{}_ptr", i)); let ptr = syn::Ident::from(format!("arg{}_ptr", i));
let len = syn::Ident::from(format!("arg{}_len", i)); let len = syn::Ident::from(format!("arg{}_len", i));
args.push(quote! { #ptr: *mut u8 }); args.push(my_quote! { #ptr: *mut u8 });
args.push(quote! { #len: usize }); args.push(my_quote! { #len: usize });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
let #ident = unsafe { let #ident = unsafe {
let vec = ::std::vec::Vec::from_raw_parts(#ptr, #len, #len); let vec = ::std::vec::Vec::from_raw_parts(#ptr, #len, #len);
::std::string::String::from_utf8_unchecked(vec) ::std::string::String::from_utf8_unchecked(vec)
@ -191,8 +195,8 @@ fn bindgen(export_name: &syn::Lit,
}); });
} }
ast::Type::ByValue(name) => { ast::Type::ByValue(name) => {
args.push(quote! { #ident: *mut ::std::cell::RefCell<#name> }); args.push(my_quote! { #ident: *mut ::std::cell::RefCell<#name> });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
assert!(!#ident.is_null()); assert!(!#ident.is_null());
let #ident = unsafe { let #ident = unsafe {
(*#ident).borrow_mut(); (*#ident).borrow_mut();
@ -201,53 +205,53 @@ fn bindgen(export_name: &syn::Lit,
}); });
} }
ast::Type::ByRef(name) => { ast::Type::ByRef(name) => {
args.push(quote! { #ident: *mut ::std::cell::RefCell<#name> }); args.push(my_quote! { #ident: *mut ::std::cell::RefCell<#name> });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
assert!(!#ident.is_null()); assert!(!#ident.is_null());
let #ident = unsafe { (*#ident).borrow() }; let #ident = unsafe { (*#ident).borrow() };
let #ident = &*#ident; let #ident = &*#ident;
}); });
} }
ast::Type::ByMutRef(name) => { ast::Type::ByMutRef(name) => {
args.push(quote! { #ident: *mut ::std::cell::RefCell<#name> }); args.push(my_quote! { #ident: *mut ::std::cell::RefCell<#name> });
arg_conversions.push(quote! { arg_conversions.push(my_quote! {
assert!(!#ident.is_null()); assert!(!#ident.is_null());
let mut #ident = unsafe { (*#ident).borrow_mut() }; let mut #ident = unsafe { (*#ident).borrow_mut() };
let #ident = &mut *#ident; let #ident = &mut *#ident;
}); });
} }
} }
converted_arguments.push(quote! { #ident }); converted_arguments.push(my_quote! { #ident });
} }
let ret_ty; let ret_ty;
let convert_ret; let convert_ret;
match ret_type { match ret_type {
Some(&ast::Type::Integer(i)) => { Some(&ast::Type::Integer(i)) => {
ret_ty = quote! { -> #i }; ret_ty = my_quote! { -> #i };
convert_ret = quote! { #ret }; convert_ret = my_quote! { #ret };
} }
Some(&ast::Type::BorrowedStr) => panic!("can't return a borrowed string"), Some(&ast::Type::BorrowedStr) => panic!("can't return a borrowed string"),
Some(&ast::Type::ByRef(_)) => panic!("can't return a borrowed ref"), Some(&ast::Type::ByRef(_)) => panic!("can't return a borrowed ref"),
Some(&ast::Type::ByMutRef(_)) => panic!("can't return a borrowed ref"), Some(&ast::Type::ByMutRef(_)) => panic!("can't return a borrowed ref"),
Some(&ast::Type::String) => { Some(&ast::Type::String) => {
boxed_str = !BOXED_STR_GENERATED.swap(true, Ordering::SeqCst); boxed_str = !BOXED_STR_GENERATED.swap(true, Ordering::SeqCst);
ret_ty = quote! { -> *mut String }; ret_ty = my_quote! { -> *mut String };
convert_ret = quote! { Box::into_raw(Box::new(#ret)) }; convert_ret = my_quote! { Box::into_raw(Box::new(#ret)) };
} }
Some(&ast::Type::ByValue(name)) => { Some(&ast::Type::ByValue(name)) => {
ret_ty = quote! { -> *mut ::std::cell::RefCell<#name> }; ret_ty = my_quote! { -> *mut ::std::cell::RefCell<#name> };
convert_ret = quote! { convert_ret = my_quote! {
Box::into_raw(Box::new(::std::cell::RefCell::new(#ret))) Box::into_raw(Box::new(::std::cell::RefCell::new(#ret)))
}; };
} }
None => { None => {
ret_ty = quote! {}; ret_ty = my_quote! {};
convert_ret = quote! {}; convert_ret = my_quote! {};
} }
} }
let malloc = if malloc { let malloc = if malloc {
quote! { my_quote! {
#[no_mangle] #[no_mangle]
pub extern fn __wbindgen_malloc(size: usize) -> *mut u8 { pub extern fn __wbindgen_malloc(size: usize) -> *mut u8 {
let mut ret = Vec::with_capacity(size); let mut ret = Vec::with_capacity(size);
@ -262,12 +266,12 @@ fn bindgen(export_name: &syn::Lit,
} }
} }
} else { } else {
quote! { my_quote! {
} }
}; };
let boxed_str = if boxed_str { let boxed_str = if boxed_str {
quote! { my_quote! {
#[no_mangle] #[no_mangle]
pub unsafe extern fn __wbindgen_boxed_str_len(ptr: *mut String) -> usize { pub unsafe extern fn __wbindgen_boxed_str_len(ptr: *mut String) -> usize {
(*ptr).len() (*ptr).len()
@ -284,11 +288,11 @@ fn bindgen(export_name: &syn::Lit,
} }
} }
} else { } else {
quote! { my_quote! {
} }
}; };
let tokens = quote! { let tokens = my_quote! {
#malloc #malloc
#boxed_str #boxed_str

View File

@ -105,3 +105,38 @@ fn strings() {
"#) "#)
.test(); .test();
} }
#[test]
fn exceptions() {
test_support::project()
.file("src/lib.rs", r#"
#![feature(proc_macro)]
extern crate wasm_bindgen;
use wasm_bindgen::prelude::*;
wasm_bindgen! {
pub struct A {
}
impl A {
pub fn new() -> A {
A {}
}
}
}
"#)
.file("test.js", r#"
import * as assert from "assert";
export function test(wasm) {
assert.throws(() => new wasm.A(), /cannot invoke `new` directly/);
let a = wasm.A.new();
a.free();
// TODO: figure out a better error message?
assert.throws(() => a.free(), /RuntimeError: unreachable/);
}
"#)
.test();
}

View File

@ -101,3 +101,29 @@ fn return_a_string() {
"#) "#)
.test(); .test();
} }
#[test]
fn exceptions() {
test_support::project()
.file("src/lib.rs", r#"
#![feature(proc_macro)]
extern crate wasm_bindgen;
use wasm_bindgen::prelude::*;
wasm_bindgen! {
pub fn foo(_a: u32) {}
pub fn bar(_a: &str) {}
}
"#)
.file("test.js", r#"
import * as assert from "assert";
export function test(wasm) {
assert.throws(() => wasm.foo('a'), /expected a number argument/);
assert.throws(() => wasm.bar(3), /expected a string argument/);
}
"#)
.test();
}