feat: add arbitrary to some types

unionlabs · May 2, 2024 · 871fb3b · 871fb3b
1 parent 3b4027f
commit 871fb3b
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 0 deletions.
diff --git a/tendermint/Cargo.toml b/tendermint/Cargo.toml
@@ -52,13 +52,15 @@ ed25519-consensus = { version = "2", optional = true, default-features = false }
 sha2 = { version = "0.10", optional = true, default-features = false }
 k256 = { version = "0.13", optional = true, default-features = false, features = ["alloc", "ecdsa"] }
 ripemd = { version = "0.1.3", optional = true, default-features = false }
+arbitrary = { version = "1.3.2", features = ["derive"], optional = true }
 
 [features]
 default = ["std", "rust-crypto"]
 std = ["flex-error/std", "clock"]
 clock = ["time/std"]
 secp256k1 = ["k256", "ripemd"]
 rust-crypto = ["sha2", "ed25519-consensus"]
+arbitrary = ["dep:arbitrary"]
 
 [dev-dependencies]
 k256 = { version = "0.13", default-features = false, features = ["ecdsa"] }

diff --git a/tendermint/src/account.rs b/tendermint/src/account.rs
@@ -19,6 +19,7 @@ pub const LENGTH: usize = 20;
 
 /// Account IDs
 #[derive(Copy, Clone, Eq, Hash, PartialEq, PartialOrd, Ord)]
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct Id([u8; LENGTH]); // JSON custom serialization for priv_validator_key.json
 
 impl Protobuf<Vec<u8>> for Id {}

diff --git a/tendermint/src/crypto/ed25519/verification_key.rs b/tendermint/src/crypto/ed25519/verification_key.rs
@@ -1,6 +1,7 @@
 use crate::Error;
 
 #[derive(Copy, Clone, Eq, PartialEq)]
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct VerificationKey([u8; 32]);
 
 impl core::fmt::Display for VerificationKey {

diff --git a/tendermint/src/public_key.rs b/tendermint/src/public_key.rs
@@ -31,6 +31,7 @@ use crate::{error::Error, prelude::*};
 #[derive(Copy, Clone, Debug, Eq, PartialEq, Serialize, Deserialize)]
 #[non_exhaustive]
 #[serde(tag = "type", content = "value")] // JSON custom serialization for priv_validator_key.json
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub enum PublicKey {
     /// Ed25519 keys
     #[serde(

diff --git a/tendermint/src/validator.rs b/tendermint/src/validator.rs
@@ -145,6 +145,7 @@ impl Set {
 /// Validator information
 // Todo: Remove address and make it into a function that generates it on the fly from pub_key.
 #[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct Info {
     /// Validator account address
     pub address: account::Id,
@@ -228,6 +229,7 @@ impl Info {
 // Todo: Is there more knowledge/restrictions about proposerPriority?
 /// Proposer priority
 #[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord, Default)]
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct ProposerPriority(i64);
 
 impl From<i64> for ProposerPriority {

diff --git a/tendermint/src/vote/power.rs b/tendermint/src/vote/power.rs
@@ -8,6 +8,7 @@ use crate::{error::Error, prelude::*};
 
 /// Voting power
 #[derive(Copy, Clone, Debug, Eq, PartialEq, PartialOrd, Ord, Default)]
+#[cfg_attr(feature = "arbitrary", derive(arbitrary::Arbitrary))]
 pub struct Power(u64);
 
 impl fmt::Display for Power {