using System.Linq; using FizzWare.NBuilder; using Moq; using NUnit.Framework; using NzbDrone.Core.ImportLists; using NzbDrone.Core.Lifecycle; using NzbDrone.Core.Music; using NzbDrone.Core.Profiles.Qualities; using NzbDrone.Core.RootFolders; using NzbDrone.Core.Test.Framework; namespace NzbDrone.Core.Test.Profiles { [TestFixture] public class ProfileServiceFixture : CoreTest { [Test] public void init_should_add_default_profiles() { Subject.Handle(new ApplicationStartedEvent()); Mocker.GetMock() .Verify(v => v.Insert(It.IsAny()), Times.Exactly(3)); } [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 Init_should_skip_if_any_profiles_already_exist() { Mocker.GetMock() .Setup(s => s.All()) .Returns(Builder.CreateListOfSize(2).Build().ToList()); Subject.Handle(new ApplicationStartedEvent()); Mocker.GetMock() .Verify(v => v.Insert(It.IsAny()), Times.Never()); } [Test] public void should_not_be_able_to_delete_profile_if_assigned_to_artist() { var profile = Builder.CreateNew() .With(p => p.Id = 2) .Build(); var artistList = Builder.CreateListOfSize(3) .Random(1) .With(c => c.QualityProfileId = profile.Id) .Build().ToList(); var importLists = Builder.CreateListOfSize(2) .All() .With(c => c.ProfileId = 1) .Build().ToList(); var rootFolders = Builder.CreateListOfSize(2) .All() .With(f => f.DefaultQualityProfileId = 1) .BuildList(); Mocker.GetMock().Setup(c => c.GetAllArtists()).Returns(artistList); Mocker.GetMock().Setup(c => c.All()).Returns(importLists); Mocker.GetMock().Setup(c => c.All()).Returns(rootFolders); Mocker.GetMock().Setup(c => c.Get(profile.Id)).Returns(profile); Assert.Throws(() => Subject.Delete(profile.Id)); Mocker.GetMock().Verify(c => c.Delete(It.IsAny()), Times.Never()); } [Test] public void should_not_be_able_to_delete_profile_if_assigned_to_import_list() { var profile = Builder.CreateNew() .With(p => p.Id = 2) .Build(); var artistList = Builder.CreateListOfSize(3) .All() .With(c => c.QualityProfileId = 1) .Build().ToList(); var importLists = Builder.CreateListOfSize(2) .Random(1) .With(c => c.ProfileId = profile.Id) .Build().ToList(); var rootFolders = Builder.CreateListOfSize(2) .All() .With(f => f.DefaultQualityProfileId = 1) .BuildList(); Mocker.GetMock().Setup(c => c.GetAllArtists()).Returns(artistList); Mocker.GetMock().Setup(c => c.All()).Returns(importLists); Mocker.GetMock().Setup(c => c.All()).Returns(rootFolders); Mocker.GetMock().Setup(c => c.Get(profile.Id)).Returns(profile); Assert.Throws(() => Subject.Delete(profile.Id)); Mocker.GetMock().Verify(c => c.Delete(It.IsAny()), Times.Never()); } [Test] public void should_not_be_able_to_delete_profile_if_assigned_to_root_folder() { var profile = Builder.CreateNew() .With(p => p.Id = 2) .Build(); var artistList = Builder.CreateListOfSize(3) .All() .With(c => c.QualityProfileId = 1) .Build().ToList(); var importLists = Builder.CreateListOfSize(2) .All() .With(c => c.ProfileId = 1) .Build().ToList(); var rootFolders = Builder.CreateListOfSize(2) .Random(1) .With(f => f.DefaultQualityProfileId = profile.Id) .BuildList(); Mocker.GetMock().Setup(c => c.GetAllArtists()).Returns(artistList); Mocker.GetMock().Setup(c => c.All()).Returns(importLists); Mocker.GetMock().Setup(c => c.All()).Returns(rootFolders); Mocker.GetMock().Setup(c => c.Get(profile.Id)).Returns(profile); Assert.Throws(() => Subject.Delete(profile.Id)); Mocker.GetMock().Verify(c => c.Delete(It.IsAny()), Times.Never()); } [Test] public void should_delete_profile_if_not_assigned_to_artist_import_list_or_root_folder() { var artistList = Builder.CreateListOfSize(3) .All() .With(c => c.QualityProfileId = 2) .Build().ToList(); var importLists = Builder.CreateListOfSize(2) .All() .With(c => c.ProfileId = 2) .Build().ToList(); var rootFolders = Builder.CreateListOfSize(2) .All() .With(f => f.DefaultQualityProfileId = 2) .BuildList(); Mocker.GetMock().Setup(c => c.GetAllArtists()).Returns(artistList); Mocker.GetMock().Setup(c => c.All()).Returns(importLists); Mocker.GetMock().Setup(c => c.All()).Returns(rootFolders); Subject.Delete(1); Mocker.GetMock().Verify(c => c.Delete(1), Times.Once()); } } }