You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
149 lines
5.3 KiB
149 lines
5.3 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common.Disk;
|
|
using NzbDrone.Core.Configuration;
|
|
using NzbDrone.Core.Extras.Metadata;
|
|
using NzbDrone.Core.Extras.Metadata.Files;
|
|
using NzbDrone.Core.Housekeeping.Housekeepers;
|
|
using NzbDrone.Core.Music;
|
|
using NzbDrone.Core.Test.Framework;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
|
|
{
|
|
[TestFixture]
|
|
public class DeleteBadMediaCoversFixture : CoreTest<DeleteBadMediaCovers>
|
|
{
|
|
private List<MetadataFile> _metadata;
|
|
private Dictionary<int, string> _artist;
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
_artist = new Dictionary<int, string>
|
|
{
|
|
{ 1, "C:\\Music\\".AsOsAgnostic() }
|
|
};
|
|
|
|
_metadata = Builder<MetadataFile>.CreateListOfSize(1)
|
|
.Build().ToList();
|
|
|
|
Mocker.GetMock<IArtistService>()
|
|
.Setup(c => c.AllArtistPaths())
|
|
.Returns(_artist);
|
|
|
|
Mocker.GetMock<IMetadataFileService>()
|
|
.Setup(c => c.GetFilesByArtist(_artist.First().Key))
|
|
.Returns(_metadata);
|
|
|
|
Mocker.GetMock<IConfigService>().SetupGet(c => c.CleanupMetadataImages).Returns(true);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_process_non_image_files()
|
|
{
|
|
_metadata.First().RelativePath = "album\\file.xml".AsOsAgnostic();
|
|
_metadata.First().Type = MetadataType.TrackMetadata;
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IDiskProvider>().Verify(c => c.OpenReadStream(It.IsAny<string>()), Times.Never());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_process_images_before_tvdb_switch()
|
|
{
|
|
_metadata.First().LastUpdated = new DateTime(2014, 12, 25);
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IDiskProvider>().Verify(c => c.OpenReadStream(It.IsAny<string>()), Times.Never());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_run_if_flag_is_false()
|
|
{
|
|
Mocker.GetMock<IConfigService>().SetupGet(c => c.CleanupMetadataImages).Returns(false);
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IConfigService>().VerifySet(c => c.CleanupMetadataImages = true, Times.Never());
|
|
Mocker.GetMock<IArtistService>().Verify(c => c.AllArtistPaths(), Times.Never());
|
|
|
|
AssertImageWasNotRemoved();
|
|
}
|
|
|
|
[Test]
|
|
public void should_set_clean_flag_to_false()
|
|
{
|
|
_metadata.First().LastUpdated = new DateTime(2014, 12, 25);
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IConfigService>().VerifySet(c => c.CleanupMetadataImages = false, Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_delete_html_images()
|
|
{
|
|
var imagePath = "C:\\Music\\Album\\image.jpg".AsOsAgnostic();
|
|
_metadata.First().LastUpdated = new DateTime(2014, 12, 29);
|
|
_metadata.First().RelativePath = "Album\\image.jpg".AsOsAgnostic();
|
|
_metadata.First().Type = MetadataType.ArtistImage;
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
.Setup(c => c.OpenReadStream(imagePath))
|
|
.Returns(new FileStream(GetTestPath("Files/html_image.jpg"), FileMode.Open, FileAccess.Read));
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IDiskProvider>().Verify(c => c.DeleteFile(imagePath), Times.Once());
|
|
Mocker.GetMock<IMetadataFileService>().Verify(c => c.Delete(_metadata.First().Id), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_delete_empty_images()
|
|
{
|
|
var imagePath = "C:\\Music\\Album\\image.jpg".AsOsAgnostic();
|
|
_metadata.First().LastUpdated = new DateTime(2014, 12, 29);
|
|
_metadata.First().Type = MetadataType.AlbumImage;
|
|
_metadata.First().RelativePath = "Album\\image.jpg".AsOsAgnostic();
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
.Setup(c => c.OpenReadStream(imagePath))
|
|
.Returns(new FileStream(GetTestPath("Files/emptyfile.txt"), FileMode.Open, FileAccess.Read));
|
|
|
|
Subject.Clean();
|
|
|
|
Mocker.GetMock<IDiskProvider>().Verify(c => c.DeleteFile(imagePath), Times.Once());
|
|
Mocker.GetMock<IMetadataFileService>().Verify(c => c.Delete(_metadata.First().Id), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_delete_non_html_files()
|
|
{
|
|
var imagePath = "C:\\Music\\Album\\image.jpg".AsOsAgnostic();
|
|
_metadata.First().LastUpdated = new DateTime(2014, 12, 29);
|
|
_metadata.First().RelativePath = "Album\\image.jpg".AsOsAgnostic();
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
.Setup(c => c.OpenReadStream(imagePath))
|
|
.Returns(new FileStream(GetTestPath("Files/Queue.txt"), FileMode.Open, FileAccess.Read));
|
|
|
|
Subject.Clean();
|
|
AssertImageWasNotRemoved();
|
|
}
|
|
|
|
private void AssertImageWasNotRemoved()
|
|
{
|
|
Mocker.GetMock<IDiskProvider>().Verify(c => c.DeleteFile(It.IsAny<string>()), Times.Never());
|
|
Mocker.GetMock<IMetadataFileService>().Verify(c => c.Delete(It.IsAny<int>()), Times.Never());
|
|
}
|
|
}
|
|
}
|