openethereum/rpc/src/v1/tests/helpers/sync_provider.rs

127 lines
3.2 KiB
Rust
Raw Normal View History

// Copyright 2015-2017 Parity Technologies (UK) Ltd.
2016-03-11 10:17:20 +01:00
// This file is part of Parity.
// Parity 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.
// Parity 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 Parity. If not, see <http://www.gnu.org/licenses/>.
2016-03-18 19:16:46 +01:00
//! Test implementation of SyncProvider.
2016-11-16 13:37:21 +01:00
use std::collections::BTreeMap;
use ethereum_types::H256;
use parking_lot::RwLock;
2018-04-10 12:13:49 +02:00
use sync::{SyncProvider, EthProtocolInfo, SyncStatus, SyncState, PeerInfo, TransactionStats};
2016-03-11 10:17:20 +01:00
2016-03-18 19:16:46 +01:00
/// TestSyncProvider config.
2016-03-11 10:17:20 +01:00
pub struct Config {
2016-03-18 19:16:46 +01:00
/// Protocol version.
2016-12-05 15:54:31 +01:00
pub network_id: u64,
2016-03-18 19:16:46 +01:00
/// Number of peers.
2016-03-11 10:17:20 +01:00
pub num_peers: usize,
}
2016-03-18 19:16:46 +01:00
/// Test sync provider.
2016-03-11 10:17:20 +01:00
pub struct TestSyncProvider {
2016-03-18 19:16:46 +01:00
/// Sync status.
2016-03-16 10:37:08 +01:00
pub status: RwLock<SyncStatus>,
2016-03-11 10:17:20 +01:00
}
impl TestSyncProvider {
/// Creates new sync provider.
2016-03-11 10:17:20 +01:00
pub fn new(config: Config) -> Self {
TestSyncProvider {
2016-03-16 10:37:08 +01:00
status: RwLock::new(SyncStatus {
2016-05-16 14:41:41 +02:00
state: SyncState::Idle,
network_id: config.network_id,
protocol_version: 63,
2016-03-11 10:17:20 +01:00
start_block_number: 0,
last_imported_block_number: None,
highest_block_number: None,
blocks_total: 0,
blocks_received: 0,
num_peers: config.num_peers,
num_active_peers: 0,
mem_used: 0,
num_snapshot_chunks: 0,
snapshot_chunks_done: 0,
last_imported_old_block_number: None,
2016-03-17 11:23:30 +01:00
}),
2016-03-11 10:17:20 +01:00
}
}
/// Simulate importing blocks.
pub fn increase_imported_block_number(&self, count: u64) {
let mut status = self.status.write();
let current_number = status.last_imported_block_number.unwrap_or(0);
status.last_imported_block_number = Some(current_number + count);
}
2016-03-11 10:17:20 +01:00
}
impl SyncProvider for TestSyncProvider {
fn status(&self) -> SyncStatus {
self.status.read().clone()
2016-03-11 10:17:20 +01:00
}
fn peers(&self) -> Vec<PeerInfo> {
vec![
PeerInfo {
id: Some("node1".to_owned()),
client_version: "Parity/1".to_owned(),
2016-11-16 13:37:21 +01:00
capabilities: vec!["eth/62".to_owned(), "eth/63".to_owned()],
remote_address: "127.0.0.1:7777".to_owned(),
local_address: "127.0.0.1:8888".to_owned(),
eth_info: Some(EthProtocolInfo {
version: 62,
difficulty: Some(40.into()),
head: 50.into(),
}),
2017-03-22 16:45:50 +01:00
pip_info: None,
},
PeerInfo {
id: None,
client_version: "Parity/2".to_owned(),
2016-11-16 13:37:21 +01:00
capabilities: vec!["eth/63".to_owned(), "eth/64".to_owned()],
remote_address: "Handshake".to_owned(),
local_address: "127.0.0.1:3333".to_owned(),
eth_info: Some(EthProtocolInfo {
version: 64,
difficulty: None,
head: 60.into()
}),
2017-03-22 16:45:50 +01:00
pip_info: None,
}
]
}
fn enode(&self) -> Option<String> {
None
}
2016-11-16 13:37:21 +01:00
fn transactions_stats(&self) -> BTreeMap<H256, TransactionStats> {
map![
1.into() => TransactionStats {
first_seen: 10,
propagated_to: map![
128.into() => 16
],
2016-11-16 13:37:21 +01:00
},
5.into() => TransactionStats {
first_seen: 16,
propagated_to: map![
16.into() => 1
],
2016-11-16 13:37:21 +01:00
}
]
}
2016-03-11 10:17:20 +01:00
}