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 = "#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"
        .parse::<MediaPlaylist>()
        .unwrap();

    assert_eq!(
        MediaPlaylist::builder()
            .target_duration_tag(ExtXTargetDuration::new(Duration::from_secs(10)))
            .media_sequence_tag(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
    )
}