sc_network_sync/
mock.rs

1// This file is part of Substrate.
2
3// Copyright (C) Parity Technologies (UK) Ltd.
4// SPDX-License-Identifier: GPL-3.0-or-later WITH Classpath-exception-2.0
5
6// This program is free software: you can redistribute it and/or modify
7// it under the terms of the GNU General Public License as published by
8// the Free Software Foundation, either version 3 of the License, or
9// (at your option) any later version.
10
11// This program is distributed in the hope that it will be useful,
12// but WITHOUT ANY WARRANTY; without even the implied warranty of
13// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14// GNU General Public License for more details.
15
16// You should have received a copy of the GNU General Public License
17// along with this program. If not, see <https://www.gnu.org/licenses/>.
18
19//! Contains mock implementations of `ChainSync` and 'BlockDownloader'.
20
21use crate::block_relay_protocol::{BlockDownloader as BlockDownloaderT, BlockResponseError};
22
23use futures::channel::oneshot;
24use sc_network::{ProtocolName, RequestFailure};
25use sc_network_common::sync::message::{BlockData, BlockRequest};
26use sc_network_types::PeerId;
27use sp_runtime::traits::Block as BlockT;
28
29mockall::mock! {
30	#[derive(Debug)]
31	pub BlockDownloader<Block: BlockT> {}
32
33	#[async_trait::async_trait]
34	impl<Block: BlockT> BlockDownloaderT<Block> for BlockDownloader<Block> {
35		fn protocol_name(&self) -> &ProtocolName;
36
37		async fn download_blocks(
38			&self,
39			who: PeerId,
40			request: BlockRequest<Block>,
41		) -> Result<Result<(Vec<u8>, ProtocolName), RequestFailure>, oneshot::Canceled>;
42		fn block_response_into_blocks(
43			&self,
44			request: &BlockRequest<Block>,
45			response: Vec<u8>,
46		) -> Result<Vec<BlockData<Block>>, BlockResponseError>;
47	}
48}
sc_network_sync/mock.rs

sc_network_sync/
mock.rs
