@@ -400,7 +400,7 @@ pub async fn process<W: Workflows>(
400
400
mod tests {
401
401
use super :: * ;
402
402
use crate :: bb02_async:: block_on;
403
- use crate :: workflow:: RealWorkflows ;
403
+ use crate :: workflow:: testing :: { Screen , TestingWorkflows } ;
404
404
use alloc:: boxed:: Box ;
405
405
use bitbox02:: testing:: { mock, mock_unlocked, Data } ;
406
406
use util:: bip32:: HARDENED ;
@@ -476,7 +476,7 @@ mod tests {
476
476
477
477
fn do_pkh_skh ( keypath_payment : & [ u32 ] , keypath_stake : & [ u32 ] ) -> Result < Response , Error > {
478
478
block_on ( process (
479
- & mut RealWorkflows ,
479
+ & mut TestingWorkflows :: new ( ) ,
480
480
& pb:: CardanoAddressRequest {
481
481
network : CardanoNetwork :: CardanoMainnet as _ ,
482
482
display : false ,
@@ -565,19 +565,11 @@ mod tests {
565
565
fn test_process_confirm ( ) {
566
566
const EXPECTED : & str = "addr1q90tlskd4mh5kncmul7vx887j30tjtfgvap5n0g0rf9qqc7znmndrdhe7rwvqkw5c7mqnp4a3yflnvu6kff7l5dungvqmvu6hs" ;
567
567
568
- mock ( Data {
569
- ui_confirm_create : Some ( Box :: new ( |params| {
570
- assert_eq ! ( params. title, "Cardano" ) ;
571
- assert_eq ! ( params. body, EXPECTED ) ;
572
- true
573
- } ) ) ,
574
- ..Default :: default ( )
575
- } ) ;
576
568
mock_unlocked ( ) ;
577
-
569
+ let mut mock_workflows = TestingWorkflows :: new ( ) ;
578
570
assert_eq ! (
579
571
block_on( process(
580
- & mut RealWorkflows ,
572
+ & mut mock_workflows ,
581
573
& pb:: CardanoAddressRequest {
582
574
network: CardanoNetwork :: CardanoMainnet as _,
583
575
display: true ,
@@ -591,6 +583,14 @@ mod tests {
591
583
r#pub: EXPECTED . into( )
592
584
} ) )
593
585
) ;
586
+ assert_eq ! (
587
+ mock_workflows. screens,
588
+ vec![ Screen :: Confirm {
589
+ title: "Cardano" . into( ) ,
590
+ body: EXPECTED . into( ) ,
591
+ longtouch: false ,
592
+ } , ]
593
+ ) ;
594
594
}
595
595
596
596
#[ test]
0 commit comments