mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-05-09 05:57:59 +00:00
Tomasz Drwięga
99 lines
3.0 KiB
Rust
99 lines
3.0 KiB
Rust
// Copyright 2017 Parity Technologies (UK) Ltd.
|
|
// This file is part of Polkadot.
|
|
|
|
// Polkadot is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Polkadot is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Polkadot. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
//! An unsigned fixed-size integer.
|
|
|
|
use serde::{Serialize, Serializer, Deserialize, Deserializer};
|
|
|
|
use bytes;
|
|
|
|
macro_rules! impl_serde {
|
|
($name: ident, $len: expr) => {
|
|
impl Serialize for $name {
|
|
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error> where S: Serializer {
|
|
let mut bytes = [0u8; $len * 8];
|
|
self.to_big_endian(&mut bytes);
|
|
bytes::serialize_uint(&bytes, serializer)
|
|
}
|
|
}
|
|
|
|
impl<'de> Deserialize<'de> for $name {
|
|
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error> where D: Deserializer<'de> {
|
|
bytes::deserialize_check_len(deserializer, bytes::ExpectedLen::Between(0, $len * 8))
|
|
.map(|x| (&*x).into())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
construct_uint!(U256, 4);
|
|
impl_serde!(U256, 4);
|
|
construct_uint!(U512, 8);
|
|
impl_serde!(U512, 8);
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use polkadot_serializer as ser;
|
|
|
|
macro_rules! test {
|
|
($name: ident, $test_name: ident) => {
|
|
#[test]
|
|
fn $test_name() {
|
|
let tests = vec![
|
|
($name::from(0), "0x0"),
|
|
($name::from(1), "0x1"),
|
|
($name::from(2), "0x2"),
|
|
($name::from(10), "0xa"),
|
|
($name::from(15), "0xf"),
|
|
($name::from(15), "0xf"),
|
|
($name::from(16), "0x10"),
|
|
($name::from(1_000), "0x3e8"),
|
|
($name::from(100_000), "0x186a0"),
|
|
($name::from(u64::max_value()), "0xffffffffffffffff"),
|
|
($name::from(u64::max_value()) + 1.into(), "0x10000000000000000"),
|
|
];
|
|
|
|
for (number, expected) in tests {
|
|
assert_eq!(format!("{:?}", expected), ser::to_string_pretty(&number));
|
|
assert_eq!(number, ser::from_str(&format!("{:?}", expected)).unwrap());
|
|
}
|
|
|
|
// Invalid examples
|
|
assert!(ser::from_str::<$name>("\"0x\"").unwrap_err().is_data());
|
|
assert!(ser::from_str::<$name>("\"0xg\"").unwrap_err().is_data());
|
|
assert!(ser::from_str::<$name>("\"\"").unwrap_err().is_data());
|
|
assert!(ser::from_str::<$name>("\"10\"").unwrap_err().is_data());
|
|
assert!(ser::from_str::<$name>("\"0\"").unwrap_err().is_data());
|
|
}
|
|
}
|
|
}
|
|
|
|
test!(U256, test_u256);
|
|
test!(U512, test_u512);
|
|
|
|
#[test]
|
|
fn test_large_values() {
|
|
assert_eq!(
|
|
ser::to_string_pretty(&!U256::zero()),
|
|
"\"0xffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff\""
|
|
);
|
|
assert!(
|
|
ser::from_str::<U256>("\"0x1ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff\"").unwrap_err().is_data()
|
|
);
|
|
}
|
|
}
|