#region Copyright // /************************************************************************ // Copyright (c) 2017 Jamie Rees // File: MovieSenderTests.cs // Created By: Jamie Rees // // Permission is hereby granted, free of charge, to any person obtaining // a copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to // permit persons to whom the Software is furnished to do so, subject to // the following conditions: // // The above copyright notice and this permission notice shall be // included in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. // ************************************************************************/ #endregion using System; using System.Collections.Generic; using System.Linq; using System.Linq.Expressions; using System.Threading.Tasks; using Moq; using NUnit.Framework; using Ombi.Api; using Ombi.Api.Interfaces; using Ombi.Api.Models.Radarr; using Ombi.Api.Models.Sonarr; using Ombi.Api.Models.Watcher; using Ombi.Core.SettingModels; using Ombi.Helpers; using Ombi.Store; using Ploeh.AutoFixture; namespace Ombi.Core.Tests { public class MovieSenderTests { private MovieSender Sender { get; set; } private Mock> CpMock { get; set; } private Mock> WatcherMock { get; set; } private Mock> RadarrMock { get; set; } private Mock CpApiMock { get; set; } private Mock WatcherApiMock { get; set; } private Mock RadarrApiMock { get; set; } private Mock CacheMock { get; set; } private Fixture F { get; set; } [SetUp] public void Setup() { F = new Fixture(); CpMock = new Mock>(); WatcherMock = new Mock>(); RadarrApiMock = new Mock(); RadarrMock = new Mock>(); CpApiMock = new Mock(); WatcherApiMock = new Mock(); CacheMock = new Mock(); RadarrMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, false).Create()); WatcherMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, false).Create()); CpMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, false).Create()); Sender = new MovieSender(CpMock.Object, WatcherMock.Object, CpApiMock.Object, WatcherApiMock.Object, RadarrApiMock.Object, RadarrMock.Object, CacheMock.Object); } [Test] public async Task SendRadarrMovie() { RadarrMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, true).Create()); RadarrApiMock.Setup(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny())).Returns(new RadarrAddMovie { title = "Abc" }); CacheMock.Setup(x => x.GetOrSet>(CacheKeys.RadarrRootFolders, It.IsAny>>(), It.IsAny())) .Returns(F.CreateMany().ToList()); var model = F.Create(); var result = await Sender.Send(model, 2.ToString()); Assert.That(result.Result, Is.True); Assert.That(result.Error, Is.False); Assert.That(result.MovieSendingEnabled, Is.True); RadarrApiMock.Verify(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), 2, It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny()), Times.Once); } [Test] public async Task SendRadarrMovie_SendingFailed() { RadarrMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, true).Create()); RadarrApiMock.Setup(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny())).Returns(new RadarrAddMovie { Error = new RadarrError{message = "Movie Already Added"}}); CacheMock.Setup(x => x.GetOrSet>(CacheKeys.RadarrRootFolders, It.IsAny>>(), It.IsAny())) .Returns(F.CreateMany().ToList()); var model = F.Create(); var result = await Sender.Send(model, 2.ToString()); Assert.That(result.Result, Is.False); Assert.That(result.Error, Is.True); Assert.That(result.MovieSendingEnabled, Is.True); RadarrApiMock.Verify(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), 2, It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny()), Times.Once); } [Test] public async Task SendCpMovie() { CpMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, true).Create()); CpApiMock.Setup(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny())).Returns(true); var model = F.Create(); var result = await Sender.Send(model); Assert.That(result.Result, Is.True); Assert.That(result.Error, Is.False); Assert.That(result.MovieSendingEnabled, Is.True); CpApiMock.Verify(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny()), Times.Once); } [Test] public async Task SendWatcherMovie() { WatcherMock.Setup(x => x.GetSettingsAsync()) .ReturnsAsync(F.Build().With(x => x.Enabled, true).Create()); WatcherApiMock.Setup(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny())).Returns(F.Create()); var model = F.Create(); var result = await Sender.Send(model); Assert.That(result.Result, Is.True); Assert.That(result.Error, Is.False); Assert.That(result.MovieSendingEnabled, Is.True); WatcherApiMock.Verify(x => x.AddMovie(It.IsAny(), It.IsAny(), It.IsAny()), Times.Once); } } }