2011-10-21 02:42:17 +03:00
|
|
|
using System.Collections.Generic;
|
2013-03-04 02:18:43 +03:00
|
|
|
using System.Linq;
|
2011-09-04 06:05:44 +03:00
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using Moq;
|
|
|
|
using NUnit.Framework;
|
2013-03-01 10:03:41 +03:00
|
|
|
using NzbDrone.Core.MediaFiles;
|
2013-02-27 06:19:22 +03:00
|
|
|
using NzbDrone.Core.Qualities;
|
2013-02-19 09:01:03 +03:00
|
|
|
using NzbDrone.Core.Tv;
|
2011-09-04 06:05:44 +03:00
|
|
|
using NzbDrone.Core.Providers;
|
|
|
|
using NzbDrone.Core.Repository;
|
2011-12-15 07:15:53 +03:00
|
|
|
using NzbDrone.Core.Test.Framework;
|
2011-11-14 03:22:18 +03:00
|
|
|
using NzbDrone.Test.Common.AutoMoq;
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-10-21 02:42:17 +03:00
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
2011-09-04 06:05:44 +03:00
|
|
|
{
|
|
|
|
[TestFixture]
|
2013-02-17 08:44:06 +03:00
|
|
|
public class MisnamedProviderTest : CoreTest
|
2011-09-04 06:05:44 +03:00
|
|
|
{
|
|
|
|
[Test]
|
|
|
|
public void no_misnamed_files()
|
|
|
|
{
|
|
|
|
//Setup
|
2011-09-04 10:45:58 +03:00
|
|
|
var series = Builder<Series>.CreateNew()
|
|
|
|
.With(s => s.Title = "SeriesTitle")
|
|
|
|
.Build();
|
|
|
|
|
2011-09-04 06:05:44 +03:00
|
|
|
var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.TheFirst(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title1.avi")
|
|
|
|
.TheNext(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title2.avi")
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.All()
|
|
|
|
.With(e => e.Series = series)
|
|
|
|
.TheFirst(1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(e => e.EpisodeFile = episodeFiles[0])
|
|
|
|
.TheNext(1)
|
|
|
|
.With(e => e.EpisodeFile = episodeFiles[1])
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
WithStrictMocker();
|
|
|
|
|
2013-02-22 03:47:09 +03:00
|
|
|
Mocker.GetMock<IEpisodeService>()
|
2011-09-04 06:05:44 +03:00
|
|
|
.Setup(c => c.EpisodesWithFiles()).Returns(episodes);
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[0] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[0]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title1");
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[1] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[1]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title2");
|
|
|
|
|
|
|
|
//Act
|
|
|
|
var totalItems = 0;
|
2011-12-15 07:15:53 +03:00
|
|
|
var misnamedEpisodes = Mocker.Resolve<MisnamedProvider>().MisnamedFiles(1, 10, out totalItems);
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
misnamedEpisodes.Should().HaveCount(0);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void all_misnamed_files()
|
|
|
|
{
|
|
|
|
//Setup
|
2011-09-04 10:45:58 +03:00
|
|
|
var series = Builder<Series>.CreateNew()
|
|
|
|
.With(s => s.Title = "SeriesTitle")
|
|
|
|
.Build();
|
|
|
|
|
2011-09-04 06:05:44 +03:00
|
|
|
var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.TheFirst(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title1.avi")
|
|
|
|
.TheNext(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title2.avi")
|
2011-09-04 06:05:44 +03:00
|
|
|
.Build();
|
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.All()
|
|
|
|
.With(e => e.Series = series)
|
|
|
|
.TheFirst(1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(e => e.EpisodeFile = episodeFiles[0])
|
|
|
|
.TheNext(1)
|
|
|
|
.With(e => e.EpisodeFile = episodeFiles[1])
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
WithStrictMocker();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2013-02-22 03:47:09 +03:00
|
|
|
Mocker.GetMock<IEpisodeService>()
|
2011-09-04 06:05:44 +03:00
|
|
|
.Setup(c => c.EpisodesWithFiles()).Returns(episodes);
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[0] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[0]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("New Title 1");
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[1] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[1]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("New Title 2");
|
|
|
|
|
|
|
|
//Act
|
|
|
|
var totalItems = 0;
|
2011-12-15 07:15:53 +03:00
|
|
|
var misnamedEpisodes = Mocker.Resolve<MisnamedProvider>().MisnamedFiles(1, 10, out totalItems);
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
misnamedEpisodes.Should().HaveCount(2);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void one_misnamed_file()
|
|
|
|
{
|
|
|
|
//Setup
|
2011-09-04 10:45:58 +03:00
|
|
|
var series = Builder<Series>.CreateNew()
|
|
|
|
.With(s => s.Title = "SeriesTitle")
|
|
|
|
.Build();
|
|
|
|
|
2011-09-04 06:05:44 +03:00
|
|
|
var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.TheFirst(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title1.avi")
|
|
|
|
.TheNext(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title2.avi")
|
2011-09-04 06:05:44 +03:00
|
|
|
.Build();
|
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.All()
|
|
|
|
.With(e => e.Series = series)
|
|
|
|
.TheFirst(1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(e => e.EpisodeFile = episodeFiles[0])
|
|
|
|
.TheNext(1)
|
|
|
|
.With(e => e.EpisodeFile = episodeFiles[1])
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
WithStrictMocker();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2013-02-22 03:47:09 +03:00
|
|
|
Mocker.GetMock<IEpisodeService>()
|
2011-09-04 06:05:44 +03:00
|
|
|
.Setup(c => c.EpisodesWithFiles()).Returns(episodes);
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[0] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[0]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("New Title 1");
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[1] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[1]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title2");
|
|
|
|
|
|
|
|
//Act
|
|
|
|
var totalItems = 0;
|
2011-12-15 07:15:53 +03:00
|
|
|
var misnamedEpisodes = Mocker.Resolve<MisnamedProvider>().MisnamedFiles(1, 10, out totalItems);
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
misnamedEpisodes.Should().HaveCount(1);
|
|
|
|
misnamedEpisodes[0].CurrentName.Should().Be("Title1");
|
|
|
|
misnamedEpisodes[0].ProperName.Should().Be("New Title 1");
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void misnamed_multi_episode_file()
|
|
|
|
{
|
|
|
|
//Setup
|
2011-09-04 10:45:58 +03:00
|
|
|
var series = Builder<Series>.CreateNew()
|
|
|
|
.With(s => s.Title = "SeriesTitle")
|
|
|
|
.Build();
|
|
|
|
|
2011-09-04 06:05:44 +03:00
|
|
|
var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.TheFirst(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title1.avi")
|
|
|
|
.TheNext(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title2.avi")
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(3)
|
2011-10-19 00:46:06 +03:00
|
|
|
.All()
|
|
|
|
.With(e => e.Series = series)
|
|
|
|
.TheFirst(2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(e => e.EpisodeFile = episodeFiles[0])
|
|
|
|
.TheNext(1)
|
|
|
|
.With(e => e.EpisodeFile = episodeFiles[1])
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
WithStrictMocker();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2013-02-22 03:47:09 +03:00
|
|
|
Mocker.GetMock<IEpisodeService>()
|
2011-09-04 06:05:44 +03:00
|
|
|
.Setup(c => c.EpisodesWithFiles()).Returns(episodes);
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[0], episodes[1] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[0]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("New Title 1");
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[2] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[1]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title2");
|
|
|
|
|
|
|
|
//Act
|
|
|
|
var totalItems = 0;
|
2011-12-15 07:15:53 +03:00
|
|
|
var misnamedEpisodes = Mocker.Resolve<MisnamedProvider>().MisnamedFiles(1, 10, out totalItems);
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
misnamedEpisodes.Should().HaveCount(1);
|
|
|
|
misnamedEpisodes[0].CurrentName.Should().Be("Title1");
|
|
|
|
misnamedEpisodes[0].ProperName.Should().Be("New Title 1");
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void no_misnamed_multi_episode_file()
|
|
|
|
{
|
|
|
|
//Setup
|
2011-09-04 10:45:58 +03:00
|
|
|
var series = Builder<Series>.CreateNew()
|
|
|
|
.With(s => s.Title = "SeriesTitle")
|
|
|
|
.Build();
|
|
|
|
|
2011-09-04 06:05:44 +03:00
|
|
|
var episodeFiles = Builder<EpisodeFile>.CreateListOfSize(2)
|
2011-10-19 00:46:06 +03:00
|
|
|
.TheFirst(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 1)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title1.avi")
|
|
|
|
.TheNext(1)
|
2013-03-01 10:03:41 +03:00
|
|
|
.With(f => f.Id = 2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(f => f.Path = @"C:\Test\Title2.avi")
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
var episodes = Builder<Episode>.CreateListOfSize(3)
|
2011-10-19 00:46:06 +03:00
|
|
|
.All()
|
|
|
|
.With(e => e.Series = series)
|
|
|
|
.TheFirst(2)
|
2011-10-23 08:39:14 +03:00
|
|
|
.With(e => e.EpisodeFile = episodeFiles[0])
|
|
|
|
.TheNext(1)
|
|
|
|
.With(e => e.EpisodeFile = episodeFiles[1])
|
2013-03-04 02:18:43 +03:00
|
|
|
.Build().ToList();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
WithStrictMocker();
|
2011-09-04 06:05:44 +03:00
|
|
|
|
2013-02-22 03:47:09 +03:00
|
|
|
Mocker.GetMock<IEpisodeService>()
|
2011-09-04 06:05:44 +03:00
|
|
|
.Setup(c => c.EpisodesWithFiles()).Returns(episodes);
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[0], episodes[1] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[0]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title1");
|
|
|
|
|
2013-03-01 10:03:41 +03:00
|
|
|
Mocker.GetMock<IMediaFileService>()
|
2013-02-27 06:19:22 +03:00
|
|
|
.Setup(c => c.GetNewFilename(new List<Episode> { episodes[2] }, It.IsAny<Series>(), It.IsAny<Quality>(), It.IsAny<bool>(), episodeFiles[1]))
|
2011-09-04 06:05:44 +03:00
|
|
|
.Returns("Title2");
|
|
|
|
|
|
|
|
//Act
|
|
|
|
var totalItems = 0;
|
2011-12-15 07:15:53 +03:00
|
|
|
var misnamedEpisodes = Mocker.Resolve<MisnamedProvider>().MisnamedFiles(1, 10, out totalItems);
|
2011-09-04 06:05:44 +03:00
|
|
|
|
|
|
|
//Assert
|
|
|
|
misnamedEpisodes.Should().HaveCount(0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|