Skip to content

Commit

Permalink
[tests] Add required audio stream snapshot tests (#851)
Browse files Browse the repository at this point in the history
  • Loading branch information
@wkozyra95
wkozyra95 authored Nov 12, 2024
1 parent c816a68 commit a4186fe
Show file tree
Hide file tree
Showing 2 changed files with 277 additions and 7 deletions.
282 changes: 276 additions & 6 deletions integration_tests/src/tests/required_inputs.rs
View file
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
use std::{thread, time::Duration};

use crate::{
compare_video_dumps, input_dump_from_disk, split_rtp_packet_dump, CommunicationProtocol,
CompositorInstance, OutputReceiver, PacketSender, VideoValidationConfig,
compare_audio_dumps, compare_video_dumps, input_dump_from_disk, split_rtp_packet_dump,
AudioValidationConfig, CommunicationProtocol, CompositorInstance, OutputReceiver, PacketSender,
VideoValidationConfig,
};
use anyhow::Result;
use serde_json::json;
Expand All @@ -12,8 +13,8 @@ use serde_json::json;
///
/// Show `input_1` and `input_2` side by side for 20 seconds.
#[test]
pub fn required_inputs_no_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_inputs_no_offset_output.rtp";
pub fn required_video_inputs_no_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_video_inputs_no_offset_output.rtp";
let instance = CompositorInstance::start(None);
let input_1_port = instance.get_port();
let input_2_port = instance.get_port();
Expand Down Expand Up @@ -129,8 +130,8 @@ pub fn required_inputs_no_offset() -> Result<()> {
///
/// Show `input_1` and `input_2` side by side for 20 seconds.
#[test]
pub fn required_inputs_with_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_inputs_with_offset_output.rtp";
pub fn required_video_inputs_with_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_video_inputs_with_offset_output.rtp";
let instance = CompositorInstance::start(None);
let input_1_port = instance.get_port();
let input_2_port = instance.get_port();
Expand Down Expand Up @@ -244,6 +245,275 @@ pub fn required_inputs_with_offset() -> Result<()> {
Ok(())
}

/// Required inputs with some packets delayed.
///
/// Countdown from 10 from the beginning
#[test]
pub fn required_audio_inputs_no_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_audio_inputs_no_offset_output.rtp";
let instance = CompositorInstance::start(None);
let input_1_port = instance.get_port();
let output_port = instance.get_port();

instance.send_request(
"output/output_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": output_port,
"audio": {
"initial": {
"inputs": [
{
"input_id": "input_1",
"volume": 1.0,
},
]
},
"encoder": {
"type": "opus",
"channels": "stereo",
}
},
}),
)?;

instance.send_request(
"output/output_1/unregister",
json!({
"schedule_time_ms": 10000,
}),
)?;

let output_receiver = OutputReceiver::start(output_port, CommunicationProtocol::Tcp)?;

instance.send_request(
"input/input_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": input_1_port,
"audio": {
"decoder": "opus"
},
"required": true,
}),
)?;

let mut input_sender = PacketSender::new(CommunicationProtocol::Tcp, input_1_port)?;
let input_dump = input_dump_from_disk("countdown_audio.rtp")?;
let (input_first_part, input_second_part) =
split_rtp_packet_dump(input_dump, Duration::from_secs(1))?;

let input_handle = thread::spawn(move || {
input_sender.send(&input_first_part).unwrap();
thread::sleep(Duration::from_secs(3));
input_sender.send(&input_second_part).unwrap();
});

instance.send_request("start", json!({}))?;

input_handle.join().unwrap();
let new_output_dump = output_receiver.wait_for_output()?;

compare_audio_dumps(
OUTPUT_DUMP_FILE,
&new_output_dump,
AudioValidationConfig {
sampling_intervals: vec![
Duration::from_millis(0)..Duration::from_millis(2000),
Duration::from_millis(2000)..Duration::from_millis(4000),
Duration::from_millis(6000)..Duration::from_millis(8000),
],
..Default::default()
},
)?;

Ok(())
}

/// Required inputs with some packets delayed. Offset set to 1000ms.
///
/// Countdown from 10 delayed by one second
#[test]
pub fn required_audio_inputs_with_offset() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_audio_inputs_with_offset_output.rtp";
let instance = CompositorInstance::start(None);
let input_1_port = instance.get_port();
let output_port = instance.get_port();

instance.send_request(
"output/output_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": output_port,
"audio": {
"initial": {
"inputs": [
{
"input_id": "input_1",
"volume": 1.0,
},
]
},
"encoder": {
"type": "opus",
"channels": "stereo",
}
},
}),
)?;

