|
|
|
using System.IO;
|
|
|
|
using FizzWare.NBuilder;
|
|
|
|
using FluentAssertions;
|
|
|
|
using Moq;
|
|
|
|
using NUnit.Framework;
|
|
|
|
using NzbDrone.Common.Disk;
|
|
|
|
using NzbDrone.Core.Configuration;
|
|
|
|
using NzbDrone.Core.MediaFiles.MovieImport.Specifications;
|
|
|
|
using NzbDrone.Core.Movies;
|
|
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
using NzbDrone.Test.Common;
|
|
|
|
|
|
|
|
namespace NzbDrone.Core.Test.MediaFiles.MovieImport.Specifications
|
|
|
|
{
|
|
|
|
[TestFixture]
|
|
|
|
public class FreeSpaceSpecificationFixture : CoreTest<FreeSpaceSpecification>
|
|
|
|
{
|
|
|
|
private Movie _movie;
|
|
|
|
private LocalMovie _localMovie;
|
|
|
|
private string _rootFolder;
|
|
|
|
|
|
|
|
[SetUp]
|
|
|
|
public void Setup()
|
|
|
|
{
|
|
|
|
_rootFolder = @"C:\Test\TV".AsOsAgnostic();
|
|
|
|
|
|
|
|
_movie = Builder<Movie>.CreateNew()
|
|
|
|
.With(s => s.Path = Path.Combine(_rootFolder, "30 Rock"))
|
|
|
|
.Build();
|
|
|
|
|
|
|
|
_localMovie = new LocalMovie()
|
|
|
|
{
|
|
|
|
Path = @"C:\Test\Unsorted\30 Rock\30.rock.s01e01.avi".AsOsAgnostic(),
|
|
|
|
Movie = _movie
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenFileSize(long size)
|
|
|
|
{
|
|
|
|
_localMovie.Size = size;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void GivenFreeSpace(long? size)
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
.Setup(s => s.GetAvailableSpace(It.IsAny<string>()))
|
|
|
|
.Returns(size);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_reject_when_there_isnt_enough_disk_space()
|
|
|
|
{
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
GivenFreeSpace(80.Megabytes());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeFalse();
|
|
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_reject_when_there_isnt_enough_space_for_file_plus_min_free_space()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IConfigService>()
|
|
|
|
.Setup(s => s.MinimumFreeSpaceWhenImporting)
|
|
|
|
.Returns(100);
|
|
|
|
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
GivenFreeSpace(150.Megabytes());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeFalse();
|
|
|
|
ExceptionVerification.ExpectedWarns(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_accept_when_there_is_enough_disk_space()
|
|
|
|
{
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
GivenFreeSpace(1.Gigabytes());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_use_series_paths_parent_for_free_space_check()
|
|
|
|
{
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
GivenFreeSpace(1.Gigabytes());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
.Verify(v => v.GetAvailableSpace(_rootFolder), Times.Once());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_pass_if_free_space_is_null()
|
|
|
|
{
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
GivenFreeSpace(null);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_pass_if_exception_is_thrown()
|
|
|
|
{
|
|
|
|
GivenFileSize(100.Megabytes());
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
.Setup(s => s.GetAvailableSpace(It.IsAny<string>()))
|
|
|
|
.Throws(new TestException());
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
ExceptionVerification.ExpectedErrors(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_skip_check_for_files_under_series_folder()
|
|
|
|
{
|
|
|
|
_localMovie.ExistingFile = true;
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
.Verify(s => s.GetAvailableSpace(It.IsAny<string>()), Times.Never());
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_true_if_free_space_is_null()
|
|
|
|
{
|
|
|
|
long? freeSpace = null;
|
|
|
|
|
|
|
|
Mocker.GetMock<IDiskProvider>()
|
|
|
|
.Setup(s => s.GetAvailableSpace(It.IsAny<string>()))
|
|
|
|
.Returns(freeSpace);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
|
|
|
|
[Test]
|
|
|
|
public void should_return_true_when_skip_check_is_enabled()
|
|
|
|
{
|
|
|
|
Mocker.GetMock<IConfigService>()
|
|
|
|
.Setup(s => s.SkipFreeSpaceCheckWhenImporting)
|
|
|
|
.Returns(true);
|
|
|
|
|
|
|
|
Subject.IsSatisfiedBy(_localMovie, null).Accepted.Should().BeTrue();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|