// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; using System.Linq.Expressions; using System.Linq; using AutoMoq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.Model.Notification; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Providers.Jobs; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; using SubSonic.Repository; namespace NzbDrone.Core.Test { [TestFixture] // ReSharper disable InconsistentNaming public class MediaFileProviderTests : TestBase { [Test] [Description("Verifies that a new file imported properly")] public void import_new_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = @"WEEDS.S03E01.DUAL.BDRip.XviD.AC3.-HELLYWOOD.avi"; const int seasonNumber = 3; const int episodeNumner = 1; const int size = 12345; //Fakes var fakeSeries = Builder.CreateNew().Build(); var fakeEpisode = Builder.CreateNew() .With(c => c.SeriesId = fakeSeries.SeriesId) .With(c => c.SeasonNumber = seasonNumber) .Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(12345).Verifiable(); mocker.GetMock() .Setup(r => r.Exists(It.IsAny>>())).Returns(false).Verifiable(); mocker.GetMock() .Setup(r => r.Add(It.IsAny())).Returns(0).Verifiable(); mocker.GetMock() .Setup(e => e.GetEpisode(fakeSeries.SeriesId, seasonNumber, episodeNumner)).Returns(fakeEpisode); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert Assert.IsNotNull(result); mocker.GetMock().Verify(r => r.Add(result), Times.Once()); mocker.VerifyAllMocks(); result.SeasonNumber.Should().Be(fakeEpisode.SeasonNumber); Assert.AreEqual(fakeEpisode.SeriesId, result.SeriesId); Assert.AreEqual(QualityTypes.DVD, result.Quality); Assert.AreEqual(Parser.NormalizePath(fileName), result.Path); Assert.AreEqual(size, result.Size); Assert.AreEqual(false, result.Proper); Assert.AreNotEqual(new DateTime(), result.DateAdded); } [Test] [Description("Verifies that a new file imported properly")] public void import_new_daily_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = @"2011.01.10 - Denis Leary - HD TV.mkv"; var airDate = new DateTime(2011, 01, 10); const int size = 12345; //Fakes var fakeSeries = Builder.CreateNew().Build(); var fakeEpisode = Builder.CreateNew().With(c => c.SeriesId = fakeSeries.SeriesId).Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock() .Setup(r => r.Exists(It.IsAny>>())).Returns(false).Verifiable(); mocker.GetMock() .Setup(r => r.Add(It.IsAny())).Returns(0).Verifiable(); mocker.GetMock() .Setup(e => e.GetEpisode(fakeSeries.SeriesId, airDate)).Returns(fakeEpisode). Verifiable(); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(size).Verifiable(); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert Assert.IsNotNull(result); mocker.GetMock().VerifyAll(); mocker.GetMock().Verify(r => r.Add(result), Times.Once()); mocker.GetMock().VerifyAll(); mocker.GetMock().VerifyAll(); //Currently can't verify this since the list of episodes are loaded //Dynamically by SubSonic //Assert.AreEqual(fakeEpisode, result.EpisodeNumbers[0]); Assert.AreEqual(fakeEpisode.SeriesId, result.SeriesId); Assert.AreEqual(QualityTypes.HDTV, result.Quality); Assert.AreEqual(Parser.NormalizePath(fileName), result.Path); Assert.AreEqual(size, result.Size); Assert.AreEqual(false, result.Proper); Assert.AreNotEqual(new DateTime(), result.DateAdded); } [Test] [Description("Verifies that a new file imported properly")] public void import_existing_season_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = @"WEEDS.S03E01.DUAL.BDRip.XviD.AC3.-HELLYWOOD.avi"; const int size = 12345; //Fakes var fakeSeries = Builder.CreateNew().Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(r => r.Exists(It.IsAny>>())).Returns(true).Verifiable(); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(size).Verifiable(); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert mocker.VerifyAllMocks(); result.Should().BeNull(); } [Test] [Description("Verifies that a un-parsable file isn't imported")] public void import_unparsable_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = @"WEEDS.avi"; const int size = 12345; //Fakes var fakeSeries = Builder.CreateNew().Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(r => r.Exists(It.IsAny>>())).Returns(false).Verifiable(); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(size).Verifiable(); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert mocker.VerifyAllMocks(); Assert.IsNull(result); ExceptionVerification.ExcpectedWarns(1); } [Test] [Description("Verifies that a new file imported properly")] public void import_sample_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = @"2011.01.10 - Denis Leary - sample - HD TV.mkv"; var airDate = new DateTime(2011, 01, 10); const int size = 12345; //Fakes var fakeSeries = Builder.CreateNew().Build(); var fakeEpisode = Builder.CreateNew().With(c => c.SeriesId = fakeSeries.SeriesId).Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock() .Setup(r => r.Exists(It.IsAny>>())).Returns(false).Verifiable(); mocker.GetMock() .Setup(r => r.Add(It.IsAny())).Returns(0).Verifiable(); mocker.GetMock() .Setup(e => e.GetEpisode(fakeSeries.SeriesId, airDate)).Returns(fakeEpisode). Verifiable(); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(size).Verifiable(); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert Assert.IsNull(result); } [Test] [Description("Verifies that an existing file will skip import")] public void import_existing_file() { //Arrange ///////////////////////////////////////// //Constants const string fileName = "WEEDS.S03E01-06.DUAL.BDRip.XviD.AC3.-HELLYWOOD.avi"; //Fakes var fakeSeries = Builder.CreateNew().Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(r => r.Exists(It.IsAny>>())).Returns(true).Verifiable(); mocker.GetMock(MockBehavior.Strict); mocker.GetMock() .Setup(e => e.GetSize(fileName)).Returns(500000).Verifiable(); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert Assert.IsNull(result); mocker.GetMock().Verify(r => r.Add(result), Times.Never()); mocker.VerifyAllMocks(); } [Test] [Description("Verifies that a file with no episode is skipped")] public void import_file_with_no_episode() { //Arrange ///////////////////////////////////////// //Constants const string fileName = "WEEDS.S03E01.DUAL.BDRip.XviD.AC3.-HELLYWOOD.avi"; const int seasonNumber = 3; const int episodeNumner = 01; //Fakes var fakeSeries = Builder.CreateNew().Build(); //Mocks var mocker = new AutoMoqer(); mocker.GetMock(MockBehavior.Strict) .Setup(r => r.Exists(It.IsAny>>())).Returns(false).Verifiable(); mocker.GetMock(MockBehavior.Strict) .Setup(e => e.GetEpisode(fakeSeries.SeriesId, seasonNumber, episodeNumner)).Returns(null). Verifiable(); mocker.GetMock(MockBehavior.Strict) .Setup(e => e.GetSize(fileName)).Returns(90000000000); //Act var result = mocker.Resolve().ImportFile(fakeSeries, fileName); //Assert mocker.VerifyAllMocks(); Assert.IsNull(result); mocker.GetMock().Verify(r => r.Add(result), Times.Never()); ExceptionVerification.ExcpectedWarns(1); } [Test] public void scan_series_should_update_last_scan_date() { var mocker = new AutoMoqer(); mocker.GetMock() .Setup(c => c.UpdateSeries(It.Is(s => s.LastDiskSync != null))).Verifiable(); mocker.Resolve().Scan(new Series()); mocker.VerifyAllMocks(); } [Test] public void scan_media_job_should_not_scan_new_series() { var mocker = new AutoMoqer(); var fakeSeries = Builder.CreateListOfSize(2) .WhereTheFirst(1).Has(c => c.Episodes = new List()) .AndTheNext(1).Has(c => c.Episodes = Builder.CreateListOfSize(10).Build()) .Build(); mocker.GetMock() .Setup(c => c.GetAllSeries()).Returns(fakeSeries); mocker.GetMock(MockBehavior.Strict) .Setup(c => c.Scan(fakeSeries.ToList()[1])).Returns(new List()).Verifiable(); mocker.Resolve().Start(new ProgressNotification("test"), 0); mocker.VerifyAllMocks(); } [Test] public void get_series_files() { var firstSeriesFiles = Builder.CreateListOfSize(10) .WhereAll().Have(s => s.SeriesId = 12).Build(); var secondSeriesFiles = Builder.CreateListOfSize(10) .WhereAll().Have(s => s.SeriesId = 20).Build(); var mocker = new AutoMoqer(); var repo = MockLib.GetEmptyRepository(true); repo.AddMany(firstSeriesFiles); repo.AddMany(secondSeriesFiles); mocker.SetConstant(repo); var result = mocker.Resolve().GetSeriesFiles(12); result.Should().HaveSameCount(firstSeriesFiles); } } }