Added EpisodesWithFiles to get episodes in a serires that have an episode file

2026-04-23 22:25:14 -04:00 · 2014-03-06 01:46:19 -08:00
parent 9533e9e403
commit fb5bd4994a
4 changed files with 92 additions and 0 deletions
@@ -0,0 +1,78 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using FizzWare.NBuilder;
+using FluentAssertions;
+using NUnit.Framework;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.Test.Framework;
+using NzbDrone.Core.Tv;
+
+namespace NzbDrone.Core.Test.TvTests.EpisodeRepositoryTests
+{
+    [TestFixture]
+    public class EpisodesWithFilesFixture : DbTest<EpisodeRepository, Episode>
+    {
+        private const int SERIES_ID = 1;
+        private List<Episode> _episodes;
+        private List<EpisodeFile> _episodeFiles;
+
+        [SetUp]
+        public void Setup()
+        {
+            _episodeFiles = Builder<EpisodeFile>.CreateListOfSize(5)
+                                                .BuildListOfNew()
+                                                .ToList();
+
+            Db.InsertMany(_episodeFiles);
+
+            _episodes = Builder<Episode>.CreateListOfSize(10)
+                                        .All()
+                                        .With(e => e.EpisodeFileId = 0)
+                                        .With(e => e.SeriesId = SERIES_ID)
+                                        .BuildListOfNew()
+                                        .ToList();
+
+            for (int i = 0; i < _episodeFiles.Count; i++)
+            {
+                _episodes[i].EpisodeFileId = _episodeFiles[i].Id;
+            }
+
+            Db.InsertMany(_episodes);
+        }
+
+
+        [Test]
+        public void should_only_get_files_that_have_episode_files()
+        {
+            var result = Subject.EpisodesWithFiles(SERIES_ID);
+            
+            result.Should().OnlyContain(e => e.EpisodeFileId > 0);
+            result.Should().HaveCount(_episodeFiles.Count);
+        }
+
+        [Test]
+        public void should_only_contain_episodes_for_the_given_series()
+        {
+            var episodeFile = Builder<EpisodeFile>.CreateNew()
+                                                  .BuildNew();
+
+            Db.Insert(episodeFile);
+
+            var episode = Builder<Episode>.CreateNew()
+                                          .With(e => e.SeriesId = SERIES_ID + 10)
+                                          .With(e => e.EpisodeFileId = episodeFile.Id)
+                                          .BuildNew();
+
+            Db.Insert(episode);
+
+            Subject.EpisodesWithFiles(episode.SeriesId).Should().OnlyContain(e => e.SeriesId == episode.SeriesId);
+        }
+
+        [Test]
+        public void should_have_episode_file_loaded()
+        {
+            Subject.EpisodesWithFiles(SERIES_ID).Should().OnlyContain(e => e.EpisodeFile.IsLoaded);
+        }
+    }
+}