@@ -1727,41 +1727,40 @@ pub mod tests {
1727
1727
assert_not_triggered ! ( & triggered_rav_request) ;
1728
1728
}
1729
1729
1730
- #[ rstest:: rstest]
1731
- async fn test_update_receipt_fees_trigger_rav (
1732
- #[ future( awt) ] basic_sender_account : TestSenderAccount ,
1733
- ) {
1730
+ #[ sqlx:: test( migrations = "../../migrations" ) ]
1731
+ async fn test_update_receipt_fees_trigger_rav ( pgpool : PgPool ) {
1732
+ let ( sender_account, notify, prefix, _) =
1733
+ create_sender_account ( ) . pgpool ( pgpool) . call ( ) . await ;
1734
+
1734
1735
// create a fake sender allocation
1735
1736
let ( triggered_rav_request, _, _) = create_mock_sender_allocation (
1736
- basic_sender_account . prefix ,
1737
+ prefix,
1737
1738
SENDER . 1 ,
1738
1739
ALLOCATION_ID_0 ,
1739
- basic_sender_account . sender_account . clone ( ) ,
1740
+ sender_account. clone ( ) ,
1740
1741
)
1741
1742
. await ;
1742
1743
1743
- basic_sender_account
1744
- . sender_account
1744
+ sender_account
1745
1745
. cast ( SenderAccountMessage :: UpdateReceiptFees (
1746
1746
ALLOCATION_ID_0 ,
1747
1747
ReceiptFees :: NewReceipt ( TRIGGER_VALUE , get_current_timestamp_u64_ns ( ) ) ,
1748
1748
) )
1749
1749
. unwrap ( ) ;
1750
1750
1751
- flush_messages ( & basic_sender_account . notify ) . await ;
1751
+ flush_messages ( & notify) . await ;
1752
1752
assert_not_triggered ! ( & triggered_rav_request) ;
1753
1753
1754
1754
// wait for it to be outside buffer
1755
1755
tokio:: time:: sleep ( BUFFER_DURATION ) . await ;
1756
1756
1757
- basic_sender_account
1758
- . sender_account
1757
+ sender_account
1759
1758
. cast ( SenderAccountMessage :: UpdateReceiptFees (
1760
1759
ALLOCATION_ID_0 ,
1761
1760
ReceiptFees :: Retry ,
1762
1761
) )
1763
1762
. unwrap ( ) ;
1764
- flush_messages ( & basic_sender_account . notify ) . await ;
1763
+ flush_messages ( & notify) . await ;
1765
1764
1766
1765
assert_triggered ! ( & triggered_rav_request) ;
1767
1766
}
0 commit comments