using System; using FizzWare.NBuilder; using FluentAssertions; using NUnit.Framework; using NzbDrone.Core.Datastore; using NzbDrone.Core.Test.Framework; using NzbDrone.Core.Tv; namespace NzbDrone.Core.Test.TvTests.EpisodeRepositoryTests { [TestFixture] public class EpisodesWithoutFilesFixture : DbTest { private Series _monitoredSeries; private Series _unmonitoredSeries; private PagingSpec _pagingSpec; [SetUp] public void Setup() { _monitoredSeries = Builder.CreateNew() .With(s => s.Id = 0) .With(s => s.Runtime = 30) .With(s => s.Monitored = true) .Build(); _unmonitoredSeries = Builder.CreateNew() .With(s => s.Id = 0) .With(s => s.Runtime = 30) .With(s => s.Monitored = false) .Build(); _monitoredSeries.Id = Db.Insert(_monitoredSeries).Id; _unmonitoredSeries.Id = Db.Insert(_unmonitoredSeries).Id; _pagingSpec = new PagingSpec { Page = 1, PageSize = 10, SortKey = "AirDate", SortDirection = SortDirection.Ascending }; var monitoredSeriesEpisodes = Builder.CreateListOfSize(3) .All() .With(e => e.Id = 0) .With(e => e.SeriesId = _monitoredSeries.Id) .With(e => e.EpisodeFileId = 0) .With(e => e.AirDate = DateTime.Now.AddDays(-5)) .With(e => e.Monitored = true) .TheFirst(1) .With(e => e.Monitored = false) .TheLast(1) .With(e => e.SeasonNumber = 0) .Build(); var unmonitoredSeriesEpisodes = Builder.CreateListOfSize(3) .All() .With(e => e.Id = 0) .With(e => e.SeriesId = _unmonitoredSeries.Id) .With(e => e.EpisodeFileId = 0) .With(e => e.AirDate = DateTime.Now.AddDays(-5)) .With(e => e.Monitored = true) .TheFirst(1) .With(e => e.Monitored = false) .TheLast(1) .With(e => e.SeasonNumber = 0) .Build(); Db.InsertMany(monitoredSeriesEpisodes); Db.InsertMany(unmonitoredSeriesEpisodes); } [Test] public void should_get_monitored_episodes() { var episodes = Subject.EpisodesWithoutFiles(_pagingSpec, false); episodes.Records.Should().HaveCount(1); } [Test] [Ignore("Specials not implemented")] public void should_get_episode_including_specials() { var episodes = Subject.EpisodesWithoutFiles(_pagingSpec, true); episodes.Records.Should().HaveCount(2); } [Test] public void should_not_include_unmonitored_episodes() { var episodes = Subject.EpisodesWithoutFiles(_pagingSpec, false); episodes.Records.Should().NotContain(e => e.Monitored == false); } [Test] public void should_not_contain_unmonitored_series() { var episodes = Subject.EpisodesWithoutFiles(_pagingSpec, false); episodes.Records.Should().NotContain(e => e.SeriesId == _unmonitoredSeries.Id); } [Test] public void should_have_count_of_one() { var episodes = Subject.EpisodesWithoutFiles(_pagingSpec, false); episodes.TotalRecords.Should().Be(1); } } }