openethereum/ethcore/src/blockchain/import_route.rs

135 lines
3.4 KiB
Rust
Raw Normal View History

2016-03-09 21:55:23 +01:00
// Copyright 2015, 2016 Ethcore (UK) Ltd.
// 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/>.
//! Import route.
2016-08-03 18:05:17 +02:00
use util::H256;
2016-03-09 21:55:23 +01:00
use blockchain::block_info::{BlockInfo, BlockLocation};
/// Import route for newly inserted block.
#[derive(Debug, PartialEq)]
pub struct ImportRoute {
/// Blocks that were invalidated by new block.
pub retracted: Vec<H256>,
2016-03-10 00:11:35 +01:00
/// Blocks that were validated by new block.
pub enacted: Vec<H256>,
2016-04-24 19:40:04 +02:00
/// Blocks which are neither retracted nor enacted.
2016-04-24 23:16:34 +02:00
pub omitted: Vec<H256>,
2016-03-09 21:55:23 +01:00
}
impl ImportRoute {
pub fn none() -> Self {
ImportRoute {
retracted: vec![],
enacted: vec![],
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-03-09 21:55:23 +01:00
}
}
}
impl From<BlockInfo> for ImportRoute {
fn from(info: BlockInfo) -> ImportRoute {
match info.location {
BlockLocation::CanonChain => ImportRoute {
retracted: vec![],
enacted: vec![info.hash],
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-04-24 19:40:04 +02:00
},
BlockLocation::Branch => ImportRoute {
retracted: vec![],
enacted: vec![],
2016-04-24 23:16:34 +02:00
omitted: vec![info.hash],
2016-03-09 21:55:23 +01:00
},
2016-04-17 17:18:25 +02:00
BlockLocation::BranchBecomingCanonChain(mut data) => {
data.enacted.push(info.hash);
2016-03-09 21:55:23 +01:00
ImportRoute {
2016-04-17 17:18:25 +02:00
retracted: data.retracted,
enacted: data.enacted,
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-03-09 21:55:23 +01:00
}
}
}
}
}
#[cfg(test)]
mod tests {
2016-08-03 18:05:17 +02:00
use util::{U256, H256};
2016-04-17 17:18:25 +02:00
use blockchain::block_info::{BlockInfo, BlockLocation, BranchBecomingCanonChainData};
2016-03-09 21:55:23 +01:00
use blockchain::ImportRoute;
#[test]
fn import_route_none() {
assert_eq!(ImportRoute::none(), ImportRoute {
enacted: vec![],
retracted: vec![],
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-03-09 21:55:23 +01:00
});
}
#[test]
fn import_route_branch() {
let info = BlockInfo {
hash: H256::from(U256::from(1)),
number: 0,
total_difficulty: U256::from(0),
location: BlockLocation::Branch,
};
2016-04-24 19:40:04 +02:00
assert_eq!(ImportRoute::from(info), ImportRoute {
retracted: vec![],
enacted: vec![],
2016-04-24 23:16:34 +02:00
omitted: vec![H256::from(U256::from(1))],
2016-04-24 19:40:04 +02:00
});
2016-03-09 21:55:23 +01:00
}
#[test]
fn import_route_canon_chain() {
let info = BlockInfo {
hash: H256::from(U256::from(1)),
number: 0,
total_difficulty: U256::from(0),
location: BlockLocation::CanonChain,
};
assert_eq!(ImportRoute::from(info), ImportRoute {
retracted: vec![],
enacted: vec![H256::from(U256::from(1))],
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-03-09 21:55:23 +01:00
});
}
#[test]
fn import_route_branch_becoming_canon_chain() {
let info = BlockInfo {
hash: H256::from(U256::from(2)),
number: 0,
total_difficulty: U256::from(0),
2016-04-17 17:18:25 +02:00
location: BlockLocation::BranchBecomingCanonChain(BranchBecomingCanonChainData {
ancestor: H256::from(U256::from(0)),
enacted: vec![H256::from(U256::from(1))],
retracted: vec![H256::from(U256::from(3)), H256::from(U256::from(4))],
2016-04-17 17:18:25 +02:00
})
2016-03-09 21:55:23 +01:00
};
assert_eq!(ImportRoute::from(info), ImportRoute {
retracted: vec![H256::from(U256::from(3)), H256::from(U256::from(4))],
enacted: vec![H256::from(U256::from(1)), H256::from(U256::from(2))],
2016-04-24 23:16:34 +02:00
omitted: vec![],
2016-03-09 21:55:23 +01:00
});
}
}