73 lines
1.9 KiB
Rust
73 lines
1.9 KiB
Rust
#![cfg(feature = "borsh")]
|
|
use std::fmt;
|
|
extern crate arrayvec;
|
|
extern crate borsh;
|
|
|
|
fn assert_ser<T: borsh::BorshSerialize>(v: &T, expected_bytes: &[u8]) {
|
|
let mut actual_bytes = Vec::new();
|
|
v.serialize(&mut actual_bytes).unwrap();
|
|
assert_eq!(actual_bytes, expected_bytes);
|
|
}
|
|
|
|
fn assert_roundtrip<T: borsh::BorshSerialize + borsh::BorshDeserialize + PartialEq + fmt::Debug>(v: &T) {
|
|
let mut bytes = Vec::new();
|
|
v.serialize(&mut bytes).unwrap();
|
|
let v_de = T::try_from_slice(&bytes).unwrap();
|
|
assert_eq!(*v, v_de);
|
|
}
|
|
|
|
mod array_vec {
|
|
use arrayvec::ArrayVec;
|
|
use super::{assert_ser, assert_roundtrip};
|
|
|
|
#[test]
|
|
fn test_empty() {
|
|
let vec = ArrayVec::<u32, 0>::new();
|
|
assert_ser(&vec, b"\0\0\0\0");
|
|
assert_roundtrip(&vec);
|
|
}
|
|
|
|
#[test]
|
|
fn test_full() {
|
|
let mut vec = ArrayVec::<u32, 3>::new();
|
|
vec.push(0xdeadbeef);
|
|
vec.push(0x123);
|
|
vec.push(0x456);
|
|
assert_ser(&vec, b"\x03\0\0\0\xef\xbe\xad\xde\x23\x01\0\0\x56\x04\0\0");
|
|
assert_roundtrip(&vec);
|
|
}
|
|
|
|
#[test]
|
|
fn test_with_free_capacity() {
|
|
let mut vec = ArrayVec::<u32, 3>::new();
|
|
vec.push(0xdeadbeef);
|
|
assert_ser(&vec, b"\x01\0\0\0\xef\xbe\xad\xde");
|
|
assert_roundtrip(&vec);
|
|
}
|
|
}
|
|
|
|
mod array_string {
|
|
use arrayvec::ArrayString;
|
|
use super::{assert_ser, assert_roundtrip};
|
|
|
|
#[test]
|
|
fn test_empty() {
|
|
let string = ArrayString::<0>::new();
|
|
assert_ser(&string, b"\0\0\0\0");
|
|
assert_roundtrip(&string);
|
|
}
|
|
|
|
#[test]
|
|
fn test_full() {
|
|
let string = ArrayString::from_byte_string(b"hello world").unwrap();
|
|
assert_ser(&string, b"\x0b\0\0\0hello world");
|
|
assert_roundtrip(&string);
|
|
}
|
|
|
|
#[test]
|
|
fn test_with_free_capacity() {
|
|
let string = ArrayString::<16>::from("hello world").unwrap();
|
|
assert_ser(&string, b"\x0b\0\0\0hello world");
|
|
assert_roundtrip(&string);
|
|
}
|
|
}
|