diff --git a/src/hydrabadger/mod.rs b/src/hydrabadger/mod.rs index 5fab6f6..f45d171 100644 --- a/src/hydrabadger/mod.rs +++ b/src/hydrabadger/mod.rs @@ -3,7 +3,7 @@ mod hydrabadger; mod state; use self::handler::Handler; -use self::state::{State, StateDsct}; +use self::state::State; use bincode; use hbbft::{ dynamic_honey_badger::Error as DhbError, @@ -14,6 +14,7 @@ use std; use {Input, Message, Uid}; pub use self::hydrabadger::{Config, Hydrabadger}; +pub use self::state::StateDsct; // Number of times to attempt wire message re-send. pub const WIRE_MESSAGE_RETRY_MAX: usize = 10; diff --git a/src/hydrabadger/state.rs b/src/hydrabadger/state.rs index fc3c4b0..c80cf0c 100644 --- a/src/hydrabadger/state.rs +++ b/src/hydrabadger/state.rs @@ -19,7 +19,7 @@ use rand; use {Contribution, Input, Message, NetworkNodeInfo, NetworkState, Step, Uid}; /// A `State` discriminant. -#[derive(Copy, Clone, Debug)] +#[derive(Copy, Clone, Debug, PartialEq, Eq)] pub enum StateDsct { Disconnected, DeterminingNetworkState, diff --git a/src/lib.rs b/src/lib.rs index a2a857e..cb2ec08 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -72,6 +72,7 @@ pub use hydrabadger::{Config, Hydrabadger}; // TODO: Create a separate, library-wide error type. pub use hydrabadger::Error; pub use hbbft::dynamic_honey_badger::Batch; +pub use hydrabadger::StateDsct; /// Transmit half of the wire message channel. // TODO: Use a bounded tx/rx (find a sensible upper bound):