def test_pick_indexers(self):
search.read_indexers_from_config()
indexers = search.pick_indexers()
self.assertEqual(3, len(indexers))
# Indexers with tv search and which support queries (actually searching for particular releases)
indexers = search.pick_indexers(query_supplied=True)
self.assertEqual(3, len(indexers))
# Indexers with tv search, including those that only provide a list of latest releases (womble) but excluding the one that needs a query (nzbclub)
indexers = search.pick_indexers(query_supplied=False)
self.assertEqual(3, len(indexers))
indexers = search.pick_indexers(identifier_key="tvdbid")
self.assertEqual(2, len(indexers))
self.assertEqual("NZBs.org", indexers[0].name)
self.assertEqual("DOGNzb", indexers[1].name)
indexers = search.pick_indexers("tv", identifier_key="imdbid")
self.assertEqual(1, len(indexers))
self.assertEqual("NZBs.org", indexers[0].name)
# Test category search, the first indexer is now only enabled for audio searches, so it will be only 2 available
search.indexers[0].indexer.settings["categories"] = ["Audio"]
indexers = search.pick_indexers(category="Movie")
self.assertEqual(2, len(indexers))
def testIndexersApiLimits(self):
config.settings.searching.generate_queries = []
self.newznab1.hitLimit = 3
self.newznab1.hitLimitResetTime = None
config.settings.indexers = [self.newznab1]
read_indexers_from_config()
search_request = SearchRequest()
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
dbsearch = Search(internal=True, time=arrow.utcnow().datetime)
dbsearch.save()
indexer = Indexer().get(name="newznab1")
#Two accesses one and 12 hours ago
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-1).datetime, type="search", url="", response_successful=True).save()
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-12).datetime, type="search", url="", response_successful=True).save()
self.assertEqual(1, len(search.pick_indexers(search_request)))
#Another one 20 hours ago, so limit should be reached
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-20).datetime, type="search", url="", response_successful=True).save()
self.assertEqual(0, len(search.pick_indexers(search_request)))
def testIndexersApiLimits(self):
config.settings.searching.generate_queries = []
self.newznab1.hitLimit = 3
self.newznab1.hitLimitResetTime = None
config.settings.indexers = [self.newznab1]
read_indexers_from_config()
search_request = SearchRequest()
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
dbsearch = Search(internal=True, time=arrow.utcnow().datetime)
dbsearch.save()
indexer = Indexer().get(name="newznab1")
#Two accesses one and 12 hours ago
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-1).datetime, type="search", url="", response_successful=True).save()
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-12).datetime, type="search", url="", response_successful=True).save()
self.assertEqual(1, len(search.pick_indexers(search_request)))
#Another one 20 hours ago, so limit should be reached
IndexerApiAccess(indexer=indexer, search=dbsearch, time=arrow.utcnow().replace(hours=-20).datetime, type="search", url="", response_successful=True).save()
self.assertEqual(0, len(search.pick_indexers(search_request)))
def test_pick_indexers(self):
config.settings.searching.generate_queries = []
config.settings.indexers.extend([self.newznab1, self.newznab2])
getIndexerSettingByName("womble").enabled = True
getIndexerSettingByName("womble").accessType = "both"
getIndexerSettingByName("nzbclub").enabled = True
getIndexerSettingByName("nzbclub").accessType = "both"
read_indexers_from_config()
search_request = SearchRequest()
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
# Indexers with tv search and which support queries (actually searching for particular releases)
search_request.query = "bla"
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
# Indexers with tv search, including those that only provide a list of latest releases (womble) but excluding the one that needs a query (nzbclub)
search_request.query = None
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
search_request.identifier_key = "tvdbid"
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
self.assertEqual("newznab1", indexers[0].name)
self.assertEqual("newznab2", indexers[1].name)
search_request.identifier_key = "imdbid"
search_request.category = getCategoryByName("movies")
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
self.assertEqual("newznab1", indexers[0].name)
# WIth query generation NZBClub should also be returned
infos.title_from_id = mock
config.settings.searching.generate_queries = ["internal"]
search_request.identifier_key = "tvdbid"
search_request.query = None
search_request.category = None
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
self.assertEqual("nzbclub", indexers[0].name)
self.assertEqual("newznab1", indexers[1].name)
self.assertEqual("newznab2", indexers[2].name)
# Test picking depending on internal, external, both
getIndexerSettingByName("womble").enabled = False
getIndexerSettingByName("nzbclub").enabled = False
getIndexerSettingByName("newznab1").accessType = "both"
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
config.settings.indexers = [self.newznab1, self.newznab2]
getIndexerSettingByName("newznab1").accessType = "external"
read_indexers_from_config()
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
getIndexerSettingByName("newznab1").accessType = "internal"
read_indexers_from_config()
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
def test_pick_indexers(self):
config.settings.searching.generate_queries = []
config.settings.indexers.extend([self.newznab1, self.newznab2])
getIndexerSettingByName("womble").enabled = True
getIndexerSettingByName("womble").accessType = "both"
getIndexerSettingByName("nzbclub").enabled = True
getIndexerSettingByName("nzbclub").accessType = "both"
read_indexers_from_config()
search_request = SearchRequest()
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
# Indexers with tv search and which support queries (actually searching for particular releases)
search_request.query = "bla"
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
# Indexers with tv search, including those that only provide a list of latest releases (womble) but excluding the one that needs a query (nzbclub)
search_request.query = None
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
search_request.identifier_key = "tvdbid"
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
self.assertEqual("newznab1", indexers[0].name)
self.assertEqual("newznab2", indexers[1].name)
search_request.identifier_key = "imdbid"
search_request.category = getCategoryByName("movies")
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
self.assertEqual("newznab1", indexers[0].name)
# WIth query generation NZBClub should also be returned
infos.title_from_id = mock
config.settings.searching.generate_queries = ["internal"]
search_request.identifier_key = "tvdbid"
search_request.query = None
search_request.category = None
indexers = search.pick_indexers(search_request)
self.assertEqual(3, len(indexers))
self.assertEqual("nzbclub", indexers[0].name)
self.assertEqual("newznab1", indexers[1].name)
self.assertEqual("newznab2", indexers[2].name)
# Test picking depending on internal, external, both
getIndexerSettingByName("womble").enabled = False
getIndexerSettingByName("nzbclub").enabled = False
getIndexerSettingByName("newznab1").accessType = "both"
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
config.settings.indexers = [self.newznab1, self.newznab2]
getIndexerSettingByName("newznab1").accessType = "external"
read_indexers_from_config()
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
getIndexerSettingByName("newznab1").accessType = "internal"
read_indexers_from_config()
search_request.internal = True
indexers = search.pick_indexers(search_request)
self.assertEqual(2, len(indexers))
search_request.internal = False
indexers = search.pick_indexers(search_request)
self.assertEqual(1, len(indexers))
