mirrors/gst-plugins-rs
mirrors/gst-plugins-rs
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gst-plugins-rs.git synced 2024-11-22 19:41:00 +00:00
Code Issues Projects Releases Wiki Activity

signalling: add and fix test for consumer disconnection

Browse source 
Fixes #92
This commit is contained in:
Mathieu Duponchelle 2022-09-05 22:08:14 +02:00
parent 08b79ab8f1
commit 5ebf9913a1
1 changed files with 52 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
signalling/src/handlers/mod.rs
View file

@ -120,7 +120,7 @@ impl Handler {
.sessions
.iter()
.filter_map(|(session_id, session)| {
if session.producer == peer_id {
if session.producer == peer_id || session.consumer == peer_id {
Some(session_id.clone())
} else {
None
@ -652,6 +652,57 @@ mod tests {
);
}
#[async_std::test]
async fn test_disconnect_consumer() {
let (mut tx, rx) = mpsc::unbounded();
let mut handler = Handler::new(Box::pin(rx));
new_peer(&mut tx, &mut handler, "producer").await;
let message = p::IncomingMessage::SetPeerStatus(p::PeerStatus {
roles: vec![p::PeerRole::Producer],
meta: None,
peer_id: None,
});
tx.send(("producer".to_string(), Some(message)))
.await
.unwrap();
new_peer(&mut tx, &mut handler, "consumer").await;
let message = p::IncomingMessage::StartSession(p::StartSessionMessage {
peer_id: "producer".to_string(),
});
tx.send(("consumer".to_string(), Some(message)))
.await
.unwrap();
let (peer_id, sent_message) = handler.next().await.unwrap();
assert_eq!(peer_id, "consumer");
let session_id = match sent_message {
p::OutgoingMessage::SessionStarted {
ref peer_id,
ref session_id,
} => {
assert_eq!(peer_id, "producer");
session_id.to_string()
}
_ => panic!("SessionStarted message missing"),
};
let _ = handler.next().await.unwrap();
tx.send(("consumer".to_string(), None)).await.unwrap();
let (peer_id, sent_message) = handler.next().await.unwrap();
assert_eq!(peer_id, "producer");
assert_eq!(
sent_message,
p::OutgoingMessage::EndSession(p::EndSessionMessage {
session_id: session_id
})
);
}
#[async_std::test]
async fn test_end_session_producer() {
let (mut tx, rx) = mpsc::unbounded();