Move around write for BorshSerialize, test

Author: af-afk

Cargo.toml

@@ -36,6 +36,7 @@ version = "1.0"
 [dev-dependencies]
 matches = { version = "0.1" }
 bencher = "0.1.4"
+borsh = { version = "1.5.5", default-features = false, features = ["derive"] }
 
 [[bench]]
 name = "extend"

src/arrayvec.rs

@@ -1428,7 +1428,7 @@ where
     }
 }
 
-#[cfg(all(feature = "borsh", not(feature = "std")))]
+#[cfg(all(feature = "borsh"))]
 /// Requires crate feature `"borsh"`
 impl<const CAP: usize> borsh::io::Write for ArrayVec<u8, CAP> {
     fn write(&mut self, data: &[u8]) -> borsh::io::Result<usize> {

tests/borsh.rs

@@ -46,6 +46,39 @@ mod array_vec {
         assert_ser(&vec, b"\x01\0\0\0\xef\xbe\xad\xde");
         assert_roundtrip(&vec);
     }
+
+    #[test]
+    fn test_borsh_one_to_one() {
+        use borsh::{BorshDeserialize, BorshSerialize};
+        #[derive(BorshSerialize, BorshDeserialize)]
+        struct Something {
+            a: String,
+        }
+        let a = Something {
+            a: "hello".to_owned(),
+        };
+        let mut vec = ArrayVec::<u8, 9>::new();
+        a.serialize(&mut vec).unwrap();
+        let mut test = Vec::new();
+        a.serialize(&mut test).unwrap();
+        assert_eq!(vec.as_slice(), test.as_slice());
+    }
+
+    #[test]
+    fn test_borsh_more_than() {
+        use borsh::{io::ErrorKind, BorshDeserialize, BorshSerialize};
+        #[derive(BorshSerialize, BorshDeserialize)]
+        struct Something {
+            a: String,
+        }
+        let mut b = ArrayVec::<u8, 5>::new();
+        let err = Something {
+            a: "Hello, world!".to_owned(),
+        }
+        .serialize(&mut b)
+        .unwrap_err();
+        assert_eq!(ErrorKind::WriteZero, err.kind());
+    }
 }
 
 mod array_string {