481 lines
19 KiB
C#
481 lines
19 KiB
C#
// ReSharper disable RedundantUsingDirective
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Linq.Expressions;
|
|
using AutoMoq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.Model;
|
|
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 PetaPoco;
|
|
|
|
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<Series>.CreateNew().Build();
|
|
var fakeEpisode = Builder<Episode>.CreateNew()
|
|
.With(c => c.SeriesId = fakeSeries.SeriesId)
|
|
.With(c => c.SeasonNumber = seasonNumber)
|
|
.Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(12345).Verifiable();
|
|
|
|
var database = mocker.GetMock<IDatabase>(MockBehavior.Strict);
|
|
database.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
database.Setup(r => r.Insert(It.IsAny<EpisodeFile>())).Returns(1).Verifiable();
|
|
|
|
|
|
mocker.GetMock<EpisodeProvider>()
|
|
.Setup(e => e.GetEpisode(fakeSeries.SeriesId, seasonNumber, episodeNumner)).Returns(fakeEpisode);
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportFile(fakeSeries, fileName);
|
|
|
|
//Assert
|
|
Assert.IsNotNull(result);
|
|
mocker.GetMock<IDatabase>().Verify(r => r.Insert(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<Series>.CreateNew().Build();
|
|
var fakeEpisode = Builder<Episode>.CreateNew().With(c => c.SeriesId = fakeSeries.SeriesId).Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
var database = mocker.GetMock<IDatabase>(MockBehavior.Strict);
|
|
database.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
database.Setup(r => r.Insert(It.IsAny<EpisodeFile>())).Returns(1).Verifiable();
|
|
|
|
mocker.GetMock<EpisodeProvider>()
|
|
.Setup(e => e.GetEpisode(fakeSeries.SeriesId, airDate)).Returns(fakeEpisode).
|
|
Verifiable();
|
|
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(size).Verifiable();
|
|
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportFile(fakeSeries, fileName);
|
|
|
|
//Assert
|
|
Assert.IsNotNull(result);
|
|
mocker.GetMock<IDatabase>().VerifyAll();
|
|
mocker.GetMock<IDatabase>().Verify(r => r.Insert(result), Times.Once());
|
|
mocker.GetMock<EpisodeProvider>().VerifyAll();
|
|
mocker.GetMock<DiskProvider>().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<Series>.CreateNew().Build();
|
|
var fakeEpisodeFile = Builder<EpisodeFile>.CreateNew().With(s => s.SeriesId = fakeSeries.SeriesId).Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<IDatabase>(MockBehavior.Strict)
|
|
.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(true).Verifiable();
|
|
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(size).Verifiable();
|
|
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().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<Series>.CreateNew().Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<IDatabase>(MockBehavior.Strict)
|
|
.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(size).Verifiable();
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().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<Series>.CreateNew().Build();
|
|
var fakeEpisode = Builder<Episode>.CreateNew().With(c => c.SeriesId = fakeSeries.SeriesId).Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
mocker.GetMock<IDatabase>()
|
|
.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>())).Returns(false).Verifiable();
|
|
mocker.GetMock<IDatabase>()
|
|
.Setup(r => r.Insert(It.IsAny<EpisodeFile>())).Returns(0).Verifiable();
|
|
|
|
mocker.GetMock<EpisodeProvider>()
|
|
.Setup(e => e.GetEpisode(fakeSeries.SeriesId, airDate)).Returns(fakeEpisode).
|
|
Verifiable();
|
|
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(size).Verifiable();
|
|
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().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<Series>.CreateNew().Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
mocker.GetMock<IDatabase>(MockBehavior.Strict)
|
|
.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(true).Verifiable();
|
|
|
|
mocker.GetMock<EpisodeProvider>(MockBehavior.Strict);
|
|
mocker.GetMock<DiskProvider>()
|
|
.Setup(e => e.GetSize(fileName)).Returns(500000).Verifiable();
|
|
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportFile(fakeSeries, fileName);
|
|
|
|
//Assert
|
|
Assert.IsNull(result);
|
|
mocker.GetMock<IDatabase>().Verify(r => r.Insert(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<Series>.CreateNew().Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
mocker.GetMock<IDatabase>(MockBehavior.Strict)
|
|
.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
|
|
mocker.GetMock<EpisodeProvider>(MockBehavior.Strict)
|
|
.Setup(e => e.GetEpisode(fakeSeries.SeriesId, seasonNumber, episodeNumner)).Returns<Episode>(null).
|
|
Verifiable();
|
|
|
|
mocker.GetMock<DiskProvider>(MockBehavior.Strict)
|
|
.Setup(e => e.GetSize(fileName)).Returns(90000000000);
|
|
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportFile(fakeSeries, fileName);
|
|
|
|
//Assert
|
|
mocker.VerifyAllMocks();
|
|
Assert.IsNull(result);
|
|
mocker.GetMock<IDatabase>().Verify(r => r.Insert(result), Times.Never());
|
|
ExceptionVerification.ExcpectedWarns(1);
|
|
}
|
|
|
|
[Test]
|
|
public void scan_series_should_update_last_scan_date()
|
|
{
|
|
|
|
var mocker = new AutoMoqer();
|
|
mocker.GetMock<SeriesProvider>()
|
|
.Setup(c => c.UpdateSeries(It.Is<Series>(s => s.LastDiskSync != null))).Verifiable();
|
|
|
|
mocker.Resolve<MediaFileProvider>().Scan(new Series());
|
|
|
|
mocker.VerifyAllMocks();
|
|
|
|
}
|
|
|
|
[Test]
|
|
public void scan_media_job_should_not_scan_new_series()
|
|
{
|
|
var mocker = new AutoMoqer();
|
|
var fakeSeries = Builder<Series>.CreateListOfSize(2)
|
|
.WhereTheFirst(1).Has(c => c.Episodes = new List<Episode>())
|
|
.AndTheNext(1).Has(c => c.Episodes = Builder<Episode>.CreateListOfSize(10).Build())
|
|
.Build();
|
|
mocker.GetMock<SeriesProvider>()
|
|
.Setup(c => c.GetAllSeries()).Returns(fakeSeries);
|
|
|
|
mocker.GetMock<MediaFileProvider>(MockBehavior.Strict)
|
|
.Setup(c => c.Scan(fakeSeries.ToList()[1])).Returns(new List<EpisodeFile>()).Verifiable();
|
|
|
|
mocker.Resolve<DiskScanJob>().Start(new ProgressNotification("test"), 0);
|
|
|
|
mocker.VerifyAllMocks();
|
|
}
|
|
|
|
[Test]
|
|
public void get_series_files()
|
|
{
|
|
var firstSeriesFiles = Builder<EpisodeFile>.CreateListOfSize(10)
|
|
.WhereAll().Have(s => s.SeriesId = 12).Build();
|
|
|
|
var secondSeriesFiles = Builder<EpisodeFile>.CreateListOfSize(10)
|
|
.WhereAll().Have(s => s.SeriesId = 20).Build();
|
|
|
|
var mocker = new AutoMoqer();
|
|
|
|
var database = MockLib.GetEmptyDatabase(true);
|
|
|
|
foreach (var file in firstSeriesFiles)
|
|
database.Insert(file);
|
|
|
|
foreach (var file in secondSeriesFiles)
|
|
database.Insert(file);
|
|
|
|
mocker.SetConstant(database);
|
|
|
|
var result = mocker.Resolve<MediaFileProvider>().GetSeriesFiles(12);
|
|
|
|
|
|
result.Should().HaveSameCount(firstSeriesFiles);
|
|
}
|
|
|
|
[Test]
|
|
[Description("Verifies that a new download will import successfully")]
|
|
public void import_new_download_success()
|
|
{
|
|
//Fakes
|
|
var fakeSeries = Builder<Series>.CreateNew()
|
|
.With(s => s.Title = "30 Rock")
|
|
.With(s => s.Path = @"C:\Test\TV\30 Rock")
|
|
.Build();
|
|
|
|
var fakeEpisode = Builder<Episode>.CreateNew()
|
|
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
|
.With(e => e.EpisodeFileId = 0)
|
|
.With(e => e.SeasonNumber = 1)
|
|
.With(e => e.EpisodeNumber = 5)
|
|
.Build();
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
var diskProvider = mocker.GetMock<DiskProvider>();
|
|
diskProvider.Setup(d => d.GetFiles(It.IsAny<string>(), "*.*", SearchOption.AllDirectories)).Returns(new string[] { @"C:\Test\30 Rock - 1x05 - Episode Title\30.Rock.S01E05.Gibberish.XviD.avi" });
|
|
diskProvider.Setup(d => d.GetSize(It.IsAny<string>())).Returns(90000000000);
|
|
diskProvider.Setup(d => d.CreateDirectory(It.IsAny<string>())).Returns("ok");
|
|
diskProvider.Setup(d => d.RenameFile(It.IsAny<string>(), It.IsAny<string>()));
|
|
diskProvider.Setup(d => d.GetExtension(It.IsAny<string>())).Returns(".avi");
|
|
|
|
var episodeProvider = mocker.GetMock<EpisodeProvider>();
|
|
episodeProvider.Setup(e => e.GetEpisodes(It.IsAny<EpisodeParseResult>())).Returns(new List<Episode> {fakeEpisode});
|
|
episodeProvider.Setup(e => e.GetEpisode(fakeSeries.SeriesId, 1, 5)).Returns(fakeEpisode);
|
|
|
|
var configProvider = mocker.GetMock<ConfigProvider>();
|
|
configProvider.SetupGet(c => c.UseSeasonFolder).Returns(true);
|
|
configProvider.SetupGet(c => c.SeasonFolderFormat).Returns(@"Season %0s");
|
|
configProvider.SetupGet(c => c.SeriesName).Returns(true);
|
|
configProvider.SetupGet(c => c.EpisodeName).Returns(true);
|
|
configProvider.SetupGet(c => c.AppendQuality).Returns(true);
|
|
configProvider.SetupGet(c => c.SeparatorStyle).Returns(0);
|
|
configProvider.SetupGet(c => c.NumberStyle).Returns(2);
|
|
configProvider.SetupGet(c => c.ReplaceSpaces).Returns(false);
|
|
|
|
var database = mocker.GetMock<IDatabase>(MockBehavior.Strict);
|
|
database.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
database.Setup(r => r.Insert(It.IsAny<EpisodeFile>())).Returns(1).Verifiable();
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportNewFiles(@"C:\Test\30 Rock - 1x05 - Episode Title", fakeSeries);
|
|
|
|
//Assert
|
|
mocker.VerifyAllMocks();
|
|
Assert.AreEqual(1, result.Count);
|
|
}
|
|
|
|
[Test]
|
|
[Description("Verifies that a new download will import successfully, deletes previous episode")]
|
|
public void import_new_download_success_delete_equal_quality()
|
|
{
|
|
//Fakes
|
|
var fakeSeries = Builder<Series>.CreateNew()
|
|
.With(s => s.Title = "30 Rock")
|
|
.With(s => s.Path = @"C:\Test\TV\30 Rock")
|
|
.Build();
|
|
|
|
var fakeEpisode = Builder<Episode>.CreateNew()
|
|
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
|
.With(e => e.SeasonNumber = 1)
|
|
.With(e => e.EpisodeNumber = 5)
|
|
.With(e => e.EpisodeFileId = 1)
|
|
.Build();
|
|
|
|
var fakeEpisodeFile = Builder<EpisodeFile>.CreateNew()
|
|
.With(e => e.SeriesId = fakeSeries.SeriesId)
|
|
.With(e => e.EpisodeFileId = 1)
|
|
.With(e => e.Quality = QualityTypes.SDTV)
|
|
.With(e => e.Episodes = new List<Episode>{ fakeEpisode })
|
|
.Build();
|
|
|
|
fakeEpisode.EpisodeFile = fakeEpisodeFile;
|
|
|
|
//Mocks
|
|
var mocker = new AutoMoqer();
|
|
|
|
var diskProvider = mocker.GetMock<DiskProvider>();
|
|
diskProvider.Setup(d => d.GetFiles(It.IsAny<string>(), "*.*", SearchOption.AllDirectories)).Returns(new string[] { @"C:\Test\30 Rock - 1x05 - Episode Title\30.Rock.S01E05.Gibberish.XviD.avi" });
|
|
diskProvider.Setup(d => d.GetSize(It.IsAny<string>())).Returns(90000000000);
|
|
diskProvider.Setup(d => d.CreateDirectory(It.IsAny<string>())).Returns("ok");
|
|
diskProvider.Setup(d => d.RenameFile(It.IsAny<string>(), It.IsAny<string>()));
|
|
diskProvider.Setup(d => d.GetExtension(It.IsAny<string>())).Returns(".avi");
|
|
|
|
var episodeProvider = mocker.GetMock<EpisodeProvider>();
|
|
episodeProvider.Setup(e => e.GetEpisodes(It.IsAny<EpisodeParseResult>())).Returns(new List<Episode> { fakeEpisode });
|
|
episodeProvider.Setup(e => e.GetEpisode(fakeSeries.SeriesId, 1, 5)).Returns(fakeEpisode);
|
|
|
|
var configProvider = mocker.GetMock<ConfigProvider>();
|
|
configProvider.SetupGet(c => c.UseSeasonFolder).Returns(true);
|
|
configProvider.SetupGet(c => c.SeasonFolderFormat).Returns(@"Season %0s");
|
|
configProvider.SetupGet(c => c.SeriesName).Returns(true);
|
|
configProvider.SetupGet(c => c.EpisodeName).Returns(true);
|
|
configProvider.SetupGet(c => c.AppendQuality).Returns(true);
|
|
configProvider.SetupGet(c => c.SeparatorStyle).Returns(0);
|
|
configProvider.SetupGet(c => c.NumberStyle).Returns(2);
|
|
configProvider.SetupGet(c => c.ReplaceSpaces).Returns(false);
|
|
|
|
var database = mocker.GetMock<IDatabase>(MockBehavior.Strict);
|
|
database.Setup(r => r.Exists<EpisodeFile>(It.IsAny<string>(), It.IsAny<object>())).Returns(false).Verifiable();
|
|
database.Setup(r => r.Insert(It.IsAny<EpisodeFile>())).Returns(1);
|
|
database.Setup(r => r.Delete<EpisodeFile>(It.IsAny<int>())).Returns(1);
|
|
|
|
//Act
|
|
var result = mocker.Resolve<MediaFileProvider>().ImportNewFiles(@"C:\Test\30 Rock - 1x05 - Episode Title", fakeSeries);
|
|
|
|
//Assert
|
|
mocker.VerifyAllMocks();
|
|
Assert.AreEqual(1, result.Count);
|
|
}
|
|
}
|
|
} |