@@ -14,9 +14,12 @@ use crate::entities::{BlockNumber, SignedEntityTypeDiscriminants};
14
14
use crate :: signable_builder:: TransactionsImporter ;
15
15
use crate :: signed_entity_type_lock:: SignedEntityTypeLock ;
16
16
use crate :: StdResult ;
17
+ #[ cfg( test) ]
18
+ use mockall:: automock;
17
19
18
20
/// CardanoTransactionsPreloaderChecker gives the ability to determine
19
21
/// if the Cardano Transactions Preloader should import the transactions.
22
+ #[ cfg_attr( test, automock) ]
20
23
#[ async_trait]
21
24
pub trait CardanoTransactionsPreloaderChecker : Send + Sync {
22
25
/// Determine if the Cardano Transactions Preloader should preload.
@@ -149,15 +152,6 @@ mod tests {
149
152
}
150
153
}
151
154
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
-
161
155
#[ tokio:: test]
162
156
async fn call_its_inner_importer_when_is_activated ( ) {
163
157
let chain_block_number = 5000 ;
@@ -216,13 +210,18 @@ mod tests {
216
210
let mut importer = MockTransactionsImporter :: new ( ) ;
217
211
importer. expect_import ( ) . never ( ) ;
218
212
213
+ let mut preloader_checker = MockCardanoTransactionsPreloaderChecker :: new ( ) ;
214
+ preloader_checker
215
+ . expect_is_activated ( )
216
+ . returning ( || Err ( anyhow:: anyhow!( "error" ) ) ) ;
217
+
219
218
let preloader = CardanoTransactionsPreloader :: new (
220
219
Arc :: new ( SignedEntityTypeLock :: default ( ) ) ,
221
220
Arc :: new ( importer) ,
222
221
542 ,
223
222
Arc :: new ( chain_observer) ,
224
223
TestLogger :: stdout ( ) ,
225
- Arc :: new ( CardanoTransactionsPreloaderActivationWithError { } ) ,
224
+ Arc :: new ( preloader_checker ) ,
226
225
) ;
227
226
228
227
preloader
0 commit comments