220 lines
11 KiB
C#
220 lines
11 KiB
C#
using System;
|
|
using System.Globalization;
|
|
using System.IO;
|
|
using System.Text.Json;
|
|
using Jellyfin.Extensions.Json;
|
|
using MediaBrowser.MediaEncoding.Probing;
|
|
using MediaBrowser.Model.Entities;
|
|
using MediaBrowser.Model.Globalization;
|
|
using MediaBrowser.Model.MediaInfo;
|
|
using Microsoft.Extensions.Logging.Abstractions;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace Jellyfin.MediaEncoding.Tests.Probing
|
|
{
|
|
public class ProbeResultNormalizerTests
|
|
{
|
|
private readonly JsonSerializerOptions _jsonOptions = JsonDefaults.Options;
|
|
private readonly ProbeResultNormalizer _probeResultNormalizer = new ProbeResultNormalizer(new NullLogger<EncoderValidatorTests>(), null);
|
|
|
|
[Theory]
|
|
[InlineData("2997/125", 23.976f)]
|
|
[InlineData("1/50", 0.02f)]
|
|
[InlineData("25/1", 25f)]
|
|
[InlineData("120/1", 120f)]
|
|
[InlineData("1704753000/71073479", 23.98578237601117f)]
|
|
[InlineData("0/0", null)]
|
|
[InlineData("1/1000", 0.001f)]
|
|
[InlineData("1/90000", 1.1111111E-05f)]
|
|
[InlineData("1/48000", 2.0833333E-05f)]
|
|
public void GetFrameRate_Success(string value, float? expected)
|
|
=> Assert.Equal(expected, ProbeResultNormalizer.GetFrameRate(value));
|
|
|
|
[Theory]
|
|
[InlineData(0.5f, "0/1", false)]
|
|
[InlineData(24.5f, "8/196", false)]
|
|
[InlineData(63.5f, "1/127", true)]
|
|
[InlineData(null, "1/60", false)]
|
|
[InlineData(30f, "2/120", true)]
|
|
[InlineData(59.999996f, "1563/187560", true)]
|
|
public void IsCodecTimeBaseDoubleTheFrameRate_Success(float? frameRate, string codecTimeBase, bool expected)
|
|
=> Assert.Equal(expected, ProbeResultNormalizer.IsCodecTimeBaseDoubleTheFrameRate(frameRate, codecTimeBase));
|
|
|
|
[Fact]
|
|
public void GetMediaInfo_MetaData_Success()
|
|
{
|
|
var bytes = File.ReadAllBytes("Test Data/Probing/video_metadata.json");
|
|
var internalMediaInfoResult = JsonSerializer.Deserialize<InternalMediaInfoResult>(bytes, _jsonOptions);
|
|
MediaInfo res = _probeResultNormalizer.GetMediaInfo(internalMediaInfoResult, VideoType.VideoFile, false, "Test Data/Probing/video_metadata.mkv", MediaProtocol.File);
|
|
|
|
Assert.Single(res.MediaStreams);
|
|
|
|
Assert.NotNull(res.VideoStream);
|
|
Assert.Equal("4:3", res.VideoStream.AspectRatio);
|
|
Assert.Equal(25f, res.VideoStream.AverageFrameRate);
|
|
Assert.Equal(8, res.VideoStream.BitDepth);
|
|
Assert.Equal(69432, res.VideoStream.BitRate);
|
|
Assert.Equal("h264", res.VideoStream.Codec);
|
|
Assert.Equal("1/50", res.VideoStream.CodecTimeBase);
|
|
Assert.Equal(240, res.VideoStream.Height);
|
|
Assert.Equal(320, res.VideoStream.Width);
|
|
Assert.Equal(0, res.VideoStream.Index);
|
|
Assert.False(res.VideoStream.IsAnamorphic);
|
|
Assert.True(res.VideoStream.IsAVC);
|
|
Assert.True(res.VideoStream.IsDefault);
|
|
Assert.False(res.VideoStream.IsExternal);
|
|
Assert.False(res.VideoStream.IsForced);
|
|
Assert.False(res.VideoStream.IsInterlaced);
|
|
Assert.False(res.VideoStream.IsTextSubtitleStream);
|
|
Assert.Equal(13d, res.VideoStream.Level);
|
|
Assert.Equal("4", res.VideoStream.NalLengthSize);
|
|
Assert.Equal("yuv444p", res.VideoStream.PixelFormat);
|
|
Assert.Equal("High 4:4:4 Predictive", res.VideoStream.Profile);
|
|
Assert.Equal(25f, res.VideoStream.RealFrameRate);
|
|
Assert.Equal(1, res.VideoStream.RefFrames);
|
|
Assert.Equal("1/1000", res.VideoStream.TimeBase);
|
|
Assert.Equal(MediaStreamType.Video, res.VideoStream.Type);
|
|
|
|
Assert.Empty(res.Chapters);
|
|
Assert.Equal("Just color bars", res.Overview);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetMediaInfo_Mp4MetaData_Success()
|
|
{
|
|
var bytes = File.ReadAllBytes("Test Data/Probing/video_mp4_metadata.json");
|
|
var internalMediaInfoResult = JsonSerializer.Deserialize<InternalMediaInfoResult>(bytes, _jsonOptions);
|
|
|
|
// subtitle handling requires a localization object, set a mock to return the input string
|
|
var mockLocalization = new Mock<ILocalizationManager>();
|
|
mockLocalization.Setup(x => x.GetLocalizedString(It.IsAny<string>())).Returns<string>(x => x);
|
|
ProbeResultNormalizer localizedProbeResultNormalizer = new ProbeResultNormalizer(new NullLogger<EncoderValidatorTests>(), mockLocalization.Object);
|
|
|
|
MediaInfo res = localizedProbeResultNormalizer.GetMediaInfo(internalMediaInfoResult, VideoType.VideoFile, false, "Test Data/Probing/video_mp4_metadata.mkv", MediaProtocol.File);
|
|
|
|
// [Video, Audio (Main), Audio (Commentary), Subtitle (Main, Spanish), Subtitle (Main, English), Subtitle (Commentary)
|
|
Assert.Equal(6, res.MediaStreams.Count);
|
|
|
|
Assert.NotNull(res.VideoStream);
|
|
Assert.Equal(res.MediaStreams[0], res.VideoStream);
|
|
Assert.Equal(0, res.VideoStream.Index);
|
|
Assert.Equal("h264", res.VideoStream.Codec);
|
|
Assert.Equal("High", res.VideoStream.Profile);
|
|
Assert.Equal(MediaStreamType.Video, res.VideoStream.Type);
|
|
Assert.Equal(358, res.VideoStream.Height);
|
|
Assert.Equal(720, res.VideoStream.Width);
|
|
Assert.Equal("2.40:1", res.VideoStream.AspectRatio);
|
|
Assert.Equal("yuv420p", res.VideoStream.PixelFormat);
|
|
Assert.Equal(31d, res.VideoStream.Level);
|
|
Assert.Equal(1, res.VideoStream.RefFrames);
|
|
Assert.True(res.VideoStream.IsAVC);
|
|
Assert.Equal(120f, res.VideoStream.RealFrameRate);
|
|
Assert.Equal("1/90000", res.VideoStream.TimeBase);
|
|
Assert.Equal(1147365, res.VideoStream.BitRate);
|
|
Assert.Equal(8, res.VideoStream.BitDepth);
|
|
Assert.True(res.VideoStream.IsDefault);
|
|
Assert.Equal("und", res.VideoStream.Language);
|
|
|
|
Assert.Equal(MediaStreamType.Audio, res.MediaStreams[1].Type);
|
|
Assert.Equal("aac", res.MediaStreams[1].Codec);
|
|
Assert.Equal(7, res.MediaStreams[1].Channels);
|
|
Assert.True(res.MediaStreams[1].IsDefault);
|
|
Assert.Equal("eng", res.MediaStreams[1].Language);
|
|
Assert.Equal("Surround 6.1", res.MediaStreams[1].Title);
|
|
|
|
Assert.Equal(MediaStreamType.Audio, res.MediaStreams[2].Type);
|
|
Assert.Equal("aac", res.MediaStreams[2].Codec);
|
|
Assert.Equal(2, res.MediaStreams[2].Channels);
|
|
Assert.False(res.MediaStreams[2].IsDefault);
|
|
Assert.Equal("eng", res.MediaStreams[2].Language);
|
|
Assert.Equal("Commentary", res.MediaStreams[2].Title);
|
|
|
|
Assert.Equal("spa", res.MediaStreams[3].Language);
|
|
Assert.Equal(MediaStreamType.Subtitle, res.MediaStreams[3].Type);
|
|
Assert.Equal("DVDSUB", res.MediaStreams[3].Codec);
|
|
Assert.Null(res.MediaStreams[3].Title);
|
|
|
|
Assert.Equal("eng", res.MediaStreams[4].Language);
|
|
Assert.Equal(MediaStreamType.Subtitle, res.MediaStreams[4].Type);
|
|
Assert.Equal("mov_text", res.MediaStreams[4].Codec);
|
|
Assert.Null(res.MediaStreams[4].Title);
|
|
|
|
Assert.Equal("eng", res.MediaStreams[5].Language);
|
|
Assert.Equal(MediaStreamType.Subtitle, res.MediaStreams[5].Type);
|
|
Assert.Equal("mov_text", res.MediaStreams[5].Codec);
|
|
Assert.Equal("Commentary", res.MediaStreams[5].Title);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetMediaInfo_MusicVideo_Success()
|
|
{
|
|
var bytes = File.ReadAllBytes("Test Data/Probing/music_video_metadata.json");
|
|
var internalMediaInfoResult = JsonSerializer.Deserialize<InternalMediaInfoResult>(bytes, _jsonOptions);
|
|
MediaInfo res = _probeResultNormalizer.GetMediaInfo(internalMediaInfoResult, VideoType.VideoFile, false, "Test Data/Probing/music_video.mkv", MediaProtocol.File);
|
|
|
|
Assert.Equal("The Title", res.Name);
|
|
Assert.Equal("Title, The", res.ForcedSortName);
|
|
Assert.Single(res.Artists);
|
|
Assert.Equal("The Artist", res.Artists[0]);
|
|
Assert.Equal("Album", res.Album);
|
|
Assert.Equal(2021, res.ProductionYear);
|
|
Assert.True(res.PremiereDate.HasValue);
|
|
Assert.Equal(DateTime.Parse("2021-01-01T00:00Z", DateTimeFormatInfo.CurrentInfo, DateTimeStyles.AdjustToUniversal), res.PremiereDate);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetMediaInfo_GivenOriginalDateContainsOnlyYear_Success()
|
|
{
|
|
var bytes = File.ReadAllBytes("Test Data/Probing/music_year_only_metadata.json");
|
|
var internalMediaInfoResult = JsonSerializer.Deserialize<InternalMediaInfoResult>(bytes, _jsonOptions);
|
|
MediaInfo res = _probeResultNormalizer.GetMediaInfo(internalMediaInfoResult, null, true, "Test Data/Probing/music.flac", MediaProtocol.File);
|
|
|
|
Assert.Equal("Baker Street", res.Name);
|
|
Assert.Single(res.Artists);
|
|
Assert.Equal("Gerry Rafferty", res.Artists[0]);
|
|
Assert.Equal("City to City", res.Album);
|
|
Assert.Equal(1978, res.ProductionYear);
|
|
Assert.True(res.PremiereDate.HasValue);
|
|
Assert.Equal(DateTime.Parse("1978-01-01T00:00Z", DateTimeFormatInfo.CurrentInfo, DateTimeStyles.AdjustToUniversal), res.PremiereDate);
|
|
Assert.Contains("Electronic", res.Genres);
|
|
Assert.Contains("Ambient", res.Genres);
|
|
Assert.Contains("Pop", res.Genres);
|
|
Assert.Contains("Jazz", res.Genres);
|
|
}
|
|
|
|
[Fact]
|
|
public void GetMediaInfo_Music_Success()
|
|
{
|
|
var bytes = File.ReadAllBytes("Test Data/Probing/music_metadata.json");
|
|
var internalMediaInfoResult = JsonSerializer.Deserialize<InternalMediaInfoResult>(bytes, _jsonOptions);
|
|
MediaInfo res = _probeResultNormalizer.GetMediaInfo(internalMediaInfoResult, null, true, "Test Data/Probing/music.flac", MediaProtocol.File);
|
|
|
|
Assert.Equal("UP NO MORE", res.Name);
|
|
Assert.Single(res.Artists);
|
|
Assert.Equal("TWICE", res.Artists[0]);
|
|
Assert.Equal("Eyes wide open", res.Album);
|
|
Assert.Equal(2020, res.ProductionYear);
|
|
Assert.True(res.PremiereDate.HasValue);
|
|
Assert.Equal(DateTime.Parse("2020-10-26T00:00Z", DateTimeFormatInfo.CurrentInfo, DateTimeStyles.AdjustToUniversal), res.PremiereDate);
|
|
Assert.Equal(22, res.People.Length);
|
|
Assert.Equal("Krysta Youngs", res.People[0].Name);
|
|
Assert.Equal(PersonType.Composer, res.People[0].Type);
|
|
Assert.Equal("Julia Ross", res.People[1].Name);
|
|
Assert.Equal(PersonType.Composer, res.People[1].Type);
|
|
Assert.Equal("Yiwoomin", res.People[2].Name);
|
|
Assert.Equal(PersonType.Composer, res.People[2].Type);
|
|
Assert.Equal("Ji-hyo Park", res.People[3].Name);
|
|
Assert.Equal(PersonType.Lyricist, res.People[3].Type);
|
|
Assert.Equal("Yiwoomin", res.People[4].Name);
|
|
Assert.Equal(PersonType.Actor, res.People[4].Type);
|
|
Assert.Equal("Electric Piano", res.People[4].Role);
|
|
Assert.Equal(4, res.Genres.Length);
|
|
Assert.Contains("Electronic", res.Genres);
|
|
Assert.Contains("Trance", res.Genres);
|
|
Assert.Contains("Dance", res.Genres);
|
|
Assert.Contains("Jazz", res.Genres);
|
|
}
|
|
}
|
|
}
|