Skip to content

Commit f4ddee9

Browse files
committed
test: handle error from failed connection
1 parent 634fe12 commit f4ddee9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -953,10 +953,12 @@ pub mod tests {
953953
.unwrap()
954954
.connect_timeout(Duration::default());
955955

956+
956957
let sender_aggregator = TapAggregatorClient::connect(endpoint)
957958
.await
958-
.unwrap()
959-
.send_compressed(tonic::codec::CompressionEncoding::Zstd);
959+
.expect("Failed to connect to the TapAggregator endpoint")
960+
.send_compressed(tonic::codec::CompressionEncoding::Zstd);
961+
960962
SenderAllocationArgs {
961963
pgpool: pgpool.clone(),
962964
allocation_id: ALLOCATION_ID_0,
@@ -1255,7 +1257,6 @@ pub mod tests {
12551257
SenderAccountMessage::UpdateReceiptFees(_, ReceiptFees::RavRequestResponse(_))
12561258
));
12571259

1258-
// Stop the TAP aggregator server.
12591260
}
12601261

12611262
#[sqlx::test(migrations = "../../migrations")]

0 commit comments

Comments
 (0)