Skip to content

Commit 6aa7024

Browse files
committed
test: re,pve fixtuer from test
1 parent 2879bc2 commit 6aa7024

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

crates/tap-agent/src/agent/sender_account.rs

+11-12
Original file line numberDiff line numberDiff line change
@@ -1727,41 +1727,40 @@ pub mod tests {
17271727
assert_not_triggered!(&triggered_rav_request);
17281728
}
17291729

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+
17341735
// create a fake sender allocation
17351736
let (triggered_rav_request, _, _) = create_mock_sender_allocation(
1736-
basic_sender_account.prefix,
1737+
prefix,
17371738
SENDER.1,
17381739
ALLOCATION_ID_0,
1739-
basic_sender_account.sender_account.clone(),
1740+
sender_account.clone(),
17401741
)
17411742
.await;
17421743

1743-
basic_sender_account
1744-
.sender_account
1744+
sender_account
17451745
.cast(SenderAccountMessage::UpdateReceiptFees(
17461746
ALLOCATION_ID_0,
17471747
ReceiptFees::NewReceipt(TRIGGER_VALUE, get_current_timestamp_u64_ns()),
17481748
))
17491749
.unwrap();
17501750

1751-
flush_messages(&basic_sender_account.notify).await;
1751+
flush_messages(&notify).await;
17521752
assert_not_triggered!(&triggered_rav_request);
17531753

17541754
// wait for it to be outside buffer
17551755
tokio::time::sleep(BUFFER_DURATION).await;
17561756

1757-
basic_sender_account
1758-
.sender_account
1757+
sender_account
17591758
.cast(SenderAccountMessage::UpdateReceiptFees(
17601759
ALLOCATION_ID_0,
17611760
ReceiptFees::Retry,
17621761
))
17631762
.unwrap();
1764-
flush_messages(&basic_sender_account.notify).await;
1763+
flush_messages(&notify).await;
17651764

17661765
assert_triggered!(&triggered_rav_request);
17671766
}

0 commit comments

Comments
 (0)