2011-05-22 19:53:21 +03:00
|
|
|
// ReSharper disable RedundantUsingDirective
|
2011-10-21 02:42:17 +03:00
|
|
|
|
2011-05-22 19:53:21 +03:00
|
|
|
using System;
|
2011-04-19 03:12:06 +03:00
|
|
|
using System.Collections.Generic;
|
2011-04-25 23:21:52 +03:00
|
|
|
using System.Net;
|
2011-04-19 03:12:06 +03:00
|
|
|
using System.ServiceModel.Syndication;
|
2011-11-14 03:22:18 +03:00
|
|
|
|
2011-06-18 05:00:44 +03:00
|
|
|
using FluentAssertions;
|
2011-06-03 00:06:46 +03:00
|
|
|
using NUnit.Framework;
|
2011-05-20 07:18:51 +03:00
|
|
|
using NzbDrone.Core.Model;
|
2011-01-29 08:10:22 +02:00
|
|
|
using NzbDrone.Core.Providers;
|
2011-04-04 06:50:12 +03:00
|
|
|
using NzbDrone.Core.Providers.Core;
|
2011-04-19 03:12:06 +03:00
|
|
|
using NzbDrone.Core.Providers.Indexer;
|
2011-05-19 06:55:35 +03:00
|
|
|
using NzbDrone.Core.Test.Framework;
|
2011-11-14 03:22:18 +03:00
|
|
|
using NzbDrone.Test.Common.AutoMoq;
|
2011-01-29 08:10:22 +02:00
|
|
|
|
2011-10-21 02:42:17 +03:00
|
|
|
namespace NzbDrone.Core.Test.ProviderTests
|
2011-01-29 08:10:22 +02:00
|
|
|
{
|
|
|
|
[TestFixture]
|
2011-04-19 03:12:06 +03:00
|
|
|
// ReSharper disable InconsistentNaming
|
2011-11-13 10:27:16 +03:00
|
|
|
public class IndexerProviderTest : CoreTest
|
2011-01-29 08:10:22 +02:00
|
|
|
{
|
2011-04-19 03:12:06 +03:00
|
|
|
[Test]
|
|
|
|
public void Init_indexer_test()
|
|
|
|
{
|
2011-12-15 07:15:53 +03:00
|
|
|
|
2011-04-19 03:12:06 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
Mocker.SetConstant(TestDbHelper.GetEmptyDatabase());
|
2011-01-29 08:10:22 +02:00
|
|
|
|
|
|
|
//Act
|
2011-12-15 07:15:53 +03:00
|
|
|
var indexerProvider = Mocker.Resolve<IndexerProvider>();
|
|
|
|
indexerProvider.InitializeIndexers(new List<IndexerBase> { Mocker.Resolve<MockIndexer>() });
|
2011-05-27 05:12:28 +03:00
|
|
|
var settings = indexerProvider.GetSettings(typeof(MockIndexer));
|
|
|
|
settings.Enable = true;
|
|
|
|
indexerProvider.SaveSettings(settings);
|
2011-01-29 08:10:22 +02:00
|
|
|
|
|
|
|
//Assert
|
2011-07-08 08:41:08 +03:00
|
|
|
indexerProvider.All();
|
2011-06-03 00:06:46 +03:00
|
|
|
|
|
|
|
|
2011-07-08 08:41:08 +03:00
|
|
|
indexerProvider.All().Should().HaveCount(1);
|
2011-06-03 00:06:46 +03:00
|
|
|
indexerProvider.GetEnabledIndexers().Should().HaveCount(1);
|
2011-04-22 22:16:52 +03:00
|
|
|
}
|
2011-05-26 07:25:59 +03:00
|
|
|
|
|
|
|
[Test]
|
2011-05-27 05:12:28 +03:00
|
|
|
public void Init_indexer_with_disabled_job()
|
2011-05-26 07:25:59 +03:00
|
|
|
{
|
2011-12-15 07:15:53 +03:00
|
|
|
|
2011-05-26 07:25:59 +03:00
|
|
|
|
2011-12-15 07:15:53 +03:00
|
|
|
Mocker.SetConstant(TestDbHelper.GetEmptyDatabase());
|
2011-05-26 07:25:59 +03:00
|
|
|
|
2011-05-27 05:12:28 +03:00
|
|
|
//Act
|
2011-12-15 07:15:53 +03:00
|
|
|
var indexerProvider = Mocker.Resolve<IndexerProvider>();
|
|
|
|
indexerProvider.InitializeIndexers(new List<IndexerBase> { Mocker.Resolve<MockIndexer>() });
|
2011-05-27 05:12:28 +03:00
|
|
|
var settings = indexerProvider.GetSettings(typeof(MockIndexer));
|
|
|
|
settings.Enable = false;
|
|
|
|
indexerProvider.SaveSettings(settings);
|
2011-05-26 07:25:59 +03:00
|
|
|
|
2011-05-27 05:12:28 +03:00
|
|
|
//Assert
|
2011-06-03 00:06:46 +03:00
|
|
|
|
2011-07-08 08:41:08 +03:00
|
|
|
indexerProvider.All().Should().HaveCount(1);
|
2011-06-03 00:06:46 +03:00
|
|
|
indexerProvider.GetEnabledIndexers().Should().BeEmpty();
|
2011-05-26 07:25:59 +03:00
|
|
|
}
|
2011-04-19 03:12:06 +03:00
|
|
|
}
|
2011-01-29 08:10:22 +02:00
|
|
|
|
2011-05-20 07:21:18 +03:00
|
|
|
public class MockIndexer : IndexerBase
|
2011-04-19 03:12:06 +03:00
|
|
|
{
|
2011-05-26 07:25:59 +03:00
|
|
|
public MockIndexer(HttpProvider httpProvider, ConfigProvider configProvider)
|
|
|
|
: base(httpProvider, configProvider)
|
2011-01-29 08:10:22 +02:00
|
|
|
{
|
2011-04-19 03:12:06 +03:00
|
|
|
}
|
2011-01-29 08:10:22 +02:00
|
|
|
|
2011-04-21 04:26:13 +03:00
|
|
|
protected override string[] Urls
|
2011-04-19 03:12:06 +03:00
|
|
|
{
|
|
|
|
get { return new[] { "www.google.com" }; }
|
|
|
|
}
|
2011-01-29 08:10:22 +02:00
|
|
|
|
2011-11-29 09:49:38 +03:00
|
|
|
protected override NetworkCredential Credentials
|
|
|
|
{
|
|
|
|
get { return null; }
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetEpisodeSearchUrls(string seriesTitle, int seasonNumber, int episodeNumber)
|
2011-05-26 07:25:59 +03:00
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
2011-11-29 09:49:38 +03:00
|
|
|
protected override IList<string> GetDailyEpisodeSearchUrls(string seriesTitle, DateTime date)
|
2011-04-25 23:21:52 +03:00
|
|
|
{
|
2011-11-29 09:49:38 +03:00
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetSeasonSearchUrls(string seriesTitle, int seasonNumber)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetPartialSeasonSearchUrls(string seriesTitle, int seasonNumber, int episodeWildcard)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
2011-04-25 23:21:52 +03:00
|
|
|
}
|
|
|
|
|
2011-04-19 03:12:06 +03:00
|
|
|
public override string Name
|
|
|
|
{
|
|
|
|
get { return "Mocked Indexer"; }
|
|
|
|
}
|
2011-01-29 08:10:22 +02:00
|
|
|
|
|
|
|
|
2011-04-19 03:12:06 +03:00
|
|
|
protected override string NzbDownloadUrl(SyndicationItem item)
|
|
|
|
{
|
|
|
|
return item.Links[0].Uri.ToString();
|
2011-01-29 08:10:22 +02:00
|
|
|
}
|
|
|
|
}
|
2011-04-25 20:48:16 +03:00
|
|
|
|
2011-05-20 07:21:18 +03:00
|
|
|
public class TestUrlIndexer : IndexerBase
|
2011-04-25 20:48:16 +03:00
|
|
|
{
|
2011-05-26 07:25:59 +03:00
|
|
|
public TestUrlIndexer(HttpProvider httpProvider, ConfigProvider configProvider)
|
|
|
|
: base(httpProvider, configProvider)
|
2011-04-25 20:48:16 +03:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
public override string Name
|
|
|
|
{
|
|
|
|
get { return "All Urls"; }
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override string[] Urls
|
|
|
|
{
|
|
|
|
get { return new[] { "http://rss.nzbmatrix.com/rss.php?cat=TV" }; }
|
|
|
|
}
|
|
|
|
|
2011-11-29 09:49:38 +03:00
|
|
|
protected override IList<string> GetEpisodeSearchUrls(string seriesTitle, int seasonNumber, int episodeNumber)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetDailyEpisodeSearchUrls(string seriesTitle, DateTime date)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetSeasonSearchUrls(string seriesTitle, int seasonNumber)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetPartialSeasonSearchUrls(string seriesTitle, int seasonNumber, int episodeWildcard)
|
2011-05-26 07:25:59 +03:00
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
2011-04-25 20:48:16 +03:00
|
|
|
protected override string NzbDownloadUrl(SyndicationItem item)
|
|
|
|
{
|
|
|
|
return "http://google.com";
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-05-20 07:21:18 +03:00
|
|
|
public class CustomParserIndexer : IndexerBase
|
2011-04-25 23:21:52 +03:00
|
|
|
{
|
2011-05-26 07:25:59 +03:00
|
|
|
public CustomParserIndexer(HttpProvider httpProvider, ConfigProvider configProvider)
|
|
|
|
: base(httpProvider, configProvider)
|
2011-04-25 23:21:52 +03:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
public override string Name
|
|
|
|
{
|
|
|
|
get { return "Custom parser"; }
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override string[] Urls
|
|
|
|
{
|
|
|
|
get { return new[] { "http://www.google.com" }; }
|
|
|
|
}
|
|
|
|
|
2011-11-29 09:49:38 +03:00
|
|
|
protected override IList<string> GetEpisodeSearchUrls(string seriesTitle, int seasonNumber, int episodeNumber)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetDailyEpisodeSearchUrls(string seriesTitle, DateTime date)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetSeasonSearchUrls(string seriesTitle, int seasonNumber)
|
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
|
|
|
protected override IList<string> GetPartialSeasonSearchUrls(string seriesTitle, int seasonNumber, int episodeWildcard)
|
2011-05-26 07:25:59 +03:00
|
|
|
{
|
|
|
|
throw new NotImplementedException();
|
|
|
|
}
|
|
|
|
|
2011-04-25 23:21:52 +03:00
|
|
|
protected override string NzbDownloadUrl(SyndicationItem item)
|
|
|
|
{
|
|
|
|
return "http://www.google.com";
|
|
|
|
}
|
|
|
|
|
2011-06-18 05:00:44 +03:00
|
|
|
protected override EpisodeParseResult CustomParser(SyndicationItem item, EpisodeParseResult currentResult)
|
2011-04-25 23:21:52 +03:00
|
|
|
{
|
2011-05-20 07:18:51 +03:00
|
|
|
if (currentResult == null) currentResult = new EpisodeParseResult();
|
2011-05-28 22:23:35 +03:00
|
|
|
currentResult.Language = LanguageType.Finnish;
|
2011-04-25 23:21:52 +03:00
|
|
|
return currentResult;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-04-10 05:44:01 +03:00
|
|
|
}
|