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.
135 lines
5.0 KiB
135 lines
5.0 KiB
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using FizzWare.NBuilder;
|
|
using FluentAssertions;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Core.CustomFormats;
|
|
using NzbDrone.Core.Datastore;
|
|
using NzbDrone.Core.Housekeeping.Housekeepers;
|
|
using NzbDrone.Core.Messaging.Events;
|
|
using NzbDrone.Core.Profiles;
|
|
using NzbDrone.Core.Profiles.Qualities;
|
|
using NzbDrone.Core.Test.Framework;
|
|
|
|
namespace NzbDrone.Core.Test.Housekeeping.Housekeepers
|
|
{
|
|
[TestFixture]
|
|
public class CleanupQualityProfileFormatItemsFixture : DbTest<CleanupQualityProfileFormatItems, QualityProfile>
|
|
{
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
Mocker.SetConstant<IQualityProfileFormatItemsCleanupRepository>(
|
|
new QualityProfileFormatItemsCleanupRepository(Mocker.Resolve<IMainDatabase>(), Mocker.Resolve<IEventAggregator>()));
|
|
|
|
Mocker.SetConstant<ICustomFormatRepository>(
|
|
new CustomFormatRepository(Mocker.Resolve<IMainDatabase>(), Mocker.Resolve<IEventAggregator>()));
|
|
}
|
|
|
|
[Test]
|
|
public void should_remove_orphaned_custom_formats()
|
|
{
|
|
var qualityProfile = Builder<QualityProfile>.CreateNew()
|
|
.With(h => h.Items = Qualities.QualityFixture.GetDefaultQualities())
|
|
.With(h => h.MinFormatScore = 50)
|
|
.With(h => h.CutoffFormatScore = 100)
|
|
.With(h => h.FormatItems = new List<ProfileFormatItem>
|
|
{
|
|
Builder<ProfileFormatItem>.CreateNew()
|
|
.With(c => c.Format = new CustomFormat("My Custom Format") { Id = 0 })
|
|
.Build()
|
|
})
|
|
.BuildNew();
|
|
|
|
Db.Insert(qualityProfile);
|
|
Subject.Clean();
|
|
|
|
var result = AllStoredModels;
|
|
|
|
result.Should().HaveCount(1);
|
|
result.First().FormatItems.Should().BeEmpty();
|
|
result.First().MinFormatScore.Should().Be(0);
|
|
result.First().CutoffFormatScore.Should().Be(0);
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_remove_unorphaned_custom_formats()
|
|
{
|
|
var minFormatScore = 50;
|
|
var cutoffFormatScore = 100;
|
|
|
|
var customFormat = Builder<CustomFormat>.CreateNew()
|
|
.With(h => h.Specifications = new List<ICustomFormatSpecification>())
|
|
.BuildNew();
|
|
|
|
Db.Insert(customFormat);
|
|
|
|
var qualityProfile = Builder<QualityProfile>.CreateNew()
|
|
.With(h => h.Items = Qualities.QualityFixture.GetDefaultQualities())
|
|
.With(h => h.MinFormatScore = minFormatScore)
|
|
.With(h => h.CutoffFormatScore = cutoffFormatScore)
|
|
.With(h => h.FormatItems = new List<ProfileFormatItem>
|
|
{
|
|
Builder<ProfileFormatItem>.CreateNew()
|
|
.With(c => c.Format = customFormat)
|
|
.Build()
|
|
})
|
|
.BuildNew();
|
|
|
|
Db.Insert(qualityProfile);
|
|
|
|
Subject.Clean();
|
|
var result = AllStoredModels;
|
|
|
|
result.Should().HaveCount(1);
|
|
result.First().FormatItems.Should().HaveCount(1);
|
|
result.First().MinFormatScore.Should().Be(minFormatScore);
|
|
result.First().CutoffFormatScore.Should().Be(cutoffFormatScore);
|
|
}
|
|
|
|
[Test]
|
|
public void should_add_missing_custom_formats()
|
|
{
|
|
var minFormatScore = 50;
|
|
var cutoffFormatScore = 100;
|
|
|
|
var customFormat1 = Builder<CustomFormat>.CreateNew()
|
|
.With(h => h.Id = 1)
|
|
.With(h => h.Name = "Custom Format 1")
|
|
.With(h => h.Specifications = new List<ICustomFormatSpecification>())
|
|
.BuildNew();
|
|
|
|
var customFormat2 = Builder<CustomFormat>.CreateNew()
|
|
.With(h => h.Id = 2)
|
|
.With(h => h.Name = "Custom Format 2")
|
|
.With(h => h.Specifications = new List<ICustomFormatSpecification>())
|
|
.BuildNew();
|
|
|
|
Db.Insert(customFormat1);
|
|
Db.Insert(customFormat2);
|
|
|
|
var qualityProfile = Builder<QualityProfile>.CreateNew()
|
|
.With(h => h.Items = Qualities.QualityFixture.GetDefaultQualities())
|
|
.With(h => h.MinFormatScore = minFormatScore)
|
|
.With(h => h.CutoffFormatScore = cutoffFormatScore)
|
|
.With(h => h.FormatItems = new List<ProfileFormatItem>
|
|
{
|
|
Builder<ProfileFormatItem>.CreateNew()
|
|
.With(c => c.Format = customFormat1)
|
|
.Build()
|
|
})
|
|
.BuildNew();
|
|
|
|
Db.Insert(qualityProfile);
|
|
|
|
Subject.Clean();
|
|
var result = AllStoredModels;
|
|
|
|
result.Should().HaveCount(1);
|
|
result.First().FormatItems.Should().HaveCount(2);
|
|
result.First().MinFormatScore.Should().Be(minFormatScore);
|
|
result.First().CutoffFormatScore.Should().Be(cutoffFormatScore);
|
|
}
|
|
}
|
|
}
|