using System; using System.Collections.Generic; using System.Linq; using System.Net.Sockets; 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; using NzbDrone.Test.Common; using NzbDrone.Test.Common.AutoMoq; namespace NzbDrone.Core.Test.ProviderTests { [TestFixture] // ReSharper disable InconsistentNaming public class NewznabProviderTest : CoreTest { [Test] public void Save_should_clean_url_before_inserting() { //Setup var newznab = new NewznabDefinition { Name = "Newznab Provider", Enable = true, Url = "http://www.nzbdrone.com/gibberish/test.aspx?hello=world" }; var expectedUrl = "http://www.nzbdrone.com"; var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); //Act var result = Mocker.Resolve().Save(newznab); //Assert db.Single(result).Url.Should().Be(expectedUrl); } [Test] public void Save_should_clean_url_before_updating() { //Setup var newznab = new NewznabDefinition { Name = "Newznab Provider", Enable = true, Url = "http://www.nzbdrone.com" }; var expectedUrl = "http://www.nzbdrone.com"; var newUrl = "http://www.nzbdrone.com/gibberish/test.aspx?hello=world"; var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); newznab.Id = Convert.ToInt32(db.Insert(newznab)); newznab.Url = newUrl; //Act var result = Mocker.Resolve().Save(newznab); //Assert db.Single(result).Url.Should().Be(expectedUrl); } [Test] public void Save_should_clean_url_before_inserting_when_url_is_not_empty() { //Setup var newznab = new NewznabDefinition { Name = "Newznab Provider", Enable = true, Url = "" }; var expectedUrl = ""; var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); //Act var result = Mocker.Resolve().Save(newznab); //Assert db.Single(result).Url.Should().Be(expectedUrl); } [Test] public void Save_should_clean_url_before_updating_when_url_is_not_empty() { //Setup var newznab = new NewznabDefinition { Name = "Newznab Provider", Enable = true, Url = "http://www.nzbdrone.com" }; var expectedUrl = ""; var newUrl = ""; var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); newznab.Id = Convert.ToInt32(db.Insert(newznab)); newznab.Url = newUrl; //Act var result = Mocker.Resolve().Save(newznab); //Assert db.Single(result).Url.Should().Be(expectedUrl); } [Test] public void SaveAll_should_clean_urls_before_updating() { //Setup var definitions = Builder.CreateListOfSize(5) .All() .With(d => d.Url = "http://www.nzbdrone.com") .Build(); var expectedUrl = "http://www.nzbdrone.com"; var newUrl = "http://www.nzbdrone.com/gibberish/test.aspx?hello=world"; var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.InsertMany(definitions); definitions.ToList().ForEach(d => d.Url = newUrl); //Act Mocker.Resolve().SaveAll(definitions); //Assert db.Fetch().Where(d => d.Url == expectedUrl).Should().HaveCount(5); } [Test] public void Enabled_should_return_all_enabled_newznab_providers() { //Setup var definitions = Builder.CreateListOfSize(5) .TheFirst(2) .With(d => d.Enable = false) .TheLast(3) .With(d => d.Enable = true) .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.InsertMany(definitions); //Act var result = Mocker.Resolve().Enabled(); //Assert result.Should().HaveCount(3); result.All(d => d.Enable).Should().BeTrue(); } [Test] public void All_should_return_all_newznab_providers() { //Setup var definitions = Builder.CreateListOfSize(5) .TheFirst(2) .With(d => d.Enable = false) .TheLast(3) .With(d => d.Enable = true) .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.InsertMany(definitions); //Act var result = Mocker.Resolve().All(); //Assert result.Should().HaveCount(5); } [Test] public void Delete_should_delete_newznab_provider() { //Setup var definitions = Builder.CreateListOfSize(5) .TheFirst(2) .With(d => d.Enable = false) .TheLast(3) .With(d => d.Enable = true) .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.InsertMany(definitions); //Act Mocker.Resolve().Delete(1); //Assert var result = db.Fetch(); result.Should().HaveCount(4); result.Any(d => d.Id == 1).Should().BeFalse(); } [Test] public void InitializeNewznabIndexers_should_initialize_new_indexers() { //Setup var definitions = Builder.CreateListOfSize(5) .All() .With(d => d.Url = "http://www.nzbdrone.com") .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); //Act Mocker.Resolve().InitializeNewznabIndexers(definitions); //Assert var result = db.Fetch(); result.Should().HaveCount(5); result.Should().OnlyContain(i => i.BuiltIn); } [Test] public void InitializeNewznabIndexers_should_initialize_new_indexers_only() { //Setup var definitions = Builder.CreateListOfSize(5) .All() .With(d => d.Id = 0) .TheFirst(2) .With(d => d.Url = "http://www.nzbdrone2.com") .TheLast(3) .With(d => d.Url = "http://www.nzbdrone.com") .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.Insert(definitions[0]); db.Insert(definitions[2]); //Act Mocker.Resolve().InitializeNewznabIndexers(definitions); //Assert var result = db.Fetch(); result.Should().HaveCount(2); result.Where(d => d.Url == "http://www.nzbdrone.com").Should().HaveCount(1); result.Where(d => d.Url == "http://www.nzbdrone2.com").Should().HaveCount(1); } [Test] public void InitializeNewznabIndexers_should_update_matching_indexer_to_be_builtin() { //Setup var definition = Builder.CreateNew() .With(d => d.Url = "http://www.nzbdrone2.com") .With(d => d.BuiltIn = false) .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.Insert(definition); //Act Mocker.Resolve().InitializeNewznabIndexers(new List{ definition }); //Assert var result = db.Fetch(); result.Should().HaveCount(1); result.First().BuiltIn.Should().BeTrue(); } [Test] public void InitializeNewznabIndexers_should_not_blow_up_if_more_than_one_indexer_with_the_same_url_is_found() { //Setup var definition = Builder.CreateNew() .With(d => d.Url = "http://www.nzbdrone2.com") .With(d => d.BuiltIn = false) .Build(); var db = TestDbHelper.GetEmptyDatabase(); Mocker.SetConstant(db); db.Insert(definition); db.Insert(definition); //Act Mocker.Resolve().InitializeNewznabIndexers(new List { definition }); //Assert var result = db.Fetch(); result.Should().HaveCount(2); } [Test] public void CheckHostname_should_do_nothing_if_hostname_is_valid() { Mocker.Resolve().CheckHostname("http://www.google.com"); } [Test] public void CheckHostname_should_log_error_and_throw_exception_if_dnsHostname_is_invalid() { Assert.Throws(() => Mocker.Resolve().CheckHostname("http://BadName")); ExceptionVerification.ExpectedErrors(1); } } }