mirror of
https://github.com/pezkuwichain/pezkuwi-subxt.git
synced 2026-05-01 04:17:57 +00:00
Wei Tang
ac1be0665e
* Support custom fork choice rule * Remove unneeded reexport * Fix network compile
91 lines
2.9 KiB
Rust
91 lines
2.9 KiB
Rust
// Copyright 2018 Parity Technologies (UK) Ltd.
|
|
// This file is part of Substrate.
|
|
|
|
// Substrate is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// Substrate is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with Substrate. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
//! Client extension for tests.
|
|
|
|
use client::{self, Client};
|
|
use consensus::{ImportBlock, BlockImport, BlockOrigin, Error as ConsensusError, ForkChoiceStrategy};
|
|
use runtime_primitives::Justification;
|
|
use runtime_primitives::generic::BlockId;
|
|
use primitives::Blake2Hasher;
|
|
use runtime;
|
|
|
|
/// Extension trait for a test client.
|
|
pub trait TestClient: Sized {
|
|
/// Import block to the chain. No finality.
|
|
fn import(&self, origin: BlockOrigin, block: runtime::Block)
|
|
-> Result<(), ConsensusError>;
|
|
|
|
/// Import block with justification, finalizes block.
|
|
fn import_justified(&self, origin: BlockOrigin, block: runtime::Block, justification: Justification)
|
|
-> Result<(), ConsensusError>;
|
|
|
|
/// Finalize a block.
|
|
fn finalize_block(&self, id: BlockId<runtime::Block>, justification: Option<Justification>) -> client::error::Result<()>;
|
|
|
|
/// Returns hash of the genesis block.
|
|
fn genesis_hash(&self) -> runtime::Hash;
|
|
}
|
|
|
|
impl<B, E, RA> TestClient for Client<B, E, runtime::Block, RA>
|
|
where
|
|
B: client::backend::Backend<runtime::Block, Blake2Hasher>,
|
|
E: client::CallExecutor<runtime::Block, Blake2Hasher>,
|
|
Self: BlockImport<runtime::Block, Error=ConsensusError>,
|
|
{
|
|
fn import(&self, origin: BlockOrigin, block: runtime::Block)
|
|
-> Result<(), ConsensusError>
|
|
{
|
|
let import = ImportBlock {
|
|
origin,
|
|
header: block.header,
|
|
justification: None,
|
|
post_digests: vec![],
|
|
body: Some(block.extrinsics),
|
|
finalized: false,
|
|
auxiliary: Vec::new(),
|
|
fork_choice: ForkChoiceStrategy::LongestChain,
|
|
};
|
|
|
|
self.import_block(import, None).map(|_| ())
|
|
}
|
|
|
|
fn import_justified(&self, origin: BlockOrigin, block: runtime::Block, justification: Justification)
|
|
-> Result<(), ConsensusError>
|
|
{
|
|
let import = ImportBlock {
|
|
origin,
|
|
header: block.header,
|
|
justification: Some(justification),
|
|
post_digests: vec![],
|
|
body: Some(block.extrinsics),
|
|
finalized: true,
|
|
auxiliary: Vec::new(),
|
|
fork_choice: ForkChoiceStrategy::LongestChain,
|
|
};
|
|
|
|
self.import_block(import, None).map(|_| ())
|
|
}
|
|
|
|
fn finalize_block(&self, id: BlockId<runtime::Block>, justification: Option<Justification>) -> client::error::Result<()> {
|
|
self.finalize_block(id, justification, true)
|
|
}
|
|
|
|
fn genesis_hash(&self) -> runtime::Hash {
|
|
self.block_hash(0).unwrap().unwrap()
|
|
}
|
|
}
|