Skip to content

Commit cc7e024

Browse files
committed
Use a mock to test errors on preloader checker
1 parent daae47a commit cc7e024

File tree

2 files changed

+10
-11
lines changed

2 files changed

+10
-11
lines changed

mithril-common/src/cardano_transactions_preloader.rs

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,12 @@ use crate::entities::{BlockNumber, SignedEntityTypeDiscriminants};
1414
use crate::signable_builder::TransactionsImporter;
1515
use crate::signed_entity_type_lock::SignedEntityTypeLock;
1616
use crate::StdResult;
17+
#[cfg(test)]
18+
use mockall::automock;
1719

1820
/// CardanoTransactionsPreloaderChecker gives the ability to determine
1921
/// if the Cardano Transactions Preloader should import the transactions.
22+
#[cfg_attr(test, automock)]
2023
#[async_trait]
2124
pub trait CardanoTransactionsPreloaderChecker: Send + Sync {
2225
/// Determine if the Cardano Transactions Preloader should preload.
@@ -149,15 +152,6 @@ mod tests {
149152
}
150153
}
151154

152-
struct CardanoTransactionsPreloaderActivationWithError {}
153-
154-
#[async_trait]
155-
impl CardanoTransactionsPreloaderChecker for CardanoTransactionsPreloaderActivationWithError {
156-
async fn is_activated(&self) -> StdResult<bool> {
157-
Err(anyhow::anyhow!("error"))
158-
}
159-
}
160-
161155
#[tokio::test]
162156
async fn call_its_inner_importer_when_is_activated() {
163157
let chain_block_number = 5000;
@@ -216,13 +210,18 @@ mod tests {
216210
let mut importer = MockTransactionsImporter::new();
217211
importer.expect_import().never();
218212

213+
let mut preloader_checker = MockCardanoTransactionsPreloaderChecker::new();
214+
preloader_checker
215+
.expect_is_activated()
216+
.returning(|| Err(anyhow::anyhow!("error")));
217+
219218
let preloader = CardanoTransactionsPreloader::new(
220219
Arc::new(SignedEntityTypeLock::default()),
221220
Arc::new(importer),
222221
542,
223222
Arc::new(chain_observer),
224223
TestLogger::stdout(),
225-
Arc::new(CardanoTransactionsPreloaderActivationWithError {}),
224+
Arc::new(preloader_checker),
226225
);
227226

228227
preloader

openapi.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -532,7 +532,7 @@ paths:
532532
components:
533533
schemas:
534534
AggregatorFeaturesMessage:
535-
description: Represents general information about Mithril project, Aggregator public information and signing capabilities
535+
description: Represents general information about Aggregator public information and signing capabilities
536536
type: object
537537
additionalProperties: false
538538
required:

0 commit comments

Comments
 (0)