vendor/wasm-bindgen/tests/wasm/enums.rs

   1 use self::inner::ColorWithCustomValues;
   2 use wasm_bindgen::prelude::*;
   3 use wasm_bindgen_test::*;
   4
   5 #[wasm_bindgen(module = "tests/wasm/enums.js")]
   6 extern "C" {
   7     fn js_c_style_enum();
   8     fn js_c_style_enum_with_custom_values();
   9     fn js_handle_optional_enums(x: Option<Color>) -> Option<Color>;
  10     fn js_expect_enum(x: Color, y: Option<Color>);
  11     fn js_expect_enum_none(x: Option<Color>);
  12     fn js_renamed_enum(b: RenamedEnum);
  13 }
  14
  15 #[wasm_bindgen]
  16 #[derive(PartialEq, Debug)]
  17 pub enum Color {
  18     Green,
  19     Yellow,
  20     Red,
  21 }
  22
  23 pub mod inner {
  24     use wasm_bindgen::prelude::*;
  25
  26     #[wasm_bindgen]
  27     pub enum ColorWithCustomValues {
  28         Green = 21,
  29         Yellow = 34,
  30         Red = 2,
  31     }
  32 }
  33
  34 #[wasm_bindgen(js_name = JsRenamedEnum)]
  35 #[derive(Copy, Clone)]
  36 pub enum RenamedEnum {
  37     A = 10,
  38     B = 20,
  39 }
  40
  41 #[wasm_bindgen]
  42 pub fn enum_cycle(color: Color) -> Color {
  43     match color {
  44         Color::Green => Color::Yellow,
  45         Color::Yellow => Color::Red,
  46         Color::Red => Color::Green,
  47     }
  48 }
  49
  50 #[wasm_bindgen]
  51 pub fn enum_with_custom_values_cycle(color: ColorWithCustomValues) -> ColorWithCustomValues {
  52     match color {
  53         ColorWithCustomValues::Green => ColorWithCustomValues::Yellow,
  54         ColorWithCustomValues::Yellow => ColorWithCustomValues::Red,
  55         ColorWithCustomValues::Red => ColorWithCustomValues::Green,
  56     }
  57 }
  58
  59 #[wasm_bindgen_test]
  60 fn c_style_enum() {
  61     js_c_style_enum();
  62 }
  63
  64 #[wasm_bindgen_test]
  65 fn c_style_enum_with_custom_values() {
  66     js_c_style_enum_with_custom_values();
  67 }
  68
  69 #[wasm_bindgen]
  70 pub fn handle_optional_enums(x: Option<Color>) -> Option<Color> {
  71     x
  72 }
  73
  74 #[wasm_bindgen_test]
  75 fn test_optional_enums() {
  76     use self::Color::*;
  77
  78     assert_eq!(js_handle_optional_enums(None), None);
  79     assert_eq!(js_handle_optional_enums(Some(Green)), Some(Green));
  80     assert_eq!(js_handle_optional_enums(Some(Yellow)), Some(Yellow));
  81     assert_eq!(js_handle_optional_enums(Some(Red)), Some(Red));
  82 }
  83
  84 #[wasm_bindgen_test]
  85 fn test_optional_enum_values() {
  86     use self::Color::*;
  87
  88     js_expect_enum(Green, Some(Green));
  89     js_expect_enum(Yellow, Some(Yellow));
  90     js_expect_enum(Red, Some(Red));
  91     js_expect_enum_none(None);
  92 }
  93
  94 #[wasm_bindgen_test]
  95 fn test_renamed_enum() {
  96     js_renamed_enum(RenamedEnum::B);
  97 }