// ReSharper disable RedundantUsingDirective using System; using System.IO; using System.Linq; using FizzWare.NBuilder; using FluentAssertions; using Moq; using NUnit.Framework; using NzbDrone.Core.Model; using NzbDrone.Core.Providers; using NzbDrone.Core.Providers.Core; using NzbDrone.Core.Repository; using NzbDrone.Core.Repository.Quality; using NzbDrone.Core.Test.Framework; using NzbDrone.Test.Common; using NzbDrone.Test.Common.AutoMoq; namespace NzbDrone.Core.Test.ProviderTests { [TestFixture] // ReSharper disable InconsistentNaming public class SabProviderTest : CoreTest { private void WithSabConfigValues() { //Setup string sabHost = "192.168.5.55"; int sabPort = 2222; string apikey = "5c770e3197e4fe763423ee7c392c25d1"; string username = "admin"; string password = "pass"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); } [Test] public void AddByUrlSuccess() { //Setup const string sabHost = "192.168.5.55"; const int sabPort = 2222; const string apikey = "5c770e3197e4fe763423ee7c392c25d1"; const string username = "admin"; const string password = "pass"; const SabnzbdPriorityType priority = SabnzbdPriorityType.Normal; const string category = "tv"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); fakeConfig.SetupGet(c => c.SabTvPriority) .Returns(priority); fakeConfig.SetupGet(c => c.SabTvCategory) .Returns(category); Mocker.GetMock(MockBehavior.Strict) .Setup( s => s.DownloadString( "http://192.168.5.55:2222/api?mode=addurl&name=http://www.nzbclub.com/nzb_download.aspx?mid=1950232&priority=0&pp=3&cat=tv&nzbname=This+is+an+Nzb&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns("ok"); //Act bool result = Mocker.Resolve().AddByUrl( "http://www.nzbclub.com/nzb_download.aspx?mid=1950232", "This is an Nzb"); //Assert result.Should().BeTrue(); } [Test] public void AddByUrlNewzbin() { //Setup const string sabHost = "192.168.5.55"; const int sabPort = 2222; const string apikey = "5c770e3197e4fe763423ee7c392c25d1"; const string username = "admin"; const string password = "pass"; const SabnzbdPriorityType priority = SabnzbdPriorityType.Normal; const string category = "tv"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); fakeConfig.SetupGet(c => c.SabTvPriority) .Returns(priority); fakeConfig.SetupGet(c => c.SabTvCategory) .Returns(category); Mocker.GetMock(MockBehavior.Strict) .Setup( s => s.DownloadString( "http://192.168.5.55:2222/api?mode=addid&name=6107863&priority=0&pp=3&cat=tv&nzbname=This+is+an+Nzb&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns("ok"); //Act bool result = Mocker.Resolve().AddByUrl( "http://www.newzbin.com/browse/post/6107863/nzb", "This is an Nzb"); //Assert result.Should().BeTrue(); } [Test] public void AddByUrlError() { //Setup string sabHost = "192.168.5.55"; int sabPort = 2222; string apikey = "5c770e3197e4fe763423ee7c392c25d1"; string username = "admin"; string password = "pass"; var priority = SabnzbdPriorityType.Normal; string category = "tv"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); fakeConfig.SetupGet(c => c.SabTvPriority) .Returns(priority); fakeConfig.SetupGet(c => c.SabTvCategory) .Returns(category); Mocker.GetMock() .Setup(s => s.DownloadString(It.IsAny())) .Returns("error"); //Act var sabProvider = Mocker.Resolve(); var result = sabProvider.AddByUrl("http://www.nzbclub.com/nzb_download.aspx?mid=1950232", "This is an nzb"); //Assert Assert.IsFalse(result); ExceptionVerification.ExpectedWarns(1); } [Test] public void IsInQueue_True() { //Setup string sabHost = "192.168.5.55"; int sabPort = 2222; string apikey = "5c770e3197e4fe763423ee7c392c25d1"; string username = "admin"; string password = "pass"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); Mocker.GetMock(MockBehavior.Strict) .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\Queue.xml")); //Act bool result = Mocker.Resolve().IsInQueue("30 Rock - 1x05 - Title [SDTV]"); //Assert result.Should().BeTrue(); } [Test] public void IsInQueue_False_Empty() { //Setup string sabHost = "192.168.5.55"; int sabPort = 2222; string apikey = "5c770e3197e4fe763423ee7c392c25d1"; string username = "admin"; string password = "pass"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); Mocker.GetMock(MockBehavior.Strict) .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\QueueEmpty.xml")); //Act bool result = Mocker.Resolve().IsInQueue(String.Empty); //Assert Assert.IsFalse(result); } [Test] [ExpectedException(typeof(ApplicationException), ExpectedMessage = "API Key Incorrect")] public void IsInQueue_False_Error() { //Setup string sabHost = "192.168.5.55"; int sabPort = 2222; string apikey = "5c770e3197e4fe763423ee7c392c25d1"; string username = "admin"; string password = "pass"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); Mocker.GetMock(MockBehavior.Strict) .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\QueueError.xml")); //Act Mocker.Resolve().IsInQueue(String.Empty); } [Test] [TestCase(1, new[] { 2 }, "My Episode Title", QualityTypes.DVD, false, "My Series Name - 1x2 - My Episode Title [DVD]")] [TestCase(1, new[] { 2 }, "My Episode Title", QualityTypes.DVD, true, "My Series Name - 1x2 - My Episode Title [DVD] [Proper]")] [TestCase(1, new[] { 2 }, "", QualityTypes.DVD, true, "My Series Name - 1x2 - [DVD] [Proper]")] [TestCase(1, new[] { 2, 4 }, "My Episode Title", QualityTypes.HDTV, false, "My Series Name - 1x2-1x4 - My Episode Title [HDTV]")] [TestCase(1, new[] { 2, 4 }, "My Episode Title", QualityTypes.HDTV, true, "My Series Name - 1x2-1x4 - My Episode Title [HDTV] [Proper]")] [TestCase(1, new[] { 2, 4 }, "", QualityTypes.HDTV, true, "My Series Name - 1x2-1x4 - [HDTV] [Proper]")] public void sab_title(int seasons, int[] episodes, string title, QualityTypes quality, bool proper, string expected) { var series = Builder.CreateNew() .With(c => c.Title = "My Series Name") .Build(); var parsResult = new EpisodeParseResult() { AirDate = DateTime.Now, EpisodeNumbers = episodes.ToList(), Quality = new Quality(quality, proper), SeasonNumber = seasons, Series = series, EpisodeTitle = title }; //Act var actual = Mocker.Resolve().GetSabTitle(parsResult); //Assert Assert.AreEqual(expected, actual); } [TestCase(true, "My Series Name - Season 1 [Bluray720p] [Proper]")] [TestCase(false, "My Series Name - Season 1 [Bluray720p]")] public void sab_season_title(bool proper, string expected) { var series = Builder.CreateNew() .With(c => c.Title = "My Series Name") .Build(); var parsResult = new EpisodeParseResult() { AirDate = DateTime.Now, Quality = new Quality(QualityTypes.Bluray720p, proper), SeasonNumber = 1, Series = series, EpisodeTitle = "My Episode Title", FullSeason = true }; //Act var actual = Mocker.Resolve().GetSabTitle(parsResult); //Assert Assert.AreEqual(expected, actual); } [TestCase(true, "My Series Name - 2011-12-01 - My Episode Title [Bluray720p] [Proper]")] [TestCase(false, "My Series Name - 2011-12-01 - My Episode Title [Bluray720p]")] public void sab_daily_series_title(bool proper, string expected) { var series = Builder.CreateNew() .With(c => c.IsDaily = true) .With(c => c.Title = "My Series Name") .Build(); var parsResult = new EpisodeParseResult { AirDate = new DateTime(2011, 12,1), Quality = new Quality(QualityTypes.Bluray720p, proper), Series = series, EpisodeTitle = "My Episode Title", }; //Act var actual = Mocker.Resolve().GetSabTitle(parsResult); //Assert Assert.AreEqual(expected, actual); } [Test] [Explicit] public void AddNewzbingByUrlSuccess() { //Setup const string sabHost = "192.168.1.50"; const int sabPort = 8080; const string apikey = "f37dc33baec2e5566f5aec666287870d"; const string username = "root"; const string password = "*************"; const SabnzbdPriorityType priority = SabnzbdPriorityType.Normal; const string category = "tv"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(sabHost); fakeConfig.SetupGet(c => c.SabPort) .Returns(sabPort); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); fakeConfig.SetupGet(c => c.SabTvPriority) .Returns(priority); fakeConfig.SetupGet(c => c.SabTvCategory) .Returns(category); Mocker.SetConstant(new HttpProvider()); //Act bool result = Mocker.Resolve().AddByUrl( "http://www.newzbin.com/browse/post/6107863/nzb", "Added by unit tests."); //Assert result.Should().BeTrue(); } [Test] public void Get_Categories_Success_Passed_Values() { //Setup const string host = "192.168.5.55"; const int port = 2222; const string apikey = "5c770e3197e4fe763423ee7c392c25d1"; const string username = "admin"; const string password = "pass"; Mocker.GetMock(MockBehavior.Strict) .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=get_cats&output=json&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\Categories_json.txt")); //Act var result = Mocker.Resolve().GetCategories(host, port, apikey, username, password); //Assert result.Should().NotBeNull(); result.categories.Should().HaveCount(c => c > 0); } [Test] public void Get_Categories_Success_Config_Values() { //Setup const string host = "192.168.5.55"; const int port = 2222; const string apikey = "5c770e3197e4fe763423ee7c392c25d1"; const string username = "admin"; const string password = "pass"; var fakeConfig = Mocker.GetMock(); fakeConfig.SetupGet(c => c.SabHost) .Returns(host); fakeConfig.SetupGet(c => c.SabPort) .Returns(port); fakeConfig.SetupGet(c => c.SabApiKey) .Returns(apikey); fakeConfig.SetupGet(c => c.SabUsername) .Returns(username); fakeConfig.SetupGet(c => c.SabPassword) .Returns(password); Mocker.GetMock(MockBehavior.Strict) .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=get_cats&output=json&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\Categories_json.txt")); //Act var result = Mocker.Resolve().GetCategories(); //Assert result.Should().NotBeNull(); result.categories.Should().HaveCount(c => c > 0); } [Test] public void GetQueue_should_return_an_empty_list_when_the_queue_is_empty() { WithSabConfigValues(); Mocker.GetMock() .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\QueueEmpty.xml")); //Act var result = Mocker.Resolve().GetQueue(); //Assert result.Should().BeEmpty(); } [Test] [ExpectedException(typeof(ApplicationException), ExpectedMessage = "API Key Incorrect")] public void GetQueue_should_return_an_empty_list_when_there_is_an_error_getting_the_queue() { WithSabConfigValues(); Mocker.GetMock() .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\QueueError.xml")); //Act var result = Mocker.Resolve().GetQueue(); //Assert result.Should().BeEmpty(); } [Test] public void GetQueue_should_return_a_list_with_items_when_the_queue_has_items() { WithSabConfigValues(); Mocker.GetMock() .Setup(s => s.DownloadString("http://192.168.5.55:2222/api?mode=queue&output=xml&apikey=5c770e3197e4fe763423ee7c392c25d1&ma_username=admin&ma_password=pass")) .Returns(File.ReadAllText(@".\Files\Queue.xml")); //Act var result = Mocker.Resolve().GetQueue(); //Assert result.Should().HaveCount(2); } } }