// ReSharper disable RedundantUsingDirective using System; using System.Collections.Generic; using AutoMoq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.Providers; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; namespace NzbDrone.Core.Test { [TestFixture] // ReSharper disable InconsistentNaming public class QualityTypeProviderTest : TestBase { [Test] public void SetupDefault_should_add_six_profiles() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); //Act mocker.Resolve().SetupDefault(); //Assert var types = mocker.Resolve().All(); types.Should().HaveCount(6); types.Should().Contain(e => e.Name == "SDTV" && e.QualityTypeId == 1); types.Should().Contain(e => e.Name == "DVD" && e.QualityTypeId == 2); types.Should().Contain(e => e.Name == "HDTV" && e.QualityTypeId == 4); types.Should().Contain(e => e.Name == "WEBDL" && e.QualityTypeId == 5); types.Should().Contain(e => e.Name == "Bluray720p" && e.QualityTypeId == 6); types.Should().Contain(e => e.Name == "Bluray1080p" && e.QualityTypeId == 7); } [Test] public void SetupDefault_already_exists() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeQualityType = Builder.CreateNew() .Build(); db.Insert(fakeQualityType); //Act mocker.Resolve().SetupDefault(); //Assert var types = mocker.Resolve().All(); types.Should().HaveCount(1); } [Test] public void GetList_single_quality_type() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeQualityTypes = Builder.CreateListOfSize(6) .Build(); var ids = new List { 1 }; db.InsertMany(fakeQualityTypes); //Act var result = mocker.Resolve().GetList(ids); //Assert result.Should().HaveCount(ids.Count); } [Test] public void GetList_multiple_quality_type() { var mocker = new AutoMoqer(); var db = MockLib.GetEmptyDatabase(); mocker.SetConstant(db); var fakeQualityTypes = Builder.CreateListOfSize(6) .Build(); var ids = new List { 1, 2 }; db.InsertMany(fakeQualityTypes); //Act var result = mocker.Resolve().GetList(ids); //Assert result.Should().HaveCount(ids.Count); } } }