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	pub BlockDownloader<Block: BlockT> {}
31
32	#[async_trait::async_trait]
33	impl<Block: BlockT> BlockDownloaderT<Block> for BlockDownloader<Block> {
34		async fn download_blocks(
35			&self,
36			who: PeerId,
37			request: BlockRequest<Block>,
38		) -> Result<Result<(Vec<u8>, ProtocolName), RequestFailure>, oneshot::Canceled>;
39		fn block_response_into_blocks(
40			&self,
41			request: &BlockRequest<Block>,
42			response: Vec<u8>,
43		) -> Result<Vec<BlockData<Block>>, BlockResponseError>;
44	}
45}
sc_network_sync/mock.rs

sc_network_sync/
mock.rs