instance.send_request(
"output/output_1/unregister",
json!({
"schedule_time_ms": 10000,
}),
)?;

let output_receiver = OutputReceiver::start(output_port, CommunicationProtocol::Tcp)?;

instance.send_request(
"input/input_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": input_1_port,
"audio": {
"decoder": "opus"
},
"offset_ms": 1000,
"required": true,
}),
)?;

let mut input_sender = PacketSender::new(CommunicationProtocol::Tcp, input_1_port)?;
let input_dump = input_dump_from_disk("countdown_audio.rtp")?;
let (input_first_part, input_second_part) =
split_rtp_packet_dump(input_dump, Duration::from_secs(1))?;

let input_handle = thread::spawn(move || {
input_sender.send(&input_first_part).unwrap();
thread::sleep(Duration::from_secs(3));
input_sender.send(&input_second_part).unwrap();
});

instance.send_request("start", json!({}))?;

input_handle.join().unwrap();
let new_output_dump = output_receiver.wait_for_output()?;

compare_audio_dumps(
OUTPUT_DUMP_FILE,
&new_output_dump,
AudioValidationConfig {
sampling_intervals: vec![
Duration::from_millis(0)..Duration::from_millis(2000),
Duration::from_millis(2000)..Duration::from_millis(4000),
Duration::from_millis(6000)..Duration::from_millis(8000),
],
..Default::default()
},
)?;

Ok(())
}

/// Required inputs with some packets delayed and some dropped. Offset set to 1000ms.
///
/// Countdown from 10
/// - 1 seconds of silence
/// - 1 second of audio (from the start of the recording)
/// - 2 seconds of silence (cuts of part of "10" and "9" from countdown)
/// - remaining part of audio (2 seconds from previous step are missing)
#[test]
pub fn required_audio_inputs_with_offset_missing_data() -> Result<()> {
const OUTPUT_DUMP_FILE: &str = "required_audio_inputs_with_offset_missing_data_output.rtp";
let instance = CompositorInstance::start(None);
let input_1_port = instance.get_port();
let output_port = instance.get_port();

instance.send_request(
"output/output_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": output_port,
"audio": {
"initial": {
"inputs": [
{
"input_id": "input_1",
"volume": 1.0,
},
]
},
"encoder": {
"type": "opus",
"channels": "stereo",
}
},
}),
)?;

instance.send_request(
"output/output_1/unregister",
json!({
"schedule_time_ms": 10000,
}),
)?;

let output_receiver = OutputReceiver::start(output_port, CommunicationProtocol::Tcp)?;

instance.send_request(
"input/input_1/register",
json!({
"type": "rtp_stream",
"transport_protocol": "tcp_server",
"port": input_1_port,
"audio": {
"decoder": "opus"
},
"offset_ms": 1000,
"required": true,
}),
)?;

let mut input_sender = PacketSender::new(CommunicationProtocol::Tcp, input_1_port)?;
let input_dump = input_dump_from_disk("countdown_audio.rtp")?;
let (input_first_part, input_second_part) =
split_rtp_packet_dump(input_dump, Duration::from_secs(1))?;
let (_dropped_2_seconds, input_second_part_2) =
split_rtp_packet_dump(input_second_part, Duration::from_secs(2))?;

let input_handle = thread::spawn(move || {
input_sender.send(&input_first_part).unwrap();
thread::sleep(Duration::from_secs(3));
input_sender.send(&input_second_part_2).unwrap();
});

instance.send_request("start", json!({}))?;

input_handle.join().unwrap();
let new_output_dump = output_receiver.wait_for_output()?;

compare_audio_dumps(
OUTPUT_DUMP_FILE,
&new_output_dump,
AudioValidationConfig {
sampling_intervals: vec![
Duration::from_millis(0)..Duration::from_millis(2000),
Duration::from_millis(2000)..Duration::from_millis(4000),
Duration::from_millis(6000)..Duration::from_millis(8000),
],
..Default::default()
},
)?;

Ok(())
}

/// Optional inputs with some packets delayed in the middle
/// No offset
///
Expand Down
2 changes: 1 addition & 1 deletion snapshot_tests/snapshots
Submodule snapshots updated 5 files
+ − rtp_packet_dumps/outputs/required_audio_inputs_no_offset_output.rtp
+ − rtp_packet_dumps/outputs/required_audio_inputs_with_offset_missing_data_output.rtp
+ − rtp_packet_dumps/outputs/required_audio_inputs_with_offset_output.rtp
+ − rtp_packet_dumps/outputs/required_video_inputs_no_offset_output.rtp
+ − rtp_packet_dumps/outputs/required_video_inputs_with_offset_output.rtp

0 comments on commit a4186fe

Please sign in to comment.