feat: fix more tests in wasm land

banyancomputer · Feb 11, 2024 · a89a687 · a89a687
1 parent 5576cfb
commit a89a687
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 6 deletions.
diff --git a/src/codec/crypto/authentication_tag.rs b/src/codec/crypto/authentication_tag.rs
@@ -70,7 +70,11 @@ mod tests {
 
     use super::*;
 
-    #[test]
+    #[cfg(target_arch = "wasm32")]
+    use wasm_bindgen_test::wasm_bindgen_test;
+
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_authentication_tag_parsing() {
         let mut rng = rand::thread_rng();
         let input: [u8; TAG_LENGTH + 4] = rng.gen();
@@ -80,7 +84,8 @@ mod tests {
         assert_eq!(tag.as_bytes(), &input[..TAG_LENGTH]);
     }
 
-    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_authentication_tag_parsing_stream_too_short() {
         let input = [0u8; TAG_LENGTH - 1];
         let result = AuthenticationTag::parse(&input);

diff --git a/src/codec/crypto/fingerprint.rs b/src/codec/crypto/fingerprint.rs
@@ -40,9 +40,13 @@ impl From<&VerifyingKey> for Fingerprint {
 pub mod tests {
     use super::*;
 
+    #[cfg(target_arch = "wasm32")]
+    use wasm_bindgen_test::wasm_bindgen_test;
+
     const REFERENCE_FINGERPRINT_BYTES: &[u8; 32] = b"UUUUUUUUaaaaaaaaUUUUUUUUaaaaaaaa";
 
-    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_fingerprint_debug_fmt() {
         let fingerprint = Fingerprint::from(*REFERENCE_FINGERPRINT_BYTES);
         let fmt_str = format!("{:?}", fingerprint);
@@ -53,7 +57,8 @@ pub mod tests {
         );
     }
 
-    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_key_id_from_fingerprint() {
         let fingerprint = Fingerprint::from(*REFERENCE_FINGERPRINT_BYTES);
         let key_id = fingerprint.key_id();

diff --git a/src/codec/crypto/nonce.rs b/src/codec/crypto/nonce.rs
@@ -68,7 +68,11 @@ impl Deref for Nonce {
 mod tests {
     use super::*;
 
-    #[test]
+    #[cfg(target_arch = "wasm32")]
+    use wasm_bindgen_test::wasm_bindgen_test;
+
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_nonce_parsing() {
         let mut rng = rand::thread_rng();
         let input: [u8; NONCE_LENGTH + 4] = rng.gen();
@@ -81,7 +85,8 @@ mod tests {
         assert!(Nonce::parse_complete(&input[..NONCE_LENGTH]).is_ok());
     }
 
-    #[test]
+    #[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
+    #[cfg_attr(not(target_arch = "wasm32"), test)]
     fn test_nonce_parsing_stream_too_short() {
         let input = [0u8; NONCE_LENGTH - 1];
         let result = Nonce::parse(&input);