blockworks-foundation
/
solana
mirror of https://github.com/blockworks-foundation/solana.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed tests

This commit is contained in:
Lijun Wang 2022-10-18 10:03:14 -07:00
parent 5fbbad991e
commit 062c7f828c
3 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tpu-client/src/nonblocking/udp_client.rs
View File

@ -73,7 +73,7 @@ mod tests {
async fn check_send_one(connection: &UdpTpuConnection, reader: &UdpSocket) {
let packet = vec![111u8; PACKET_DATA_SIZE];
connection.send_wire_transaction(&packet).await.unwrap();
connection.send_wire_transaction(&packet, &mut None).await.unwrap();
let mut packets = vec![Packet::default(); 32];
let recv = recv_mmsg(reader, &mut packets[..]).await.unwrap();
assert_eq!(1, recv);
@ -82,7 +82,7 @@ mod tests {
async fn check_send_batch(connection: &UdpTpuConnection, reader: &UdpSocket) {
let packets: Vec<_> = (0..32).map(|_| vec![0u8; PACKET_DATA_SIZE]).collect();
connection
.send_wire_transaction_batch(&packets)
.send_wire_transaction_batch(&packets, &mut None)
.await
.unwrap();
let mut packets = vec![Packet::default(); 32];

7
tpu-client/src/tpu_connection_cache.rs
View File

@ -341,7 +341,7 @@ mod tests {
use {
super::*,
crate::{
nonblocking::tpu_connection::TpuConnection as NonblockingTpuConnection,
nonblocking::tpu_connection::{SendTransactionCallbackOption, TpuConnection as NonblockingTpuConnection},
tpu_connection::TpuConnection as BlockingTpuConnection,
},
async_trait::async_trait,
@ -468,13 +468,14 @@ mod tests {
fn tpu_addr(&self) -> &SocketAddr {
&self.addr
}
async fn send_wire_transaction<T>(&self, _wire_transaction: T) -> TransportResult<()>
async fn send_wire_transaction<T>(&self, _wire_transaction: T, _callback: &mut SendTransactionCallbackOption,
) -> TransportResult<()>
where
T: AsRef<[u8]> + Send + Sync,
{
unimplemented!()
}
async fn send_wire_transaction_batch<T>(&self, _buffers: &[T]) -> TransportResult<()>
async fn send_wire_transaction_batch<T>(&self, _buffers: &[T], _callback: &mut SendTransactionCallbackOption,) -> TransportResult<()>
where
T: AsRef<[u8]> + Send + Sync,
{

2
tpu-client/tests/quic_client.rs
View File

@ -141,7 +141,7 @@ mod tests {
let num_expected_packets: usize = 3000;
let packets = vec![vec![0u8; PACKET_DATA_SIZE]; num_expected_packets];
assert!(client.send_wire_transaction_batch(&packets).await.is_ok());
assert!(client.send_wire_transaction_batch(&packets, &mut None).await.is_ok());
check_packets(receiver, num_bytes, num_expected_packets);
exit.store(true, Ordering::Relaxed);