osu-lazer/osu.Game.Tests/Editing/Checks/CheckZeroByteFilesTest.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.IO;
using System.Linq;
using Moq;
using NUnit.Framework;
using NUnit.Framework.Legacy;
using osu.Game.Beatmaps;
using osu.Game.Rulesets.Edit;
using osu.Game.Rulesets.Edit.Checks;
using osu.Game.Rulesets.Objects;

namespace osu.Game.Tests.Editing.Checks
{
    [TestFixture]
    public class CheckZeroByteFilesTest
    {
        private CheckZeroByteFiles check = null!;
        private IBeatmap beatmap = null!;

        [SetUp]
        public void Setup()
        {
            check = new CheckZeroByteFiles();
            beatmap = new Beatmap<HitObject>
            {
                BeatmapInfo = new BeatmapInfo
                {
                    BeatmapSet = new BeatmapSetInfo
                    {
                        Files =
                        {
                            CheckTestHelpers.CreateMockFile("jpg"),
                        }
                    }
                }
            };
        }

        [Test]
        public void TestNonZeroBytes()
        {
            ClassicAssert.IsEmpty(check.Run(getContext(byteLength: 44)));
        }

        [Test]
        public void TestZeroBytes()
        {
            var issues = check.Run(getContext(byteLength: 0)).ToList();

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

        [Test]
        public void TestMissing()
        {
            ClassicAssert.IsEmpty(check.Run(getContextMissing()));
        }

        private BeatmapVerifierContext getContext(long byteLength)
        {
            var mockStream = new Mock<Stream>();
            mockStream.Setup(s => s.Length).Returns(byteLength);

            var mockWorkingBeatmap = new Mock<IWorkingBeatmap>();
            mockWorkingBeatmap.Setup(w => w.GetStream(It.IsAny<string>())).Returns(mockStream.Object);

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

        private BeatmapVerifierContext getContextMissing()
        {
            var mockWorkingBeatmap = new Mock<IWorkingBeatmap>();
            mockWorkingBeatmap.Setup(w => w.GetStream(It.IsAny<string>())).Returns((Stream)null!);

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