add test_peer_connection_close_is_send

Author: yngrtc

webrtc/src/peer_connection/peer_connection_test.rs

@@ -1,14 +1,22 @@
 use super::*;
 
+use crate::api::interceptor_registry::register_default_interceptors;
+use crate::api::media_engine::MediaEngine;
 use crate::api::media_engine::MIME_TYPE_VP8;
 use crate::api::APIBuilder;
 use crate::ice_transport::ice_candidate_pair::RTCIceCandidatePair;
+use crate::ice_transport::ice_server::RTCIceServer;
+use crate::peer_connection::configuration::RTCConfiguration;
 use crate::rtp_transceiver::rtp_codec::RTCRtpCodecCapability;
 use crate::stats::StatsReportType;
 use crate::track::track_local::track_local_static_sample::TrackLocalStaticSample;
+use crate::Error;
+use interceptor::registry::Registry;
+
 use bytes::Bytes;
 use media::Sample;
 use std::sync::atomic::AtomicU32;
+use std::sync::Arc;
 use tokio::time::Duration;
 use util::vnet::net::{Net, NetConfig};
 use util::vnet::router::{Router, RouterConfig};
@@ -374,3 +382,41 @@ async fn test_get_stats() -> Result<()> {
 
     Ok(())
 }
+
+#[tokio::test]
+async fn test_peer_connection_close_is_send() -> Result<()> {
+    let handle = tokio::spawn(async move { peer().await });
+    tokio::join!(handle).0.unwrap()
+}
+
+async fn peer() -> Result<()> {
+    let mut m = MediaEngine::default();
+    m.register_default_codecs()?;
+    let mut registry = Registry::new();
+    registry = register_default_interceptors(registry, &mut m)?;
+    let api = APIBuilder::new()
+        .with_media_engine(m)
+        .with_interceptor_registry(registry)
+        .build();
+
+    let config = RTCConfiguration {
+        ice_servers: vec![RTCIceServer {
+            urls: vec!["stun:stun.l.google.com:19302".to_owned()],
+            ..Default::default()
+        }],
+        ..Default::default()
+    };
+
+    let peer_connection = Arc::new(api.new_peer_connection(config).await?);
+
+    let offer = peer_connection.create_offer(None).await?;
+    let mut gather_complete = peer_connection.gathering_complete_promise().await;
+    peer_connection.set_local_description(offer).await?;
+    let _ = gather_complete.recv().await;
+
+    if let Some(_) = peer_connection.local_description().await {}
+
+    peer_connection.close().await?;
+
+    Ok(())
+}

webrtc/src/rtp_transceiver/mod.rs

@@ -175,7 +175,7 @@ pub type TriggerNegotiationNeededFnOption =
 
 /// RTPTransceiver represents a combination of an RTPSender and an RTPReceiver that share a common mid.
 pub struct RTCRtpTransceiver {
-    mid: OnceCell<SmallStr>,                  //atomic.Value
+    mid: OnceCell<SmallStr>,              //atomic.Value
     sender: Mutex<Arc<RTCRtpSender>>,     //atomic.Value
     receiver: Mutex<Arc<RTCRtpReceiver>>, //atomic.Value