leyoda/rqbit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix e2e streaming test

Browse source
This commit is contained in:
Ivan 2024-07-21 15:08:23 +02:00
parent 1c9aa8ca72
commit fdacb7bc16
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
crates/librqbit/src/tests/e2e_stream.rs
View file

@ -1,6 +1,7 @@
use std::{net::SocketAddr, time::Duration}; use std::{net::SocketAddr, time::Duration};
use anyhow::Context; use anyhow::Context;
use tempfile::TempDir;
use tokio::{io::AsyncReadExt, time::timeout}; use tokio::{io::AsyncReadExt, time::timeout};
use tracing::info; use tracing::info;
@ -22,7 +23,7 @@ async fn e2e_stream() -> anyhow::Result<()> {
let orig_content = std::fs::read(files.path().join("0.data")).unwrap(); let orig_content = std::fs::read(files.path().join("0.data")).unwrap();
let server_session = Session::new_with_opts( let server_session = Session::new_with_opts(
"/does-not-matter".into(), files.path().into(),
crate::SessionOptions { crate::SessionOptions {
disable_dht: true, disable_dht: true,
persistence: false, persistence: false,
@ -63,8 +64,10 @@ async fn e2e_stream() -> anyhow::Result<()> {
server_session.tcp_listen_port().unwrap(), server_session.tcp_listen_port().unwrap(),
); );
let client_dir = TempDir::with_prefix("test_e2e_stream_client")?;
let client_session = Session::new_with_opts( let client_session = Session::new_with_opts(
"/does-not-matter".into(), client_dir.path().into(),
crate::SessionOptions { crate::SessionOptions {
disable_dht: true, disable_dht: true,
persistence: false, persistence: false,