8cb6368fe6
Lets the user annotate fields of a struct to skip for signed root hashing. Also added tests in a `eth2/utils/tests` crate, so that we can test whether these derived macros work as intended.
95 lines
2.4 KiB
Rust
95 lines
2.4 KiB
Rust
use ssz::{SignedRoot, TreeHash};
|
|
use ssz_derive::{SignedRoot, TreeHash};
|
|
|
|
#[derive(TreeHash, SignedRoot)]
|
|
struct CryptoKitties {
|
|
best_kitty: u64,
|
|
worst_kitty: u8,
|
|
kitties: Vec<u32>,
|
|
}
|
|
|
|
impl CryptoKitties {
|
|
fn new() -> Self {
|
|
CryptoKitties {
|
|
best_kitty: 9999,
|
|
worst_kitty: 1,
|
|
kitties: vec![2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43],
|
|
}
|
|
}
|
|
|
|
fn hash(&self) -> Vec<u8> {
|
|
let mut list: Vec<Vec<u8>> = Vec::new();
|
|
list.push(self.best_kitty.hash_tree_root());
|
|
list.push(self.worst_kitty.hash_tree_root());
|
|
list.push(self.kitties.hash_tree_root());
|
|
ssz::merkle_hash(&mut list)
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_cryptokitties_hash() {
|
|
let kitties = CryptoKitties::new();
|
|
let expected_hash = vec![
|
|
201, 9, 139, 14, 24, 247, 21, 55, 132, 211, 51, 125, 183, 186, 177, 33, 147, 210, 42, 108,
|
|
174, 162, 221, 227, 157, 179, 15, 7, 97, 239, 82, 220,
|
|
];
|
|
assert_eq!(kitties.hash(), expected_hash);
|
|
}
|
|
|
|
#[test]
|
|
fn test_simple_tree_hash_derive() {
|
|
let kitties = CryptoKitties::new();
|
|
assert_eq!(kitties.hash_tree_root(), kitties.hash());
|
|
}
|
|
|
|
#[test]
|
|
fn test_simple_signed_root_derive() {
|
|
let kitties = CryptoKitties::new();
|
|
assert_eq!(kitties.signed_root(), kitties.hash());
|
|
}
|
|
|
|
#[derive(TreeHash, SignedRoot)]
|
|
struct Casper {
|
|
friendly: bool,
|
|
#[tree_hash(skip_hashing)]
|
|
friends: Vec<u32>,
|
|
#[signed_root(skip_hashing)]
|
|
dead: bool,
|
|
}
|
|
|
|
impl Casper {
|
|
fn new() -> Self {
|
|
Casper {
|
|
friendly: true,
|
|
friends: vec![1, 2, 3, 4, 5, 6, 7, 8, 9, 10],
|
|
dead: true,
|
|
}
|
|
}
|
|
|
|
fn expected_signed_hash(&self) -> Vec<u8> {
|
|
let mut list = Vec::new();
|
|
list.push(self.friendly.hash_tree_root());
|
|
list.push(self.friends.hash_tree_root());
|
|
ssz::merkle_hash(&mut list)
|
|
}
|
|
|
|
fn expected_tree_hash(&self) -> Vec<u8> {
|
|
let mut list = Vec::new();
|
|
list.push(self.friendly.hash_tree_root());
|
|
list.push(self.dead.hash_tree_root());
|
|
ssz::merkle_hash(&mut list)
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_annotated_tree_hash_derive() {
|
|
let casper = Casper::new();
|
|
assert_eq!(casper.hash_tree_root(), casper.expected_tree_hash());
|
|
}
|
|
|
|
#[test]
|
|
fn test_annotated_signed_root_derive() {
|
|
let casper = Casper::new();
|
|
assert_eq!(casper.signed_root(), casper.expected_signed_hash());
|
|
}
|