src/libserialize/tests/leb128.rs

   1 extern crate serialize as rustc_serialize;
   2 use rustc_serialize::leb128::*;
   3
   4 macro_rules! impl_test_unsigned_leb128 {
   5     ($test_name:ident, $write_fn_name:ident, $read_fn_name:ident, $int_ty:ident) => (
   6         #[test]
   7         fn $test_name() {
   8             let mut stream = Vec::new();
   9
  10             for x in 0..62 {
  11                 $write_fn_name(&mut stream, (3u64 << x) as $int_ty);
  12             }
  13
  14             let mut position = 0;
  15             for x in 0..62 {
  16                 let expected = (3u64 << x) as $int_ty;
  17                 let (actual, bytes_read) = $read_fn_name(&stream[position ..]);
  18                 assert_eq!(expected, actual);
  19                 position += bytes_read;
  20             }
  21             assert_eq!(stream.len(), position);
  22         }
  23     )
  24 }
  25
  26 impl_test_unsigned_leb128!(test_u16_leb128, write_u16_leb128, read_u16_leb128, u16);
  27 impl_test_unsigned_leb128!(test_u32_leb128, write_u32_leb128, read_u32_leb128, u32);
  28 impl_test_unsigned_leb128!(test_u64_leb128, write_u64_leb128, read_u64_leb128, u64);
  29 impl_test_unsigned_leb128!(test_u128_leb128, write_u128_leb128, read_u128_leb128, u128);
  30 impl_test_unsigned_leb128!(test_usize_leb128, write_usize_leb128, read_usize_leb128, usize);
  31
  32 #[test]
  33 fn test_signed_leb128() {
  34     let values: Vec<_> = (-500..500).map(|i| i * 0x12345789ABCDEF).collect();
  35     let mut stream = Vec::new();
  36     for &x in &values {
  37         write_signed_leb128(&mut stream, x);
  38     }
  39     let mut pos = 0;
  40     for &x in &values {
  41         let (value, bytes_read) = read_signed_leb128(&mut stream, pos);
  42         pos += bytes_read;
  43         assert_eq!(x, value);
  44     }
  45     assert_eq!(pos, stream.len());
  46 }