Radarr/NzbDrone.Core.Test/QualityProfileTest.cs

171 lines
5.6 KiB
C#
Raw Normal View History

2011-05-22 16:53:21 +00:00
// ReSharper disable RedundantUsingDirective
2011-02-03 20:09:19 +00:00
using System;
2010-09-30 06:59:00 +00:00
using System.Collections.Generic;
using FizzWare.NBuilder;
2011-06-02 21:06:46 +00:00
using FluentAssertions;
using NUnit.Framework;
using NzbDrone.Core.Providers;
using NzbDrone.Core.Repository;
using NzbDrone.Core.Repository.Quality;
using NzbDrone.Core.Test.Framework;
using NzbDrone.Test.Common.AutoMoq;
2010-09-29 17:19:18 +00:00
namespace NzbDrone.Core.Test
{
[TestFixture]
2010-10-07 22:17:24 +00:00
// ReSharper disable InconsistentNaming
2011-11-13 07:27:16 +00:00
public class QualityProfileTest : CoreTest
2010-09-29 17:19:18 +00:00
{
[Test]
public void Test_Storage()
{
//Arrange
var database = TestDbHelper.GetEmptyDatabase();
2010-09-29 17:19:18 +00:00
var testProfile = new QualityProfile
2011-02-03 20:09:19 +00:00
{
Name = Guid.NewGuid().ToString(),
2011-05-23 17:20:43 +00:00
Cutoff = QualityTypes.SDTV,
Allowed = new List<QualityTypes> { QualityTypes.HDTV, QualityTypes.DVD },
2011-02-03 20:09:19 +00:00
};
2010-09-29 17:19:18 +00:00
//Act
2011-06-17 20:31:25 +00:00
var id = Convert.ToInt32(database.Insert(testProfile));
var fetch = database.SingleOrDefault<QualityProfile>(id);
2010-09-29 17:19:18 +00:00
//Assert
Assert.AreEqual(id, fetch.QualityProfileId);
2011-02-03 20:09:19 +00:00
Assert.AreEqual(testProfile.Name, fetch.Name);
2010-09-29 17:19:18 +00:00
Assert.AreEqual(testProfile.Cutoff, fetch.Cutoff);
2010-09-30 06:59:00 +00:00
Assert.AreEqual(testProfile.Allowed, fetch.Allowed);
2010-09-29 17:19:18 +00:00
}
[Test]
public void Test_Storage_no_allowed()
{
//Arrange
var database = TestDbHelper.GetEmptyDatabase();
var testProfile = new QualityProfile
{
Name = Guid.NewGuid().ToString(),
Cutoff = QualityTypes.SDTV
};
//Act
var id = Convert.ToInt32(database.Insert(testProfile));
var fetch = database.SingleOrDefault<QualityProfile>(id);
//Assert
Assert.AreEqual(id, fetch.QualityProfileId);
Assert.AreEqual(testProfile.Name, fetch.Name);
Assert.AreEqual(testProfile.Cutoff, fetch.Cutoff);
fetch.Allowed.Should().HaveCount(0);
}
[Test]
public void Update_Success()
{
//Arrange
2011-12-15 04:15:53 +00:00
var db = TestDbHelper.GetEmptyDatabase();
2011-12-15 04:15:53 +00:00
Mocker.SetConstant(db);
var testProfile = new QualityProfile
{
Name = Guid.NewGuid().ToString(),
Cutoff = QualityTypes.SDTV
};
//Act
var id = Convert.ToInt32(db.Insert(testProfile));
var currentProfile = db.SingleOrDefault<QualityProfile>(id);
//Update
currentProfile.Cutoff = QualityTypes.Bluray720p;
2011-12-15 04:15:53 +00:00
Mocker.Resolve<QualityProvider>().Update(currentProfile);
2011-12-15 04:15:53 +00:00
var updated = Mocker.Resolve<QualityProvider>().Get(currentProfile.QualityProfileId);
//Assert
updated.Name.Should().Be(currentProfile.Name);
updated.Cutoff.Should().Be(QualityTypes.Bluray720p);
updated.AllowedString.Should().Be(currentProfile.AllowedString);
}
[Test]
public void Test_Series_Quality()
{
//Arrange
var database = TestDbHelper.GetEmptyDatabase();
var testProfile = new QualityProfile
2011-04-10 02:44:01 +00:00
{
Name = Guid.NewGuid().ToString(),
2011-05-23 17:20:43 +00:00
Cutoff = QualityTypes.SDTV,
Allowed = new List<QualityTypes> { QualityTypes.HDTV, QualityTypes.DVD },
2011-04-10 02:44:01 +00:00
};
2011-06-17 20:31:25 +00:00
var profileId = Convert.ToInt32(database.Insert(testProfile));
var series = Builder<Series>.CreateNew().Build();
series.QualityProfileId = profileId;
2011-06-17 20:31:25 +00:00
database.Insert(testProfile);
database.Insert(series);
2011-06-17 20:31:25 +00:00
var result = database.Fetch<Series>();
2011-06-02 21:06:46 +00:00
result.Should().HaveCount(1);
2011-06-17 20:31:25 +00:00
var profile = database.SingleOrDefault<QualityProfile>(result[0].QualityProfileId);
Assert.AreEqual(profileId, result[0].QualityProfileId);
Assert.AreEqual(testProfile.Name, profile.Name);
}
[Test]
public void SetupInitial_should_add_two_profiles()
{
2011-12-15 04:15:53 +00:00
var db = TestDbHelper.GetEmptyDatabase();
2011-12-15 04:15:53 +00:00
Mocker.SetConstant(db);
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<QualityProvider>().SetupDefaultProfiles();
//Assert
2011-12-15 04:15:53 +00:00
var profiles = Mocker.Resolve<QualityProvider>().All();
profiles.Should().HaveCount(2);
profiles.Should().Contain(e => e.Name == "HD");
profiles.Should().Contain(e => e.Name == "SD");
}
[Test]
//This confirms that new profiles are added only if no other profiles exists.
//We don't want to keep adding them back if a user deleted them on purpose.
public void SetupInitial_should_skip_if_any_profile_exists()
{
2011-12-15 04:15:53 +00:00
var db = TestDbHelper.GetEmptyDatabase();
2011-12-15 04:15:53 +00:00
Mocker.SetConstant(db);
2012-10-14 00:54:46 +00:00
var fakeProfile = Builder<QualityProfile>.CreateNew().With(p => p.Cutoff = QualityTypes.DVD).With(p => p.Allowed = new List<QualityTypes> { QualityTypes.SDTV, QualityTypes.DVD }).Build();
//Act
2011-12-15 04:15:53 +00:00
Mocker.Resolve<QualityProvider>().Add(fakeProfile);
Mocker.Resolve<QualityProvider>().SetupDefaultProfiles();
//Assert
2011-12-15 04:15:53 +00:00
var profiles = Mocker.Resolve<QualityProvider>().All();
profiles.Should().HaveCount(1);
}
2010-09-29 17:19:18 +00:00
}
}