Rename project from xserial to pipeview
Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus <clio-agent@sisyphuslabs.ai>
This commit is contained in:
334
crates/pipeview-client/tests/session_lifecycle.rs
Normal file
334
crates/pipeview-client/tests/session_lifecycle.rs
Normal file
@@ -0,0 +1,334 @@
|
||||
use std::time::Duration;
|
||||
|
||||
use tokio::io::{AsyncReadExt, AsyncWriteExt};
|
||||
use tokio::net::TcpListener;
|
||||
|
||||
use pipeview_client::SessionManager;
|
||||
use pipeview_client::config::{DecoderConfig, FramerConfig, PipelineConfig, SessionConfig};
|
||||
use pipeview_client::session::{Session, SessionEvent};
|
||||
use pipeview_core::protocol::DecodedData;
|
||||
use pipeview_core::transport::TransportConfig;
|
||||
|
||||
fn tcp_config(addr: String) -> SessionConfig {
|
||||
SessionConfig {
|
||||
transport: TransportConfig::Tcp { addr },
|
||||
pipelines: vec![PipelineConfig {
|
||||
name: "text".into(),
|
||||
framer: FramerConfig::Line {
|
||||
strip_cr: true,
|
||||
max_line_len: 1024 * 1024,
|
||||
},
|
||||
decoder: DecoderConfig::Text {
|
||||
encoding: pipeview_core::protocol::text::TextEncoding::Utf8,
|
||||
},
|
||||
}],
|
||||
history_limit: 100,
|
||||
auto_reconnect: false,
|
||||
}
|
||||
}
|
||||
|
||||
// ── Session spawn + basic send/read ──────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_echo_send_and_read() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
let mut buf = [0u8; 256];
|
||||
let n = stream.read(&mut buf).await.unwrap();
|
||||
assert_eq!(&buf[..n], b"ping\n");
|
||||
stream.write_all(b"pong\n").await.unwrap();
|
||||
});
|
||||
|
||||
let handle = Session::spawn(1, tcp_config(addr));
|
||||
handle.send(b"ping\n".to_vec()).await.unwrap();
|
||||
|
||||
let entry = handle.read(5000).await.expect("should receive response");
|
||||
match entry.data {
|
||||
DecodedData::Text(s) => assert_eq!(s, "pong"),
|
||||
other => panic!("expected Text, got {:?}", other),
|
||||
}
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server.await.unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_multiple_reads() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
stream.write_all(b"a\nb\nc\n").await.unwrap();
|
||||
});
|
||||
|
||||
let handle = Session::spawn(2, tcp_config(addr));
|
||||
|
||||
// Subscribe once and collect all text events
|
||||
let mut rx = handle.subscribe();
|
||||
let mut texts = Vec::new();
|
||||
tokio::time::timeout(Duration::from_secs(5), async {
|
||||
loop {
|
||||
if let SessionEvent::Data(_, entry) = rx.recv().await.unwrap()
|
||||
&& let DecodedData::Text(s) = entry.data
|
||||
{
|
||||
texts.push(s);
|
||||
if texts.len() == 3 {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert_eq!(texts, vec!["a", "b", "c"]);
|
||||
handle.close().await.unwrap();
|
||||
server.await.unwrap();
|
||||
}
|
||||
|
||||
// ── Session with multiple pipelines ──────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_multi_pipeline_text_and_hex() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
stream.write_all(b"hello\n").await.unwrap();
|
||||
});
|
||||
|
||||
let mut config = tcp_config(addr);
|
||||
config.pipelines.push(PipelineConfig {
|
||||
name: "hex".into(),
|
||||
framer: FramerConfig::Line {
|
||||
strip_cr: true,
|
||||
max_line_len: 1024 * 1024,
|
||||
},
|
||||
decoder: DecoderConfig::Hex {
|
||||
uppercase: false,
|
||||
separator: " ".into(),
|
||||
bytes_per_group: 1,
|
||||
endian: pipeview_core::protocol::Endian::Big,
|
||||
},
|
||||
});
|
||||
|
||||
let handle = Session::spawn(3, config);
|
||||
|
||||
// Subscribe and collect both pipeline results
|
||||
let mut rx = handle.subscribe();
|
||||
let mut results = Vec::new();
|
||||
tokio::time::timeout(Duration::from_secs(5), async {
|
||||
loop {
|
||||
if let SessionEvent::Data(_, entry) = rx.recv().await.unwrap() {
|
||||
results.push((entry.pipeline_name.clone(), entry.data.clone()));
|
||||
if results.len() == 2 {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
assert_eq!(results.len(), 2);
|
||||
assert_eq!(results[0].0, "text");
|
||||
assert_eq!(results[1].0, "hex");
|
||||
assert!(matches!(results[1].1, DecodedData::Hex(_)));
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server.await.unwrap();
|
||||
}
|
||||
|
||||
// ── SessionClose + read timeout ──────────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_read_timeout_returns_none() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (_stream, _) = listener.accept().await.unwrap();
|
||||
// Accept but send nothing — client should time out
|
||||
tokio::time::sleep(Duration::from_secs(10)).await;
|
||||
});
|
||||
|
||||
let handle = Session::spawn(4, tcp_config(addr));
|
||||
|
||||
// Short timeout — no data sent by server
|
||||
let result = handle.read(200).await;
|
||||
assert!(result.is_none(), "read should time out and return None");
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server.abort();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_close_stops_events() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let _server = tokio::spawn(async move {
|
||||
let (_stream, _) = listener.accept().await.unwrap();
|
||||
tokio::time::sleep(Duration::from_secs(10)).await;
|
||||
});
|
||||
|
||||
let handle = Session::spawn(5, tcp_config(addr));
|
||||
handle.close().await.unwrap();
|
||||
|
||||
// After close, read should return None immediately
|
||||
let result = handle.read(100).await;
|
||||
assert!(result.is_none());
|
||||
}
|
||||
|
||||
// ── SessionManager ───────────────────────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn manager_create_and_list() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let _server = tokio::spawn(async move {
|
||||
let (_stream, _) = listener.accept().await.unwrap();
|
||||
tokio::time::sleep(Duration::from_secs(10)).await;
|
||||
});
|
||||
|
||||
let mgr = SessionManager::new();
|
||||
assert_eq!(mgr.count(), 0);
|
||||
assert!(mgr.list_ids().is_empty());
|
||||
|
||||
let h1 = mgr.create(tcp_config(addr.clone()));
|
||||
assert_eq!(mgr.count(), 1);
|
||||
assert_eq!(mgr.list_ids(), vec![1]);
|
||||
|
||||
let h2 = mgr.create(tcp_config(addr));
|
||||
assert_eq!(mgr.count(), 2);
|
||||
|
||||
assert!(mgr.get(1).is_some());
|
||||
assert!(mgr.get(2).is_some());
|
||||
assert!(mgr.get(99).is_none());
|
||||
|
||||
mgr.remove(1);
|
||||
assert_eq!(mgr.count(), 1);
|
||||
assert!(mgr.get(1).is_none());
|
||||
|
||||
h1.close().await.unwrap();
|
||||
h2.close().await.unwrap();
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn manager_event_subscription() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
stream.write_all(b"data\n").await.unwrap();
|
||||
});
|
||||
|
||||
let mgr = SessionManager::new();
|
||||
let mut rx = mgr.subscribe();
|
||||
|
||||
let handle = mgr.create(tcp_config(addr));
|
||||
|
||||
// Wait for Connected event
|
||||
let event = tokio::time::timeout(Duration::from_secs(5), rx.recv())
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
assert!(matches!(event, SessionEvent::Connected(1)));
|
||||
|
||||
// Wait for Data event
|
||||
let event = tokio::time::timeout(Duration::from_secs(5), rx.recv())
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
assert!(matches!(event, SessionEvent::Data(1, _)));
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server.await.unwrap();
|
||||
}
|
||||
|
||||
// ── Reconfigure ──────────────────────────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_reconfigure_changes_pipeline() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
stream.write_all(b"old\n").await.unwrap();
|
||||
});
|
||||
|
||||
let handle = Session::spawn(6, tcp_config(addr));
|
||||
let entry = handle.read(5000).await.unwrap();
|
||||
assert!(matches!(entry.data, DecodedData::Text(s) if s == "old"));
|
||||
server.await.unwrap();
|
||||
|
||||
// Reconfigure to hex pipeline on a new address (don't close first)
|
||||
let listener2 = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr2 = listener2.local_addr().unwrap().to_string();
|
||||
|
||||
let server2 = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener2.accept().await.unwrap();
|
||||
stream.write_all(b"ABC\n").await.unwrap();
|
||||
});
|
||||
|
||||
let mut new_cfg = tcp_config(addr2);
|
||||
new_cfg.pipelines[0].name = "hex".into();
|
||||
new_cfg.pipelines[0].decoder = DecoderConfig::Hex {
|
||||
uppercase: true,
|
||||
separator: " ".into(),
|
||||
bytes_per_group: 1,
|
||||
endian: pipeview_core::protocol::Endian::Big,
|
||||
};
|
||||
|
||||
handle.reconfigure(new_cfg).await.unwrap();
|
||||
let entry = handle.read(5000).await.unwrap();
|
||||
assert!(matches!(entry.data, DecodedData::Hex(s) if s == "41 42 43"));
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server2.await.unwrap();
|
||||
}
|
||||
|
||||
// ── SessionHandle event subscription ─────────────────────────────
|
||||
|
||||
#[tokio::test]
|
||||
async fn session_handle_subscribe_sees_events() {
|
||||
let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
|
||||
let addr = listener.local_addr().unwrap().to_string();
|
||||
|
||||
let server = tokio::spawn(async move {
|
||||
let (mut stream, _) = listener.accept().await.unwrap();
|
||||
stream.write_all(b"event_test\n").await.unwrap();
|
||||
});
|
||||
|
||||
let handle = Session::spawn(7, tcp_config(addr));
|
||||
let mut rx = handle.subscribe();
|
||||
|
||||
// Should see Connected then Data
|
||||
let e1 = tokio::time::timeout(Duration::from_secs(5), rx.recv())
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
assert!(matches!(e1, SessionEvent::Connected(7)));
|
||||
|
||||
let e2 = tokio::time::timeout(Duration::from_secs(5), rx.recv())
|
||||
.await
|
||||
.unwrap()
|
||||
.unwrap();
|
||||
match e2 {
|
||||
SessionEvent::Data(7, entry) => {
|
||||
assert_eq!(entry.pipeline_name, "text");
|
||||
}
|
||||
other => panic!("expected Data, got {:?}", other),
|
||||
}
|
||||
|
||||
handle.close().await.unwrap();
|
||||
server.await.unwrap();
|
||||
}
|
||||
Reference in New Issue
Block a user