Lidarr/NzbDrone.Core.Test/ProviderTests/DiskScanProviderTest.cs

214 lines
6.9 KiB
C#
Raw Normal View History

2011-06-21 05:44:01 +00:00
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
2011-06-21 05:44:01 +00:00
using FizzWare.NBuilder;
using FluentAssertions;
2011-06-21 05:44:01 +00:00
using Moq;
using NUnit.Framework;
2011-11-13 04:07:06 +00:00
using NzbDrone.Common;
2011-06-21 05:44:01 +00:00
using NzbDrone.Core.Providers;
using NzbDrone.Core.Providers.Core;
using NzbDrone.Core.Repository;
using NzbDrone.Core.Repository.Quality;
2011-06-21 05:44:01 +00:00
using NzbDrone.Core.Test.Framework;
2011-10-24 05:54:09 +00:00
using NzbDrone.Test.Common;
using NzbDrone.Test.Common.AutoMoq;
2011-06-21 05:44:01 +00:00
namespace NzbDrone.Core.Test.ProviderTests
2011-06-21 05:44:01 +00:00
{
// ReSharper disable InconsistentNaming
2011-11-13 07:27:16 +00:00
public class DiskScanProviderTest : CoreTest
2011-06-21 05:44:01 +00:00
{
[Test]
public void scan_series_should_update_the_last_scan_date()
2011-06-21 05:44:01 +00:00
{
2011-12-15 04:15:53 +00:00
Mocker.GetMock<SeriesProvider>()
.Setup(c => c.UpdateSeries(It.Is<Series>(s => s.LastDiskSync != null))).Verifiable();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Setup(c => c.GetEpisodeBySeries(It.IsAny<long>()))
.Returns(new List<Episode> { new Episode() });
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<DiskProvider>()
2011-10-23 00:13:49 +00:00
.Setup(c => c.FolderExists(It.IsAny<string>()))
.Returns(true);
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Setup(c => c.GetSeriesFiles(It.IsAny<int>()))
.Returns(new List<EpisodeFile>());
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.Resolve<DiskScanProvider>().Scan(new Series());
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.VerifyAllMocks();
2011-06-21 05:44:01 +00:00
2011-06-22 01:12:20 +00:00
}
2011-06-21 05:44:01 +00:00
[Test]
public void cleanup_should_skip_existing_files()
2011-06-21 05:44:01 +00:00
{
2011-12-15 04:15:53 +00:00
WithStrictMocker();
var episodes = Builder<EpisodeFile>.CreateListOfSize(10).Build();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<DiskProvider>()
.Setup(e => e.FileExists(It.IsAny<String>()))
.Returns(true);
2011-06-21 05:44:01 +00:00
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<DiskScanProvider>().CleanUp(episodes);
2011-06-21 05:44:01 +00:00
//Assert
2011-12-15 04:15:53 +00:00
Mocker.VerifyAllMocks();
2011-06-21 05:44:01 +00:00
}
[Test]
public void cleanup_should_delete_none_existing_files()
2011-06-21 05:44:01 +00:00
{
2011-12-15 04:15:53 +00:00
WithStrictMocker();
var episodes = Builder<EpisodeFile>.CreateListOfSize(10).Build();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<DiskProvider>()
.Setup(e => e.FileExists(It.IsAny<String>()))
.Returns(false);
2011-06-22 01:12:20 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Setup(e => e.GetEpisodesByFileId(It.IsAny<int>()))
.Returns(new List<Episode>());
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Setup(e => e.Delete(It.IsAny<int>()));
2011-06-21 05:44:01 +00:00
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<DiskScanProvider>().CleanUp(episodes);
2011-06-21 05:44:01 +00:00
//Assert
2011-12-15 04:15:53 +00:00
Mocker.VerifyAllMocks();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Verify(e => e.GetEpisodesByFileId(It.IsAny<int>()), Times.Exactly(10));
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Verify(e => e.Delete(It.IsAny<int>()), Times.Exactly(10));
2011-06-21 05:44:01 +00:00
}
[Test]
public void cleanup_should_delete_none_existing_files_remove_links_to_episodes()
2011-06-21 05:44:01 +00:00
{
2011-12-15 04:15:53 +00:00
WithStrictMocker();
var episodes = Builder<EpisodeFile>.CreateListOfSize(10).Build();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<DiskProvider>()
.Setup(e => e.FileExists(It.IsAny<String>()))
.Returns(false);
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Setup(e => e.GetEpisodesByFileId(It.IsAny<int>()))
.Returns(new List<Episode> { new Episode { EpisodeFileId = 10 }, new Episode { EpisodeFileId = 10 } });
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Setup(e => e.UpdateEpisode(It.IsAny<Episode>()));
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Setup(e => e.Delete(It.IsAny<int>()));
2011-06-22 01:12:20 +00:00
2011-06-21 05:44:01 +00:00
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<DiskScanProvider>().CleanUp(episodes);
2011-06-21 05:44:01 +00:00
//Assert
2011-12-15 04:15:53 +00:00
Mocker.VerifyAllMocks();
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Verify(e => e.GetEpisodesByFileId(It.IsAny<int>()), Times.Exactly(10));
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<EpisodeProvider>()
.Verify(e => e.UpdateEpisode(It.Is<Episode>(g=>g.EpisodeFileId == 0)), Times.Exactly(20));
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Verify(e => e.Delete(It.IsAny<int>()), Times.Exactly(10));
2011-06-21 05:44:01 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Verify(e => e.Delete(It.IsAny<int>()), Times.Exactly(10));
2011-06-21 05:44:01 +00:00
}
2011-06-22 01:12:20 +00:00
[Test]
public void scan_series_should_log_warning_if_path_doesnt_exist_on_disk()
{
//Setup
2011-12-15 04:15:53 +00:00
WithStrictMocker();
var series = Builder<Series>.CreateNew()
.With(s => s.Path = @"C:\Test\TV\SeriesName\")
.Build();
2011-12-15 04:15:53 +00:00
Mocker.GetMock<MediaFileProvider>()
.Setup(c => c.CleanUpDatabase());
2011-06-22 01:12:20 +00:00
2011-12-15 04:15:53 +00:00
Mocker.GetMock<DiskProvider>()
.Setup(c => c.FolderExists(series.Path))
.Returns(false);
2011-06-22 01:12:20 +00:00
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<DiskScanProvider>().Scan(series, series.Path);
//Assert
2011-12-15 04:15:53 +00:00
Mocker.VerifyAllMocks();
ExceptionVerification.ExpectedWarns(1);
}
[Test]
public void move_should_not_move_file_if_source_and_destination_are_the_same_path()
{
var fakeSeries = Builder<Series>.CreateNew()
.With(s => s.SeriesId = 5)
.With(s => s.Title = "30 Rock")
.Build();
var fakeEpisode = Builder<Episode>.CreateListOfSize(1)
.All()
.With(e => e.SeriesId = fakeSeries.SeriesId)
.With(e => e.SeasonNumber = 1)
.With(e => e.EpisodeNumber = 1)
.Build();
const string filename = @"30 Rock - S01E01 - TBD";
var fi = new FileInfo(Path.Combine(@"C:\Test\TV\30 Rock\Season 01\", filename + ".avi"));
var file = Builder<EpisodeFile>.CreateNew()
.With(f => f.SeriesId = fakeSeries.SeriesId)
.With(f => f.Path = fi.FullName)
.Build();
Mocker.GetMock<SeriesProvider>()
.Setup(e => e.GetSeries(fakeSeries.SeriesId))
.Returns(fakeSeries);
Mocker.GetMock<EpisodeProvider>()
.Setup(e => e.GetEpisodesByFileId(file.EpisodeFileId))
.Returns(fakeEpisode);
Mocker.GetMock<MediaFileProvider>()
.Setup(e => e.GetNewFilename(fakeEpisode, fakeSeries.Title, It.IsAny<QualityTypes>(), It.IsAny<bool>()))
.Returns(filename);
Mocker.GetMock<MediaFileProvider>()
.Setup(e => e.CalculateFilePath(It.IsAny<Series>(), fakeEpisode.First().SeasonNumber, filename, ".avi"))
.Returns(fi);
//Act
var result = Mocker.Resolve<DiskScanProvider>().MoveEpisodeFile(file, false);
//Assert
result.Should().BeFalse();
}
2011-06-21 05:44:01 +00:00
}
}