osu-lazer/osu.Game.Tests/Editing/Checks/CheckAudioInVideoTest.cs

// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
// See the LICENCE file in the repository root for full licence text.

using System.Collections.Generic;
using System.IO;
using System.Linq;
using Moq;
using NUnit.Framework;
using osu.Game.Beatmaps;
using osu.Game.Rulesets.Edit;
using osu.Game.Rulesets.Edit.Checks;
using osu.Game.Rulesets.Objects;
using osu.Game.Storyboards;
using osu.Game.Tests.Beatmaps;
using osu.Game.Tests.Resources;
using FileInfo = osu.Game.IO.FileInfo;

namespace osu.Game.Tests.Editing.Checks
{
    [TestFixture]
    public class CheckAudioInVideoTest
    {
        private CheckAudioInVideo check;
        private IBeatmap beatmap;

        [SetUp]
        public void Setup()
        {
            check = new CheckAudioInVideo();
            beatmap = new Beatmap<HitObject>
            {
                BeatmapInfo = new BeatmapInfo
                {
                    BeatmapSet = new BeatmapSetInfo
                    {
                        Files = new List<BeatmapSetFileInfo>(new[]
                        {
                            new BeatmapSetFileInfo
                            {
                                Filename = "abc123.mp4",
                                FileInfo = new FileInfo { Hash = "abcdef" }
                            }
                        })
                    }
                }
            };
        }

        [Test]
        public void TestRegularVideoFile()
        {
            Assert.IsEmpty(check.Run(getContext("Videos/test-video.mp4")));
        }

        [Test]
        public void TestVideoFileWithAudio()
        {
            var issues = check.Run(getContext("Videos/test-video-with-audio.mp4")).ToList();

            Assert.That(issues, Has.Count.EqualTo(1));
            Assert.That(issues.Single().Template is CheckAudioInVideo.IssueTemplateHasAudioTrack);
        }

        [Test]
        public void TestVideoFileWithTrackButNoAudio()
        {
            var issues = check.Run(getContext("Videos/test-video-with-track-but-no-audio.mp4")).ToList();

            Assert.That(issues, Has.Count.EqualTo(1));
            Assert.That(issues.Single().Template is CheckAudioInVideo.IssueTemplateHasAudioTrack);
        }

        [Test]
        public void TestMissingFile()
        {
            beatmap.BeatmapInfo.BeatmapSet.Files.Clear();

            var issues = check.Run(getContext("Videos/missing.mp4", allowMissing: true)).ToList();

            Assert.That(issues, Has.Count.EqualTo(1));
            Assert.That(issues.Single().Template is CheckAudioInVideo.IssueTemplateMissingFile);
        }

        private BeatmapVerifierContext getContext(string resourceName, bool allowMissing = false)
        {
            using Stream resourceStream = string.IsNullOrEmpty(resourceName) ? null : TestResources.OpenResource(resourceName);
            if (!allowMissing && resourceStream == null)
                throw new FileNotFoundException($"The requested test resource \"{resourceName}\" does not exist.");

            var storyboard = new Storyboard();
            var layer = storyboard.GetLayer("Video");
            layer.Add(new StoryboardVideo("abc123.mp4", 0));

            var mockWorkingBeatmap = new Mock<TestWorkingBeatmap>(beatmap, null, null);
            mockWorkingBeatmap.Setup(w => w.GetStream(It.IsAny<string>())).Returns(resourceStream);
            mockWorkingBeatmap.As<IWorkingBeatmap>().SetupGet(w => w.Storyboard).Returns(storyboard);

            return new BeatmapVerifierContext(beatmap, mockWorkingBeatmap.Object);
        }
    }
}