2012-02-21 06:25:19 +03:00
|
|
|
// ReSharper disable RedundantUsingDirective
|
2013-02-19 09:01:03 +03:00
|
|
|
|
2012-02-21 06:25:19 +03:00
|
|
|
using System;
|
|
|
|
using System.Collections.Generic;
|
|
|
|
using System.Linq;
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using NUnit.Framework;
|
2013-02-19 09:01:03 +03:00
|
|
|
using NzbDrone.Core.Tv;
|
2012-02-21 06:25:19 +03:00
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
|
2013-02-19 09:01:03 +03:00
|
|
|
namespace NzbDrone.Core.Test.TvTests
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
|
|
|
[TestFixture]
|
|
|
|
// ReSharper disable InconsistentNaming
|
2013-02-02 23:54:03 +03:00
|
|
|
public class SeasonProviderTest : SqlCeTest
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
2012-02-28 08:50:56 +03:00
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
|
|
|
WithRealDb();
|
|
|
|
}
|
|
|
|
|
2012-02-28 08:50:56 +03:00
|
|
|
|
2012-02-21 06:25:19 +03:00
|
|
|
[TestCase(true)]
|
|
|
|
[TestCase(false)]
|
2012-02-28 08:50:56 +03:00
|
|
|
public void SetIgnore_should_update_ignored_status(bool ignoreFlag)
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
|
|
|
var fakeSeason = Builder<Season>.CreateNew()
|
2012-02-28 08:50:56 +03:00
|
|
|
.With(s => s.Ignored = !ignoreFlag)
|
2012-02-21 06:25:19 +03:00
|
|
|
.Build();
|
|
|
|
|
2012-02-28 08:50:56 +03:00
|
|
|
var fakeEpisodes = Builder<Episode>.CreateListOfSize(4)
|
|
|
|
.All()
|
|
|
|
.With(c => c.SeriesId = fakeSeason.SeriesId)
|
2013-02-26 06:58:57 +03:00
|
|
|
.With(c => c.SeasonNumber = fakeSeason.Id)
|
2012-02-28 08:50:56 +03:00
|
|
|
.With(c => c.Ignored = !ignoreFlag)
|
|
|
|
.Build().ToList();
|
|
|
|
|
|
|
|
fakeEpisodes.ForEach(c => Db.Insert(c));
|
|
|
|
|
2012-02-21 06:25:19 +03:00
|
|
|
var id = Db.Insert(fakeSeason);
|
|
|
|
|
|
|
|
//Act
|
2013-02-20 05:05:15 +03:00
|
|
|
Mocker.Resolve<ISeasonService>().SetIgnore(fakeSeason.SeriesId, fakeSeason.SeasonNumber, ignoreFlag);
|
2012-02-21 06:25:19 +03:00
|
|
|
|
|
|
|
//Assert
|
2012-02-28 08:50:56 +03:00
|
|
|
var season = Db.SingleOrDefault<Season>(id);
|
|
|
|
season.Ignored.Should().Be(ignoreFlag);
|
|
|
|
|
|
|
|
var episodes = Db.Fetch<Episode>();
|
|
|
|
episodes.Should().HaveSameCount(fakeEpisodes);
|
|
|
|
episodes.Should().OnlyContain(c => c.Ignored == ignoreFlag);
|
2012-02-21 06:25:19 +03:00
|
|
|
}
|
|
|
|
|
2012-02-28 08:50:56 +03:00
|
|
|
[TestCase(true)]
|
|
|
|
[TestCase(false)]
|
|
|
|
public void IsIgnored_should_return_ignored_status_of_season(bool ignoreFlag)
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
|
|
|
//Setup
|
|
|
|
var fakeSeason = Builder<Season>.CreateNew()
|
2012-02-28 08:50:56 +03:00
|
|
|
.With(s => s.Ignored = ignoreFlag)
|
2012-02-21 06:25:19 +03:00
|
|
|
.Build();
|
|
|
|
|
|
|
|
Db.Insert(fakeSeason);
|
|
|
|
|
|
|
|
//Act
|
2013-02-20 05:05:15 +03:00
|
|
|
var result = Mocker.Resolve<SeasonRepository>().IsIgnored(fakeSeason.SeriesId, fakeSeason.SeasonNumber);
|
2012-02-21 06:25:19 +03:00
|
|
|
|
|
|
|
//Assert
|
2012-02-28 08:50:56 +03:00
|
|
|
result.Should().Be(ignoreFlag);
|
2012-02-21 06:25:19 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
2012-02-28 08:50:56 +03:00
|
|
|
public void IsIgnored_should_throw_an_exception_if_not_in_db()
|
2012-02-21 06:25:19 +03:00
|
|
|
{
|
2013-02-20 05:05:15 +03:00
|
|
|
Assert.Throws<InvalidOperationException>(() => Mocker.Resolve<SeasonRepository>().IsIgnored(10, 0));
|
2012-02-21 09:50:38 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void All_should_return_seasons_with_episodes()
|
|
|
|
{
|
|
|
|
const int seriesId = 10;
|
|
|
|
|
|
|
|
var season = Builder<Season>.CreateNew()
|
|
|
|
.With(s => s.SeriesId = seriesId)
|
|
|
|
.With(s => s.SeasonNumber = 4)
|
|
|
|
.With(s => s.Ignored = true)
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(10)
|
|
|
|
.All()
|
|
|
|
.With(e => e.SeriesId = seriesId)
|
|
|
|
.With(e => e.SeasonNumber = season.SeasonNumber)
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
Db.Insert(season);
|
|
|
|
Db.InsertMany(episodes);
|
|
|
|
|
|
|
|
//Act
|
2013-02-20 05:05:15 +03:00
|
|
|
var result = Mocker.Resolve<SeasonRepository>().GetSeasonBySeries(seriesId);
|
2012-02-21 09:50:38 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
result.Should().HaveCount(1);
|
|
|
|
result.First().Episodes.Should().HaveCount(episodes.Count);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void All_should_return_all_seasons_with_episodes()
|
|
|
|
{
|
|
|
|
const int seriesId = 10;
|
|
|
|
|
|
|
|
//Setup
|
|
|
|
WithRealDb();
|
|
|
|
|
|
|
|
var seasons = Builder<Season>.CreateListOfSize(5)
|
|
|
|
.All()
|
|
|
|
.With(s => s.SeriesId = seriesId)
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
var episodes = new List<Episode>();
|
|
|
|
|
|
|
|
for (int i = 0; i < seasons.Count; i++)
|
|
|
|
{
|
|
|
|
var newEps = Builder<Episode>.CreateListOfSize(2)
|
|
|
|
.All()
|
|
|
|
.With(e => e.SeriesId = seriesId)
|
|
|
|
.With(e => e.SeasonNumber = i + 1)
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
episodes.AddRange(newEps);
|
|
|
|
}
|
|
|
|
|
|
|
|
Db.InsertMany(seasons);
|
|
|
|
Db.InsertMany(episodes);
|
|
|
|
|
|
|
|
//Act
|
2013-02-20 05:05:15 +03:00
|
|
|
var result = Mocker.Resolve<SeasonRepository>().GetSeasonBySeries(seriesId);
|
2012-02-21 09:50:38 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
result.Should().HaveCount(5);
|
|
|
|
|
2012-02-28 08:50:56 +03:00
|
|
|
foreach (var season in result)
|
2012-02-21 09:50:38 +03:00
|
|
|
{
|
|
|
|
season.Episodes.Count.Should().Be(2);
|
2012-02-28 08:50:56 +03:00
|
|
|
season.Episodes.Should().OnlyContain(c => c.SeasonNumber == season.SeasonNumber);
|
2012-02-21 09:50:38 +03:00
|
|
|
}
|
|
|
|
}
|
2012-02-28 08:50:56 +03:00
|
|
|
|
2013-02-20 05:05:15 +03:00
|
|
|
|
2012-02-28 08:50:56 +03:00
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void GetSeason_should_return_seasons_for_specified_series_only()
|
|
|
|
{
|
|
|
|
var seriesA = new[] { 1, 2, 3 };
|
|
|
|
var seriesB = new[] { 4, 5, 6 };
|
|
|
|
|
2013-02-20 05:05:15 +03:00
|
|
|
Mocker.Resolve<SeasonRepository>().GetSeasonNumbers(1).Should().Equal(seriesA);
|
|
|
|
Mocker.Resolve<SeasonRepository>().GetSeasonNumbers(2).Should().Equal(seriesB);
|
2012-02-28 08:50:56 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void GetSeason_should_return_emptylist_if_series_doesnt_exist()
|
|
|
|
{
|
2013-02-20 05:05:15 +03:00
|
|
|
Mocker.Resolve<SeasonRepository>().GetSeasonNumbers(1).Should().BeEmpty();
|
2012-02-28 08:50:56 +03:00
|
|
|
}
|
|
|
|
|
2012-02-21 06:25:19 +03:00
|
|
|
}
|
|
|
|
}
|