hls_m3u8/tests/media_playlist.rs

use std::time::Duration;

use hls_m3u8::tags::{ExtInf, ExtXByteRange, ExtXMediaSequence, ExtXTargetDuration};
use hls_m3u8::{MediaPlaylist, MediaSegment};
use pretty_assertions::assert_eq;

#[test]
fn test_media_playlist_with_byterange() {
    let media_playlist = concat!(
        "#EXTM3U\n",
        "#EXT-X-TARGETDURATION:10\n",
        "#EXT-X-VERSION:4\n",
        "#EXT-X-MEDIA-SEQUENCE:0\n",
        "#EXTINF:10.0,\n",
        "#EXT-X-BYTERANGE:75232@0\n",
        "video.ts\n",
        "#EXT-X-BYTERANGE:82112@752321\n",
        "#EXTINF:10.0,\n",
        "video.ts\n",
        "#EXTINF:10.0,\n",
        "#EXT-X-BYTERANGE:69864\n",
        "video.ts\n"
    )
    .parse::<MediaPlaylist>()
    .unwrap();

    assert_eq!(
        MediaPlaylist::builder()
            .target_duration(ExtXTargetDuration::new(Duration::from_secs(10)))
            .media_sequence(ExtXMediaSequence::new(0))
            .segments(vec![
                MediaSegment::builder()
                    .inf_tag(ExtInf::new(Duration::from_secs_f64(10.0)))
                    .byte_range_tag(ExtXByteRange::new(75232, Some(0)))
                    .uri("video.ts")
                    .build()
                    .unwrap(),
                MediaSegment::builder()
                    .inf_tag(ExtInf::new(Duration::from_secs_f64(10.0)))
                    .byte_range_tag(ExtXByteRange::new(82112, Some(752321)))
                    .uri("video.ts")
                    .build()
                    .unwrap(),
                MediaSegment::builder()
                    .inf_tag(ExtInf::new(Duration::from_secs_f64(10.0)))
                    .byte_range_tag(ExtXByteRange::new(69864, None))
                    .uri("video.ts")
                    .build()
                    .unwrap(),
            ])
            .build()
            .unwrap(),
        media_playlist
    )
}