jellyfin-ha-src/tests/Jellyfin.LiveTv.Tests/M3UTunerHostTests.cs

using System.Collections.Generic;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
using Jellyfin.LiveTv.TunerHosts;
using MediaBrowser.Common.Net;
using MediaBrowser.Controller;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.LiveTv;
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.IO;
using MediaBrowser.Model.LiveTv;
using MediaBrowser.Model.MediaInfo;
using Microsoft.Extensions.Logging;
using Moq;
using Xunit;

namespace Jellyfin.LiveTv.Tests
{
    public class M3UTunerHostTests
    {
        [Theory]
        // A manifest is not a byte stream, so it must never be offered for direct play.
        [InlineData("http://example.com/live/1234.m3u8", false)]
        [InlineData("http://example.com/live/1234.m3u8?token=abc", false)]
        [InlineData("http://example.com/live/1234.mpd", false)]
        // Byte streams are unaffected.
        [InlineData("http://example.com/live/1234.ts", true)]
        [InlineData("http://example.com/live/1234", true)]
        public async Task GetChannelStreamMediaSources_ManifestPath_DisablesDirectPlay(string path, bool expectDirectPlay)
        {
            var mediaSourceManager = new Mock<IMediaSourceManager>();
            mediaSourceManager.Setup(x => x.GetPathProtocol(It.IsAny<string>())).Returns(MediaProtocol.Http);

            var host = new TestableM3UTunerHost(
                Mock.Of<IServerConfigurationManager>(),
                mediaSourceManager.Object,
                Mock.Of<ILogger<M3UTunerHost>>(),
                Mock.Of<IFileSystem>(),
                Mock.Of<IHttpClientFactory>(),
                Mock.Of<IServerApplicationHost>(),
                Mock.Of<INetworkManager>(),
                Mock.Of<IStreamHelper>());

            var sources = await host.GetMediaSources(
                new TunerHostInfo { TunerCount = 0, EnableStreamLooping = false },
                new ChannelInfo { Path = path });

            Assert.Equal(expectDirectPlay, sources[0].SupportsDirectPlay);
        }

        private sealed class TestableM3UTunerHost : M3UTunerHost
        {
            public TestableM3UTunerHost(
                IServerConfigurationManager config,
                IMediaSourceManager mediaSourceManager,
                ILogger<M3UTunerHost> logger,
                IFileSystem fileSystem,
                IHttpClientFactory httpClientFactory,
                IServerApplicationHost appHost,
                INetworkManager networkManager,
                IStreamHelper streamHelper)
                : base(config, mediaSourceManager, logger, fileSystem, httpClientFactory, appHost, networkManager, streamHelper)
            {
            }

            public Task<List<MediaSourceInfo>> GetMediaSources(TunerHostInfo tuner, ChannelInfo channel)
                => GetChannelStreamMediaSources(tuner, channel, CancellationToken.None);
        }
    }
}