Add Array#flat and Array#flatMap to js-sys (fixes #1454)

This commit is contained in:
Teddy Katz
2019-06-03 18:32:58 -04:00
parent 6ac61b5feb
commit 5c5c13cf9e
2 changed files with 55 additions and 0 deletions

View File

@ -43,6 +43,44 @@ fn filter() {
);
}
#[wasm_bindgen_test]
fn flat() {
let array = js_array![
js_array!["a", "b", "c"],
"d",
js_array!["e", js_array!["f", "g"]]
];
assert_eq!(
to_rust(&array.flat(1).slice(0, 5)),
vec!["a", "b", "c", "d", "e"]
);
assert_eq!(array.flat(1).length(), 6);
assert_eq!(
to_rust(&array.flat(2)),
vec!["a", "b", "c", "d", "e", "f", "g"]
);
}
#[wasm_bindgen_test]
fn flat_map() {
let array = js_array![1, 2, 3, 1];
assert_eq!(
to_rust(
&array.flat_map(&mut |val, _, _| match val.as_f64().map(|v| v as i32) {
Some(1) => vec![JsString::from("x").into(), JsString::from("x").into()],
Some(2) => vec![],
Some(3) => vec![JsString::from("z").into()],
_ => panic!("Unexpected conversion"),
})
),
vec!["x", "x", "z", "x", "x"]
);
}
#[wasm_bindgen_test]
fn index_of() {
let chars = js_array!["a", "c", "x", "n"];