Add test cases for verification of reportDoubleVote

Author: HoOngEe

core/src/client/test_client.rs

@@ -36,9 +36,10 @@ use std::ops::Range;
 use std::sync::atomic::{AtomicUsize, Ordering as AtomicOrder};
 use std::sync::Arc;
 
-use ckey::{public_to_address, Address, Generator, NetworkId, PlatformAddress, Public, Random};
+use ckey::{public_to_address, Address, Generator, KeyPair, NetworkId, PlatformAddress, Private, Public, Random};
 use cmerkle::skewed_merkle_root;
 use cnetwork::NodeId;
+use cstate::tests::helpers::empty_top_state;
 use cstate::{FindActionHandler, StateDB, TopLevelState};
 use ctimer::{TimeoutHandler, TimerToken};
 use ctypes::transaction::{Action, Transaction};
@@ -58,6 +59,7 @@ use crate::client::{
     AccountData, BlockChainClient, BlockChainTrait, BlockProducer, BlockStatus, EngineInfo, ImportBlock,
     MiningBlockChainClient, StateInfo, StateOrBlock, TermInfo,
 };
+use crate::consensus::stake::{Validator, Validators};
 use crate::consensus::EngineError;
 use crate::db::{COL_STATE, NUM_COLUMNS};
 use crate::encoded;
@@ -102,6 +104,10 @@ pub struct TestBlockChainClient {
     pub history: RwLock<Option<u64>>,
     /// Term ID
     pub term_id: Option<u64>,
+    /// Fixed validator keys
+    pub validator_keys: RwLock<HashMap<Public, Private>>,
+    /// Fixed validators
+    pub validators: Validators,
 }
 
 impl Default for TestBlockChainClient {
@@ -154,6 +160,8 @@ impl TestBlockChainClient {
             latest_block_timestamp: RwLock::new(10_000_000),
             history: RwLock::new(None),
             term_id: None,
+            validator_keys: RwLock::new(HashMap::new()),
+            validators: Validators::from_vector_to_test(vec![]),
         };
 
         // insert genesis hash.
@@ -308,6 +316,26 @@ impl TestBlockChainClient {
     pub fn set_history(&self, h: Option<u64>) {
         *self.history.write() = h;
     }
+
+    /// Set validators which can be brought from state.
+    pub fn set_random_validators(&mut self, count: usize) {
+        let mut pubkeys: Vec<Public> = vec![];
+        for _ in 0..count {
+            let random_priv_key = Private::from(H256::random());
+            let key_pair = KeyPair::from_private(random_priv_key).unwrap();
+            self.validator_keys.write().insert(*key_pair.public(), *key_pair.private());
+            pubkeys.push(*key_pair.public());
+        }
+        let fixed_validators: Validators = Validators::from_vector_to_test(
+            pubkeys.into_iter().map(|pubkey| Validator::new_for_test(0, 0, pubkey)).collect(),
+        );
+
+        self.validators = fixed_validators;
+    }
+
+    pub fn get_validators(&self) -> &Validators {
+        &self.validators
+    }
 }
 
 pub fn get_temp_state_db() -> StateDB {
@@ -627,6 +655,10 @@ impl TermInfo for TestBlockChainClient {
 
 impl StateInfo for TestBlockChainClient {
     fn state_at(&self, _id: BlockId) -> Option<TopLevelState> {
-        None
+        let statedb = StateDB::new_with_memorydb();
+        let mut top_state = empty_top_state(statedb);
+        let _ = self.validators.save_to_state(&mut top_state);
+
+        Some(top_state)
     }
 }

core/src/consensus/mod.rs

@@ -21,7 +21,7 @@ mod null_engine;
 mod signer;
 mod simple_poa;
 mod solo;
-mod stake;
+pub mod stake;
 mod tendermint;
 mod validator_set;
 mod vote_collector;
@@ -31,7 +31,10 @@ pub use self::cuckoo::Cuckoo;
 pub use self::null_engine::NullEngine;
 pub use self::simple_poa::SimplePoA;
 pub use self::solo::Solo;
-pub use self::tendermint::{Tendermint, TendermintParams, TimeGapParams};
+pub use self::tendermint::{
+    message_info_rlp, ConsensusMessage, Height, Step, Tendermint, TendermintParams, TimeGapParams, View, VoteOn,
+    VoteStep,
+};
 pub use self::validator_set::validator_list::RoundRobinValidator;
 pub use self::validator_set::ValidatorSet;
 pub use self::vote_collector::Message;
@@ -49,7 +52,6 @@ use ctypes::{CommonParams, Header};
 use primitives::{Bytes, H256, U256};
 
 use self::bit_set::BitSet;
-use self::tendermint::types::View;
 use crate::account_provider::AccountProvider;
 use crate::block::{ExecutedBlock, SealedBlock};
 use crate::client::ConsensusClient;

core/src/consensus/stake/action_data.rs

@@ -238,6 +238,15 @@ pub struct Validator {
 }
 
 impl Validator {
+    pub fn new_for_test(delegation: StakeQuantity, deposit: Deposit, pubkey: Public) -> Self {
+        Self {
+            weight: delegation,
+            delegation,
+            deposit,
+            pubkey,
+        }
+    }
+
     fn new(delegation: StakeQuantity, deposit: Deposit, pubkey: Public) -> Self {
         Self {
             weight: delegation,
@@ -263,6 +272,10 @@ impl Validator {
 #[derive(Debug)]
 pub struct Validators(Vec<Validator>);
 impl Validators {
+    pub fn from_vector_to_test(vec: Vec<Validator>) -> Self {
+        Validators(vec)
+    }
+
     pub fn load_from_state(state: &TopLevelState) -> StateResult<Self> {
         let key = &*VALIDATORS_KEY;
         let validators = state.action_data(&key)?.map(|data| decode_list(&data)).unwrap_or_default();