zksync_consensus_network 0.13.0

ZKsync consensus network component
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

//! Random message generators for testing.
//! Implementations of Distribution are supposed to generate realistic data,
//! but in fact they are "best-effort realistic" - they might need an upgrade,
//! if tests require stricter properties of the generated data.
use std::sync::Arc;

use rand::{
    distributions::{Distribution, Standard},
    Rng,
};
use zksync_consensus_roles::validator;

use crate::rpc;

impl Distribution<rpc::consensus::Req> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::consensus::Req {
        rpc::consensus::Req(rng.gen())
    }
}

impl Distribution<rpc::consensus::Resp> for Standard {
    fn sample<R: Rng + ?Sized>(&self, _rng: &mut R) -> rpc::consensus::Resp {
        rpc::consensus::Resp
    }
}

impl Distribution<rpc::push_validator_addrs::Req> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::push_validator_addrs::Req {
        let n = rng.gen_range(5..10);
        rpc::push_validator_addrs::Req(
            (0..n)
                .map(|_| {
                    let key: validator::SecretKey = rng.gen();
                    let addr: validator::NetAddress = rng.gen();
                    Arc::new(key.sign_msg(addr))
                })
                .collect(),
        )
    }
}

impl Distribution<rpc::push_tx::Req> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::push_tx::Req {
        rpc::push_tx::Req(rng.gen())
    }
}

impl Distribution<rpc::push_block_store_state::Req> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::push_block_store_state::Req {
        rpc::push_block_store_state::Req { state: rng.gen() }
    }
}

impl Distribution<rpc::get_block::Req> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::get_block::Req {
        rpc::get_block::Req(rng.gen())
    }
}

impl Distribution<rpc::get_block::Resp> for Standard {
    fn sample<R: Rng + ?Sized>(&self, rng: &mut R) -> rpc::get_block::Resp {
        rpc::get_block::Resp(Some(rng.gen()))
    }
}