tests/pagegenerators_tests.py
#!/usr/bin/env python3
"""Test pagegenerators module."""
#
# (C) Pywikibot team, 2009-2024
#
# Distributed under the terms of the MIT license.
from __future__ import annotations
import calendar
import datetime
import logging
import sys
import unittest
from contextlib import suppress
from unittest import mock
import pywikibot
from pywikibot import date, pagegenerators
from pywikibot.exceptions import (
NoPageError,
ServerError,
UnknownExtensionError,
)
from pywikibot.pagegenerators import (
CategorizedPageGenerator,
PagesFromTitlesGenerator,
PreloadingGenerator,
WikibaseItemFilterPageGenerator,
)
from pywikibot.tools import has_module
from tests import join_data_path, unittest_print
from tests.aspects import (
DefaultSiteTestCase,
DeprecationTestCase,
RecentChangesTestCase,
TestCase,
WikidataTestCase,
)
from tests.tools_tests import GeneratorIntersectTestCase
from tests.utils import skipping
en_wp_page_titles = (
# just a bunch of randomly selected titles for English Wikipedia tests
'Eastern Sayan',
'The Addams Family (pinball)',
'Talk:Nowy Sącz',
'Talk:Battle of Węgierska Górka',
'Template:!',
'Template:Template',
)
en_wp_nopage_titles = (
'Cities in Burkina Faso',
'Talk:Hispanic (U.S. Census)',
'Talk:Stołpce',
'Template:!/Doc',
'Template:!/Meta',
'Template:Template/Doc',
'Template:Template/Meta',
)
class TestDryPageGenerators(TestCase):
"""Test pagegenerators methods."""
family = 'wikipedia'
code = 'en'
dry = True
titles = en_wp_page_titles + en_wp_nopage_titles
def setUp(self):
"""Setup test."""
super().setUp()
self.site = self.get_site()
def assertFunction(self, obj):
"""Assert function test."""
self.assertTrue(hasattr(pagegenerators, obj))
self.assertTrue(callable(getattr(pagegenerators, obj)))
def test_module_import(self):
"""Test module import."""
self.assertIn('pywikibot.pagegenerators', sys.modules)
def test_PagesFromTitlesGenerator(self):
"""Test PagesFromTitlesGenerator."""
self.assertFunction('PagesFromTitlesGenerator')
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
self.assertPageTitlesEqual(gen, self.titles)
def test_NamespaceFilterPageGenerator(self):
"""Test NamespaceFilterPageGenerator."""
self.assertFunction('NamespaceFilterPageGenerator')
site = self.site
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.NamespaceFilterPageGenerator(gen, 0, site)
self.assertLength(tuple(gen), 3)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.NamespaceFilterPageGenerator(gen, 1, site)
self.assertLength(tuple(gen), 4)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.NamespaceFilterPageGenerator(gen, 10, site)
self.assertLength(tuple(gen), 6)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.NamespaceFilterPageGenerator(gen, (1, 10), site)
self.assertLength(tuple(gen), 10)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.NamespaceFilterPageGenerator(
gen, ('Talk', 'Template'), site)
self.assertLength(tuple(gen), 10)
def test_RegexFilterPageGenerator(self):
"""Test RegexFilterPageGenerator."""
self.assertFunction('RegexFilterPageGenerator')
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(gen, '/doc')
self.assertPageTitlesEqual(gen,
('Template:!/Doc', 'Template:Template/Doc'))
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(gen, '/doc',
quantifier='none')
self.assertLength(tuple(gen), 11)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(gen, ['/doc', '/meta'])
self.assertPageTitlesEqual(gen,
('Template:!/Doc',
'Template:!/Meta',
'Template:Template/Doc',
'Template:Template/Meta'))
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(gen, ['/doc', '/meta'],
quantifier='none')
self.assertLength(tuple(gen), 9)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(gen, ['/doc', '/meta'],
quantifier='all')
self.assertPageTitlesEqual(gen, [])
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(
gen, ['Template', '/meta'], quantifier='all')
self.assertPageTitlesEqual(gen, ('Template:Template/Meta', ))
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.RegexFilterPageGenerator(
gen, ['template', '/meta'], quantifier='any')
self.assertPageTitlesEqual(gen,
('Template:Template',
'Template:!/Meta',
'Template:Template/Doc',
'Template:Template/Meta'))
gen = pagegenerators.PagesFromTitlesGenerator(self.titles,
site=self.site)
gen = pagegenerators.RegexFilterPageGenerator(
gen, ['template', '/meta'], quantifier='any',
ignore_namespace=False)
self.assertLength(tuple(gen), 6)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles,
site=self.site)
gen = pagegenerators.RegexFilterPageGenerator(
gen, ['template', '/meta'], quantifier='all',
ignore_namespace=False)
self.assertPageTitlesEqual(gen,
('Template:!/Meta',
'Template:Template/Meta'))
gen = pagegenerators.PagesFromTitlesGenerator(self.titles,
site=self.site)
gen = pagegenerators.RegexFilterPageGenerator(
gen, ['template', '/meta'], quantifier='none',
ignore_namespace=False)
self.assertLength(tuple(gen), 7)
def test_RegexBodyFilterPageGenerator(self):
"""Test RegexBodyFilterPageGenerator."""
self.assertFunction('RegexBodyFilterPageGenerator')
gen = pagegenerators.PagesFromTitlesGenerator(self.titles,
site=self.site)
pages = []
for p in gen:
p.text = f'This is the content of {p.title()} as a sample'
pages.append(p)
gen = pagegenerators.RegexBodyFilterPageGenerator(iter(pages), '/doc')
self.assertPageTitlesEqual(gen,
('Template:!/Doc', 'Template:Template/Doc'))
gen = pagegenerators.RegexBodyFilterPageGenerator(iter(pages), 'This')
self.assertPageTitlesEqual(gen, self.titles)
gen = pagegenerators.RegexBodyFilterPageGenerator(iter(pages), 'talk',
quantifier='none')
self.assertLength(tuple(gen), 9)
class BasetitleTestCase(TestCase):
"""Class providing base_title attribute."""
family = 'wikisource'
code = 'en'
base_title = ('Page:06-24-1920 -The Story of the Jones County '
'Calf Case.pdf/{}')
def setUp(self):
"""Setup tests."""
super().setUp()
self.site = self.get_site()
self.titles = [self.base_title.format(i) for i in range(1, 11)]
class TestPagesFromPageidGenerator(BasetitleTestCase):
"""Test PagesFromPageidGenerator method."""
def test_PagesFromPageidGenerator(self):
"""Test PagesFromPageidGenerator."""
gen_pages = pagegenerators.PagesFromTitlesGenerator(self.titles,
self.site)
pageids = (page.pageid for page in gen_pages)
gen = pagegenerators.PagesFromPageidGenerator(pageids, self.site)
self.assertPageTitlesEqual(gen, self.titles)
class TestCategoryFilterPageGenerator(BasetitleTestCase):
"""Test CategoryFilterPageGenerator method."""
category_list = ['Category:Validated']
def setUp(self):
"""Setup tests."""
super().setUp()
self.catfilter_list = [pywikibot.Category(self.site, cat)
for cat in self.category_list]
def test_CategoryFilterPageGenerator(self):
"""Test CategoryFilterPageGenerator."""
site = self.site
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.CategoryFilterPageGenerator(
gen, self.catfilter_list)
self.assertLength(tuple(gen), 10)
class TestQualityFilterPageGenerator(BasetitleTestCase):
"""Test QualityFilterPageGenerator methods."""
cached = True
base_title = 'Page:Popular Science Monthly Volume 1.djvu/{}'
def test_QualityFilterPageGenerator(self):
"""Test QualityFilterPageGenerator."""
site = self.site
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.QualityFilterPageGenerator(gen, [0, 3])
self.assertLength(tuple(gen), 7)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
gen = pagegenerators.QualityFilterPageGenerator(gen, [4])
self.assertLength(tuple(gen), 3)
gen = pagegenerators.PagesFromTitlesGenerator(self.titles, site)
self.assertLength(tuple(gen), 10)
class EdittimeFilterPageGeneratorTestCase(TestCase):
"""Test EdittimeFilterPageGenerator."""
family = 'wikipedia'
code = 'en'
titles = en_wp_page_titles
def test_first_edit(self):
"""Test first edit."""
expect = (
'The Addams Family (pinball)',
'Talk:Nowy Sącz',
'Template:Template',
)
gen = PagesFromTitlesGenerator(self.titles, self.site)
gen = pagegenerators.EdittimeFilterPageGenerator(
gen, first_edit_end=datetime.datetime(2006, 1, 1))
self.assertPageTitlesEqual(gen, titles=expect, site=self.site)
gen = PagesFromTitlesGenerator(self.titles, self.site)
opposite_pages = pagegenerators.EdittimeFilterPageGenerator(
gen, first_edit_start=datetime.datetime(2006, 1, 1))
for p in opposite_pages:
self.assertIsInstance(p, pywikibot.Page)
self.assertNotIn(p.title(), expect)
def test_last_edit(self):
"""Test last edit."""
two_days_ago = datetime.datetime.now() - datetime.timedelta(days=2)
nine_days_ago = datetime.datetime.now() - datetime.timedelta(days=9)
gen = PagesFromTitlesGenerator(['Wikipedia:Sandbox'], self.site)
gen = pagegenerators.EdittimeFilterPageGenerator(
gen, last_edit_start=two_days_ago)
self.assertLength(list(gen), 1)
gen = PagesFromTitlesGenerator(['Wikipedia:Sandbox'], self.site)
gen = pagegenerators.EdittimeFilterPageGenerator(
gen, last_edit_end=two_days_ago)
self.assertIsEmpty(list(gen))
gen = PagesFromTitlesGenerator(['Template:Side box'], self.site)
gen = pagegenerators.EdittimeFilterPageGenerator(
gen, last_edit_end=nine_days_ago)
self.assertLength(list(gen), 1)
gen = PagesFromTitlesGenerator(['Template:Side box'], self.site)
gen = pagegenerators.EdittimeFilterPageGenerator(
gen, last_edit_start=nine_days_ago)
self.assertIsEmpty(list(gen))
class RedirectFilterPageGeneratorTestCase(TestCase):
"""Test RedirectFilterPageGenerator."""
family = 'wikipedia'
code = 'en'
def test_redirect_filter(self):
"""Test RedirectFilterPageGenerator with handle_args()."""
from pywikibot.pagegenerators import RedirectFilterPageGenerator
gf = pagegenerators.GeneratorFactory(site=self.site)
args = gf.handle_args(['-randomredirect:3', '-page:Main_Page'])
self.assertIsEmpty(args)
gen = gf.getCombinedGenerator()
pages = list(gen)
gen = RedirectFilterPageGenerator(pages, no_redirects=True)
self.assertLength(list(gen), 1)
gen = RedirectFilterPageGenerator(pages, no_redirects=False)
self.assertLength(list(gen), 3)
class SubpageFilterGeneratorTestCase(TestCase):
"""Test SubpageFilterGenerator."""
family = 'wikipedia'
code = 'test'
def test_subpage_filter(self):
"""Test SubpageFilterGenerator."""
site = self.get_site()
test_cat = pywikibot.Category(site, 'Subpage testing')
gen = CategorizedPageGenerator(test_cat)
gen = pagegenerators.SubpageFilterGenerator(gen, 0)
expect_0 = ('/home/test',)
self.assertPageTitlesEqual(gen, titles=expect_0, site=site)
gen = CategorizedPageGenerator(test_cat)
gen = pagegenerators.SubpageFilterGenerator(gen, 3)
expect_3 = (
'/home/test',
'User:Sn1per/ProtectTest1/test',
'User:Sn1per/ProtectTest1/test/test',
)
self.assertPageTitlesEqual(gen, titles=expect_3, site=site)
class PetScanPageGeneratorTestCase(TestCase):
"""Test PetScanPageGenerator."""
family = 'wikipedia'
code = 'test'
def test_petscan(self):
"""Test PetScanPageGenerator."""
site = self.get_site()
gen = pagegenerators.PetScanPageGenerator(['Pywikibot Protect Test'],
True, None, site)
with skipping(ServerError):
self.assertPageTitlesEqual(gen, titles=(
'User:Sn1per/ProtectTest1', 'User:Sn1per/ProtectTest2'),
site=site)
gen = pagegenerators.PetScanPageGenerator(['Pywikibot Protect Test'],
False, None, site)
self.assertPageTitlesEqual(gen, titles=('User:Sn1per/ProtectTest1',
'User:Sn1per/ProtectTest2'),
site=site)
gen = pagegenerators.PetScanPageGenerator(
['Pywikibot PetScan Test',
'Pywikibot Category That Needs&ToBe!Encoded',
'Test'], True, None, site)
self.assertPageTitlesEqual(gen, titles=('User:Sn1per/PetScanTest1', ),
site=site)
class PagePilePageGeneratorTestCase(TestCase):
"""Test PagePilePageGenerator."""
family = 'wikipedia'
code = 'ro'
def test_PagePilePageGenerator(self):
"""Test PagePilePageGenerator."""
gen = pagegenerators.PagePilePageGenerator(id=53158)
self.assertLength(list(gen), 215)
self.assertEqual(gen.site, pywikibot.Site('wikipedia:ro'))
class TestRepeatingGenerator(RecentChangesTestCase):
"""Test RepeatingGenerator."""
def test_RepeatingGenerator(self):
"""Test RepeatingGenerator."""
gen = pagegenerators.RepeatingGenerator(
self.site.recentchanges,
key_func=lambda x: x['revid'],
sleep_duration=10,
reverse=True,
namespaces=[0],
total=self.length)
items = list(gen)
self.assertLength(items, self.length)
timestamps = [pywikibot.Timestamp.fromISOformat(item['timestamp'])
for item in items]
self.assertEqual(sorted(timestamps), timestamps)
itemsset = set()
for item in items:
self.assertEqual(item['ns'], 0)
revid = item['revid']
self.assertNotIn(revid, itemsset)
itemsset.add(revid)
class TestTextIOPageGenerator(DefaultSiteTestCase):
"""Test loading pages from a textfile."""
dry = True
title_columns = {
'case-sensitive': 0,
'first-letter': 1,
}
expected_titles = (
('file', 'File'),
('bracket', 'Bracket'),
('MediaWiki:Test', 'MediaWiki:Test'),
('under score', 'Under score'),
('Upper case', 'Upper case'),
)
def test_brackets(self):
"""Test TextIOPageGenerator with brackets."""
filename = join_data_path('pagelist-brackets.txt')
site = self.get_site()
titles = list(pagegenerators.TextIOPageGenerator(filename, site))
self.assertLength(titles, self.expected_titles)
expected_titles = [
expected_title[self.title_columns[site.namespaces[page.namespace()]
.case]]
for expected_title, page in zip(self.expected_titles, titles)]
self.assertPageTitlesEqual(titles, expected_titles)
def test_lines(self):
"""Test TextIOPageGenerator with newlines."""
filename = join_data_path('pagelist-lines.txt')
site = self.get_site()
titles = list(pagegenerators.TextIOPageGenerator(filename, site))
self.assertLength(titles, self.expected_titles)
expected_titles = [
expected_title[self.title_columns[site.namespaces[page.namespace()]
.case]]
for expected_title, page in zip(self.expected_titles, titles)]
self.assertPageTitlesEqual(titles, expected_titles)
@unittest.mock.patch('pywikibot.comms.http.fetch', autospec=True)
def test_url(self, mock_fetch):
"""Test TextIOPageGenerator with URL."""
# Mock return value of fetch()
fetch_return = unittest.mock.Mock()
fetch_return.text = '\n'.join(
[title[0] for title in self.expected_titles])
mock_fetch.return_value = fetch_return
site = self.get_site()
titles = list(
pagegenerators.TextIOPageGenerator('http://www.someurl.org', site))
self.assertLength(titles, self.expected_titles)
class TestYearPageGenerator(DefaultSiteTestCase):
"""Test the year page generator."""
def test_basic(self):
"""Test YearPageGenerator."""
site = self.get_site()
# Some languages are missing (T85681)
if site.lang not in date.formats['YearBC']:
self.skipTest(f'Date formats for {site.lang!r} language are'
' missing from date.py')
start = -20
end = 2026
i = 0
for page in pagegenerators.YearPageGenerator(start, end, site):
self.assertIsInstance(page, pywikibot.Page)
self.assertEqual(date.formatYear(site.lang, start + i),
page.title())
self.assertNotEqual(page.title(), '0')
i += 1
if start + i == 0:
i += 1
self.assertEqual(start + i - 1, end)
class TestDayPageGenerator(DefaultSiteTestCase):
"""Test the day page generator."""
def _run_test(self, start_month=1, end_month=12, year=2000):
"""Test method for DayPageGenerator."""
params = {
'start_month': start_month,
'end_month': end_month,
'site': self.site,
}
if year != 2000:
params['year'] = year
# use positional parameter
gen1 = pagegenerators.DayPageGenerator(
start_month, end_month, self.site, year)
# use keyworded parameter and default for year
gen2 = pagegenerators.DayPageGenerator(**params)
for page in gen1:
self.assertIsInstance(page, pywikibot.Page)
self.assertTrue(page.isAutoTitle)
expected = []
for month in range(start_month, end_month + 1):
expected += [
date.format_date(month, day, self.site)
for day in range(1, calendar.monthrange(year, month)[1] + 1)
]
self.assertPageTitlesEqual(gen2, expected)
def test_basic(self):
"""General test for day page generator."""
self._run_test()
def test_year_2001(self):
"""Test for day page generator of year 2001."""
self._run_test(2, year=2001)
def test_year_2100(self):
"""Test for day page generator of year 2100."""
self._run_test(end_month=2, year=2100)
def test_start_0(self):
"""Test for day page generator with startMonth 0."""
with self.assertRaisesRegex(
calendar.IllegalMonthError,
'bad month number 0; must be 1-12'):
self._run_test(0)
def test_end_13(self):
"""Test for day page generator with endMonth 13."""
with self.assertRaisesRegex(
calendar.IllegalMonthError,
'bad month number 13; must be 1-12'):
self._run_test(12, 13)
class TestPreloadingGenerator(DefaultSiteTestCase):
"""Test preloading generator on lists."""
def test_basic(self):
"""Test PreloadingGenerator with a list of pages."""
mainpage = self.get_mainpage()
links = [page for page in self.site.pagelinks(mainpage, total=20)
if page.exists()]
count = 0
for count, page in enumerate(
PreloadingGenerator(links, groupsize=20), start=1):
self.assertIsInstance(page, pywikibot.Page)
self.assertIsInstance(page.exists(), bool)
self.assertLength(page._revisions, 1)
self.assertIsNotNone(page._revisions[page._revid].text)
self.assertFalse(hasattr(page, '_pageprops'))
self.assertLength(links, count)
def test_low_step(self):
"""Test PreloadingGenerator with a list of pages."""
mainpage = self.get_mainpage()
links = [page for page in self.site.pagelinks(mainpage, total=20)
if page.exists()]
count = 0
for count, page in enumerate(
PreloadingGenerator(links, groupsize=10), start=1):
self.assertIsInstance(page, pywikibot.Page)
self.assertIsInstance(page.exists(), bool)
self.assertLength(page._revisions, 1)
self.assertIsNotNone(page._revisions[page._revid].text)
self.assertFalse(hasattr(page, '_pageprops'))
self.assertLength(links, count)
def test_order(self):
"""Test outcome is following same order of input."""
mainpage = self.get_mainpage()
links = [page for page in self.site.pagelinks(mainpage, total=20)
if page.exists()]
count = -1
for count, page in enumerate(PreloadingGenerator(links, groupsize=10)):
self.assertIsInstance(page, pywikibot.Page)
self.assertIsInstance(page.exists(), bool)
self.assertLength(page._revisions, 1)
self.assertIsNotNone(page._revisions[page._revid].text)
self.assertFalse(hasattr(page, '_pageprops'))
self.assertEqual(page, links[count])
self.assertLength(links, count + 1)
class TestDequePreloadingGenerator(DefaultSiteTestCase):
"""Test preloading generator on lists."""
def test_deque_preloading(self):
"""Test pages being added to a DequePreloadingGenerator."""
mainpage = self.get_mainpage()
pages = pywikibot.tools.collections.DequeGenerator([mainpage])
gen = pagegenerators.DequePreloadingGenerator(pages)
pages_out = []
for page in gen:
pages_out.append(page)
# Add a page to the generator
if not page.isTalkPage():
pages.extend([page.toggleTalkPage()])
self.assertIn(mainpage, pages_out)
self.assertIn(mainpage.toggleTalkPage(), pages_out)
self.assertLength(pages_out, 2)
self.assertTrue(pages_out[1].isTalkPage())
for page in pages_out:
self.assertIsInstance(page, pywikibot.Page)
class TestPreloadingEntityGenerator(WikidataTestCase):
"""Test preloading item generator."""
def test_non_item_gen(self):
"""Test TestPreloadingEntityGenerator with getReferences."""
site = self.get_site()
page = pywikibot.Page(site, 'Property:P31')
ref_gen = page.getReferences(follow_redirects=False, total=5)
gen = pagegenerators.PreloadingEntityGenerator(ref_gen)
for ipage in gen:
self.assertIsInstance(ipage, pywikibot.ItemPage)
class WikibaseItemFilterPageGeneratorTestCase(TestCase):
"""Test WikibaseItemFilterPageGenerator."""
family = 'wikipedia'
code = 'en'
def test_filter_pages_with_item(self):
"""Test WikibaseItemFilterPageGenerator on pages with item."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Main_Page')
gen = gf.getCombinedGenerator()
pages = list(gen)
gen = WikibaseItemFilterPageGenerator(pages, has_item=True)
self.assertLength(list(gen), 1)
gen = WikibaseItemFilterPageGenerator(pages, has_item=False)
self.assertLength(list(gen), 0)
def test_filter_pages_without_item(self):
"""Test WikibaseItemFilterPageGenerator on pages without item."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Talk:Main_Page')
gen = gf.getCombinedGenerator()
pages = list(gen)
gen = WikibaseItemFilterPageGenerator(pages, has_item=True)
self.assertLength(list(gen), 0)
gen = WikibaseItemFilterPageGenerator(pages, has_item=False)
self.assertLength(list(gen), 1)
class DryFactoryGeneratorTest(TestCase):
"""Dry tests for pagegenerators.GeneratorFactory."""
family = 'wikipedia'
code = 'en'
dry = True
def test_one_namespace(self):
"""Test one namespace."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:2')
self.assertEqual(gf.namespaces, {2})
def test_two_namespaces(self):
"""Test two namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:2')
gf.handle_arg('-ns:Talk')
self.assertEqual(gf.namespaces, {2, 1})
def test_two_named_namespaces(self):
"""Test two named namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:Talk,File')
self.assertEqual(gf.namespaces, {1, 6})
def test_two_numeric_namespaces(self):
"""Test two namespaces delimited by colon."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:1,6')
self.assertEqual(gf.namespaces, {1, 6})
def test_immutable_namespaces_on_read(self):
"""Test immutable namespaces on read."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:1,6')
self.assertEqual(gf.namespaces, {1, 6})
self.assertIsInstance(gf.namespaces, frozenset)
with self.assertRaises(RuntimeError):
gf.handle_arg('-ns:0')
self.assertEqual(gf.namespaces, {1, 6})
def test_unsupported_quality_level_filter(self):
"""Test unsupported option."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
with self.assertRaises(UnknownExtensionError):
gf.handle_arg('-ql:2')
def test_one_excluded_namespaces(self):
"""Test one excluded namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:not:2')
ns = set(range(16))
ns.remove(2)
self.assertTrue(ns.issubset(gf.namespaces))
def test_two_excluded_namespaces(self):
"""Test two excluded namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:not:2')
gf.handle_arg('-ns:not:Talk')
ns = set(range(16))
ns.remove(2)
ns.remove(1)
self.assertTrue(ns.issubset(gf.namespaces))
def test_two_excluded_named_namespaces(self):
"""Test two excluded named namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:not:Talk,File')
ns = set(range(16))
ns.remove(1)
ns.remove(6)
self.assertTrue(ns.issubset(gf.namespaces))
def test_two_excluded_numeric_namespaces(self):
"""Test two excluded namespaces delimited by colon."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:not:1,6')
ns = set(range(16))
ns.remove(1)
ns.remove(6)
self.assertTrue(ns.issubset(gf.namespaces))
def test_mixed_namespaces_with_exclusion(self):
"""Test mixed namespaces with exclusion."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:not:2,File')
gf.handle_arg('-ns:not:3,4,5')
gf.handle_arg('-ns:6,7')
ns = set(range(16))
for i in range(2, 6):
ns.remove(i)
self.assertTrue(ns.issubset(gf.namespaces))
def test_given_namespaces_with_exclusion(self):
"""Test mixed namespaces with exclusion."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg('-ns:1,2,3,4,5')
gf.handle_arg('-ns:not:User')
self.assertEqual(gf.namespaces, {1, 3, 4, 5})
def test_invalid_arg(self):
"""Test invalid / non-generator arguments."""
gf = pagegenerators.GeneratorFactory(site=self.get_site())
self.assertFalse(gf.handle_arg('-foobar'))
self.assertFalse(gf.handle_arg('barbaz'))
self.assertFalse(gf.handle_arg('-ì'))
self.assertFalse(gf.handle_arg('ì'))
class TestItemClaimFilterPageGenerator(WikidataTestCase):
"""Test item claim filter page generator generator."""
def _simple_claim_test(self, prop, claim, qualifiers, valid, negate=False):
"""Test given claim on sample (India) page.
:param prop: the property to check
:param claim: the claim the property should contain
:param qualifiers: qualifiers to check or None
:param valid: true if the page should be yielded by the generator,
false otherwise
:param negate: true to swap the filters' behavior
"""
item = pywikibot.ItemPage(self.get_repo(), 'Q668')
gen = pagegenerators.ItemClaimFilterPageGenerator([item], prop, claim,
qualifiers, negate)
pages = set(gen)
if valid:
self.assertLength(pages, 1)
else:
self.assertIsEmpty(pages)
def _get_council_page(self):
"""Return United Nations Security Council Wikidata page."""
site = self.get_site()
return pywikibot.Page(site, 'Q37470')
def test_valid_qualifiers(self):
"""Test ItemClaimFilterPageGenerator using valid qualifiers."""
qualifiers = {
'P580': pywikibot.WbTime(1950, 1, 1, precision=9,
site=self.get_site()),
'P582': '1951',
}
self._simple_claim_test('P463', self._get_council_page(), qualifiers,
True)
def test_invalid_qualifiers(self):
"""Test ItemClaimFilterPageGenerator with invalid qualifiers."""
qualifiers = {
'P580': 1950,
'P582': pywikibot.WbTime(1960, 1, 1, precision=9,
site=self.site),
}
self._simple_claim_test('P463', self._get_council_page(), qualifiers,
False)
def test_nonexisting_qualifiers(self):
"""Test ItemClaimFilterPageGenerator on sample page.
The item does not have the searched qualifiers.
"""
qualifiers = {
'P370': pywikibot.WbTime(1950, 1, 1, precision=9,
site=self.get_site()),
'P232': pywikibot.WbTime(1960, 1, 1, precision=9,
site=self.get_site()),
}
self._simple_claim_test('P463', self._get_council_page(), qualifiers,
False)
def test_no_qualifiers(self):
"""Test ItemClaimFilterPageGenerator without qualifiers."""
self._simple_claim_test('P474', '+91', None, True)
self._simple_claim_test('P463', 'Q37470', None, True)
self._simple_claim_test('P1334', '28,97.4,0.1', None, True)
self._simple_claim_test('P1334', '28,96,0.01', None, False)
def test_negative_filter(self):
"""Test negative ItemClaimFilterPageGenerator."""
self._simple_claim_test('P463', 'Q37470', None, False, True)
self._simple_claim_test('P463', 'Q37471', None, True, True)
def test_item_from_page(self):
"""Test ItemPage can be obtained form Page."""
site = pywikibot.Site('en', 'wikipedia')
page = pywikibot.Page(site, 'India')
gen = pagegenerators.ItemClaimFilterPageGenerator(
[page], 'P463', self._get_council_page())
pages = set(gen)
self.assertEqual(pages.pop(), page)
class TestFactoryGenerator(DefaultSiteTestCase):
"""Test pagegenerators.GeneratorFactory."""
def test_combined_generator(self):
"""Test getCombinedGenerator with generator parameter."""
gf = pagegenerators.GeneratorFactory()
gen = gf.getCombinedGenerator(gen='ABC')
self.assertEqual(tuple(gen), ('A', 'B', 'C'))
def test_intersect_generator(self):
"""Test getCombinedGenerator with -intersect option."""
gf = pagegenerators.GeneratorFactory()
gf.handle_arg('-intersect')
# check whether the generator works for both directions
patterns = ['Python 3.7-dev', 'Pywikibot 7.0.dev']
for index in range(2):
with self.subTest(index=index):
gf.gens = [patterns[index]]
gen = gf.getCombinedGenerator(gen=patterns[index - 1])
self.assertEqual(''.join(gen), 'Pyot 7.dev')
# check whether the generator works for a very long text
patterns.append('PWB 7+ unittest developed with a very long text.')
with self.subTest(patterns=patterns):
gf.gens = patterns
gen = gf.getCombinedGenerator()
self.assertEqual(''.join(gen), 'P 7tedvoy.')
# check whether an early stop fits
with self.subTest(comment='Early stop'):
gf.gens = 'ABC', 'A Big City'
gen = gf.getCombinedGenerator()
self.assertEqual(''.join(gen), 'ABC')
with self.subTest(comment='Commutative'):
gf.gens = 'ABB', 'BB'
gen1 = gf.getCombinedGenerator()
gf2 = pagegenerators.GeneratorFactory()
gf2.handle_arg('-intersect')
gf2.gens = 'BB', 'ABB'
gen2 = gf2.getCombinedGenerator()
self.assertEqual(list(gen1), list(gen2))
def test_ns(self):
"""Test namespace option."""
gf = pagegenerators.GeneratorFactory()
gf.handle_arg('-ns:1')
gen = gf.getCombinedGenerator()
self.assertIsNone(gen)
def test_allpages_default(self):
"""Test allpages generator."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start:!'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLessEqual(len(pages), 10)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
self.assertEqual(page.namespace(), 0)
def test_allpages_ns(self):
"""Test allpages generator with namespace argument."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start:!'))
gf.handle_arg('-limit:10')
gf.handle_arg('-ns:1')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLessEqual(len(pages), 10)
self.assertPagesInNamespaces(gen, 1)
def test_regexfilter_default(self):
"""Test allpages generator with titleregex filter."""
gf = pagegenerators.GeneratorFactory()
# Matches titles with the same two or more continuous characters
self.assertTrue(gf.handle_arg('-start'))
self.assertTrue(gf.handle_arg('-titleregex:(.)\\1+'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 10)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
self.assertRegex(page.title().lower(), '(.)\\1+')
def test_regexfilter_ns_after(self):
"""Test allpages generator with titleregex and namespace filter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
self.assertTrue(gf.handle_arg('-titleregex:.*'))
gf.handle_arg('-ns:1')
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
pages = list(gen)
self.assertLessEqual(len(pages), 10)
self.assertPagesInNamespaces(pages, 1)
def test_regexfilter_ns_before(self):
"""Test allpages generator with namespace and titleregex filter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
gf.handle_arg('-ns:1')
self.assertTrue(gf.handle_arg('-titleregex:.*'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 10)
self.assertPagesInNamespaces(pages, 1)
def test_regexfilternot_default(self):
"""Test allpages generator with titleregexnot filter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
# matches titles with less than 11 characters
self.assertTrue(gf.handle_arg('-titleregexnot:.{11,}'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 10)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
self.assertNotRegex(page.title().lower(), '.{11,}')
def test_regexfilternot_ns_after(self):
"""Test allpages generator with titleregexnot and namespace filter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
self.assertTrue(gf.handle_arg('-titleregexnot:zzzz'))
gf.handle_arg('-ns:1')
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
pages = list(gen)
self.assertLessEqual(len(pages), 10)
self.assertPagesInNamespaces(pages, 1)
def test_regexfilternot_ns_before(self):
"""Test allpages generator with namespace and titleregexnot filter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
gf.handle_arg('-ns:1')
self.assertTrue(gf.handle_arg('-titleregexnot:zzzz'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 10)
self.assertPagesInNamespaces(pages, 1)
def test_allpages_with_two_ns(self):
"""Test that allpages fails with two ns as parameter."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-start'))
gf.handle_arg('-ns:3,1')
# allpages only accepts a single namespace, and will raise a
# TypeError if self.namespaces contains more than one namespace.
with self.assertRaisesRegex(
TypeError,
'allpages module does not support multiple namespaces'):
gf.getCombinedGenerator()
def test_prefixing_default(self):
"""Test prefixindex generator."""
gf = pagegenerators.GeneratorFactory()
self.assertTrue(gf.handle_arg('-prefixindex:a'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLessEqual(len(pages), 10)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
self.assertTrue(page.title().lower().startswith('a'))
def test_prefixing_ns(self):
"""Test prefixindex generator with namespace filter."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-prefixindex:a')
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 1)
def test_recentchanges_timespan(self):
"""Test recentchanges generator with offset and duration params."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-recentchanges:120,70')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
with self.assertRaises(ValueError):
gf.handle_arg('-recentchanges:3,2,1')
with self.assertRaises(ValueError):
gf.handle_arg('-recentchanges:12,-12')
with self.assertRaises(ValueError):
gf.handle_arg('-recentchanges:visualeditor,3,2,1')
with self.assertRaises(ValueError):
gf.handle_arg('-recentchanges:"mobile edit,-10,20"')
def test_recentchanges_rctag(self):
"""Test recentchanges generator with recent changes tag."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:0,2,4')
gf.handle_arg('-recentchanges:visualeditor,500')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespacesAll(gen, {0, 2, 4}, skip=True)
def test_recentchanges_default(self):
"""Test recentchanges generator with default namespace setting."""
if self.site.family.name in ('wpbeta', 'wsbeta'):
self.skipTest(
f'Skipping {self.site} due to too many autoblocked users')
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:0,1,2')
gf.handle_arg('-recentchanges:50')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespacesAll(gen, {0, 1, 2}, skip=True)
def test_recentchanges_ns(self):
"""Test recentchanges generator with namespace."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-recentchanges:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 1)
def test_recentchanges_ns_multi(self):
"""Test recentchanges generator with multiple namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-ns:3')
gf.handle_arg('-recentchanges:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, {1, 3})
def test_pageid(self):
"""Test pageid parameter."""
# Get reference pages and their pageids.
gf = pagegenerators.GeneratorFactory(site=self.get_site())
self.assertTrue(gf.handle_arg('-random'))
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
pages = list(gen)
self.assertLength(pages, 10)
# pipe-separated used as test reference.
pageids = '|'.join(str(page.pageid) for page in pages)
# Get by pageids.
gf = pagegenerators.GeneratorFactory(site=self.get_site())
gf.handle_arg(f'-pageid:{pageids}')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages_from_pageid = list(gen)
self.assertLength(pages_from_pageid, 10)
for page_a, page_b in zip(pages, pages_from_pageid):
self.assertIsInstance(page_a, pywikibot.Page)
self.assertIsInstance(page_b, pywikibot.Page)
self.assertTrue(page_a, page_b)
def test_pagegenerator(self):
"""Test page generator."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Main Page')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
def test_random_generator_default(self):
"""Test random generator."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-random:1')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLength(pages, 1)
def test_random_generator_ns(self):
"""Test random generator with namespace."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-random:1')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 1)
def test_random_generator_ns_multi(self):
"""Test random generator with multiple namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-ns:3')
gf.handle_arg('-random:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, {1, 3})
def test_randomredirect_generator_default(self):
"""Test random generator."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-randomredirect:1')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLength(pages, 1)
def test_randomredirect_generator_ns(self):
"""Test random generator with namespace."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-randomredirect:1')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 1)
def test_randomredirect_generator_ns_multi(self):
"""Test random generator with multiple namespaces."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-ns:3')
gf.handle_arg('-randomredirect:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, {1, 3})
def test_pages_with_property_generator(self):
"""Test the pages_with_property_generator method."""
mysite = self.get_site()
for item in ('defaultsort', 'disambiguation', 'displaytitle',
'hiddencat', 'invalid_property'):
if item in mysite.get_property_names():
gf = pagegenerators.GeneratorFactory()
gf.handle_arg(f'-property:{item}')
gf.handle_arg('-limit:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 10)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
if item == 'disambiguation':
self.assertTrue(page.isDisambig())
else:
with self.assertRaises(NotImplementedError):
mysite.pages_with_property(item)
self.fail(
f'NotImplementedError not raised for {item}')
def test_empty_generator(self):
"""Test empty generator."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gen = gf.getCombinedGenerator()
self.assertIsNone(gen)
def test_positionalargument(self):
"""Test page generator with positional argument."""
gf1 = pagegenerators.GeneratorFactory(site=self.site,
positional_arg_name='page')
gf1.handle_arg('Main Page')
gen1 = gf1.getCombinedGenerator()
self.assertIsNotNone(gen1)
gf2 = pagegenerators.GeneratorFactory(site=self.site)
gf2.handle_arg('-page:Main Page')
gen2 = gf2.getCombinedGenerator()
self.assertIsNotNone(gen2)
self.assertEqual(list(gen1), list(gen2))
def test_positionalargument_with_colon(self):
"""Test page generator with positional argument with colon."""
gf1 = pagegenerators.GeneratorFactory(site=self.site,
positional_arg_name='page')
gf1.handle_arg('Project:Main Page')
gen1 = gf1.getCombinedGenerator()
self.assertIsNotNone(gen1)
gf2 = pagegenerators.GeneratorFactory(site=self.site)
gf2.handle_arg('-page:Project:Main Page')
gen2 = gf2.getCombinedGenerator()
self.assertIsNotNone(gen2)
self.assertEqual(list(gen1), list(gen2))
def test_linter_generator_ns_valid_cat(self):
"""Test generator of pages with lint errors."""
if not self.site.has_extension('Linter'):
self.skipTest(
f'The site {self.site} does not use Linter extension')
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-ns:1')
gf.handle_arg('-limit:3')
gf.handle_arg('-linter:obsolete-tag')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 5)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
self.assertEqual(page._lintinfo['category'], 'obsolete-tag')
self.assertPagesInNamespaces(pages, {1})
def test_linter_generator_invalid_cat(self):
"""Test generator of pages with lint errors."""
if not self.site.has_extension('Linter'):
self.skipTest(
f'The site {self.site} does not use Linter extension')
gf = pagegenerators.GeneratorFactory(site=self.site)
with self.assertRaises(AssertionError):
gf.handle_arg('-linter:dummy')
def test_linter_generator_show(self):
"""Test generator of pages with lint errors."""
gf = pagegenerators.GeneratorFactory(site=self.site)
if self.site.has_extension('Linter'):
with self.assertRaises(SystemExit) as cm:
gf.handle_arg('-linter:show')
self.assertEqual(cm.exception.code, 0)
else:
with self.assertRaises(UnknownExtensionError):
gf.handle_arg('-linter:show')
def test_querypage_generator_with_valid_page(self):
"""Test generator of pages with lint errors."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-querypage:Ancientpages')
gf.handle_arg('-limit:5')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 5)
for page in pages:
self.assertIsInstance(page, pywikibot.Page)
def test_querypage_generator_with_invalid_page(self):
"""Test generator of pages with lint errors."""
gf = pagegenerators.GeneratorFactory(site=self.site)
with self.assertRaises(ValueError):
gf.handle_arg('-querypage:dummy')
def test_querypage_generator_with_no_page(self):
"""Test generator of pages with lint errors."""
gf = pagegenerators.GeneratorFactory(site=self.site)
with self.assertRaises(SystemExit) as cm:
gf.handle_arg('-querypage')
self.assertEqual(cm.exception.code, 0)
class TestFactoryGeneratorNewpages(TestCase):
"""Test pagegenerators.GeneratorFactory for newpages."""
# Detached from TestFactoryGenerator due to T159029
sites = {
'eswiki': {
'family': 'wikipedia',
'code': 'es',
},
'commons': {
'family': 'commons',
'code': 'commons',
},
'ruwikt': {
'family': 'wiktionary',
'code': 'ru',
},
'meta': {
'family': 'meta',
'code': 'meta',
},
'frsource': {
'family': 'wikisource',
'code': 'fr',
},
'devoy': {
'family': 'wikivoyage',
'code': 'de',
},
}
def test_newpages_default(self, key):
"""Test newpages generator."""
site = self.get_site(key)
gf = pagegenerators.GeneratorFactory(site=site)
gf.handle_arg('-newpages:60')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLessEqual(len(pages), 60)
newpages_url = self.site.base_url(
self.site.path() + '?title=Special:NewPages&uselang=en')
failure_message = 'No new pages returned by -newpages. If this is ' \
f'the only failure, check whether {newpages_url} contains any ' \
'pages. If not, create a new page on the site to make the test ' \
'pass again.'
self.assertIsNotEmpty(pages, msg=failure_message)
def test_newpages_ns_default(self, key):
"""Test newpages generator with limit argument."""
site = self.get_site(key)
gf = pagegenerators.GeneratorFactory(site=site)
gf.handle_arg('-newpages:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 0)
def test_newpages_ns(self, key):
"""Test newpages generator with limit argument and namespace filter."""
site = self.get_site(key)
gf = pagegenerators.GeneratorFactory(site=site)
gf.handle_arg('-ns:1')
gf.handle_arg('-newpages:10')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertPagesInNamespaces(gen, 1)
class TestWantedFactoryGenerator(DefaultSiteTestCase):
"""Test pagegenerators.GeneratorFactory for wanted pages."""
def setUp(self):
"""Setup tests."""
super().setUp()
self.gf = pagegenerators.GeneratorFactory(site=self.site)
def _generator_with_tests(self):
"""Test generator."""
gen = self.gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = list(gen)
self.assertLessEqual(len(pages), 5)
yield from pages
def test_wanted_pages(self):
"""Test wantedpages generator."""
self.gf.handle_arg('-wantedpages:5')
for page in self._generator_with_tests():
self.assertIsInstance(page, pywikibot.Page)
def test_wanted_files(self):
"""Test wantedfiles generator."""
if self.site.sitename == 'wowwiki:uk':
self.skipTest(f'skipping {self.site} due to T362384)')
self.gf.handle_arg('-wantedfiles:5')
for page in self._generator_with_tests():
self.assertIsInstance(page, pywikibot.Page)
if not isinstance(page, pywikibot.FilePage):
with self.assertRaisesRegex(ValueError,
'does not have a valid extension'):
pywikibot.FilePage(page)
else:
self.assertIsInstance(page.latest_file_info.mime, str)
def test_wanted_templates(self):
"""Test wantedtemplates generator."""
self.gf.handle_arg('-wantedtemplates:5')
for page in self._generator_with_tests():
self.assertIsInstance(page, pywikibot.Page)
self.assertEqual(page.namespace(), 10)
def test_wanted_categories(self):
"""Test wantedcategories generator."""
self.gf.handle_arg('-wantedcategories:5')
for page in self._generator_with_tests():
self.assertIsInstance(page, pywikibot.Category)
class TestFactoryGeneratorWikibase(WikidataTestCase):
"""Test pagegenerators.GeneratorFactory on Wikibase site."""
def test_onlyif(self):
"""Test -onlyif without qualifiers."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Q15745378')
self.assertTrue(gf.handle_arg(
'-onlyif:P1476=International Journal of Minerals\\, '
'Metallurgy\\, and Materials'))
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertLength(set(gen), 1)
def test_onlyifnot(self):
"""Test -onlyifnot without qualifiers."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Q15745378')
gf.handle_arg('-onlyifnot:P1476=International Journal of Minerals\\, '
'Metallurgy\\, and Materials')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertIsEmpty(set(gen))
def test_onlyif_qualifiers(self):
"""Test -onlyif with qualifiers."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-page:Q668')
gf.handle_arg('-onlyif:P47=Q837,P805=Q3088768')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertLength(set(gen), 1)
def test_searchitem(self):
"""Test -searchitem."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-searchitem:abc')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
self.assertIsNotNone(next(gen))
def test_searchitem_language(self):
"""Test -searchitem with custom language specified."""
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg('-searchitem:en:abc')
gf.handle_arg('-limit:5')
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
result = {page.title() for page in gen}
self.assertIn('Q169889', result) # American Broadcasting Company
self.assertIn('Q9779', result) # alphabet, also known as ABC
self.assertIn('Q1057802', result) # ABC programming language
def test_get_category_site(self):
"""Test the getCategory method."""
# With default site
gf = pagegenerators.GeneratorFactory()
cat = gf.getCategory('foo')[0]
self.assertEqual(cat.site, pywikibot.Site())
# With a user-specified site
fa_wikisource = pywikibot.Site('fa', 'wikisource')
gf = pagegenerators.GeneratorFactory(fa_wikisource)
cat = gf.getCategory('foo')[0]
self.assertEqual(cat.site, fa_wikisource)
class TestLogeventsFactoryGenerator(DefaultSiteTestCase,
DeprecationTestCase):
"""Test GeneratorFactory with pagegenerators.LogeventsPageGenerator."""
@classmethod
def setUpClass(cls):
"""Setup test class."""
super().setUpClass()
site = pywikibot.Site()
newuser_logevents = list(site.logevents(logtype='newusers', total=1))
if not newuser_logevents:
raise unittest.SkipTest('No newuser logs found to test with.')
login = True
def test_logevents_parse(self):
"""Test wrong logevents option."""
factory = pagegenerators.GeneratorFactory
gf = factory()
self.assertFalse(gf.handle_arg('-log'))
self.assertFalse(gf.handle_arg('-log:text_here'))
with self.assertRaises(NotImplementedError):
gf.handle_arg('-logevents:anyevent')
# test that old format log option is not handled by any handler method.
gf_mock = mock.create_autospec(gf)
self.assertFalse(factory.handle_arg(gf_mock, '-anotherlog'))
self.assertFalse(gf_mock.method_calls)
def test_logevents_with_start_timestamp(self):
"""Test -logevents which uses timestamp for start."""
gf = pagegenerators.GeneratorFactory(site=self.site)
# We limit the results to 1 as running this on large websites like
# Wikipedia will give an insane number of results as it asks for all
# logevents since beginning till now.
self.assertTrue(gf.handle_arg('-limit:1'))
self.assertTrue(gf.handle_arg('-logevents:newusers,,21000101'))
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertIsNotEmpty(pages)
for item in pages:
self.assertIsInstance(item, pywikibot.User)
def test_logevents_with_start_and_end_timestamp(self):
"""Test -logevents which uses timestamps for start and end."""
gf = pagegenerators.GeneratorFactory(site=self.site)
self.assertTrue(
gf.handle_arg('-logevents:newusers,,21000101,20990101'))
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertIsEmpty(pages)
def test_logevents_with_total(self):
"""Test -logevents which uses total."""
gf = pagegenerators.GeneratorFactory(site=self.site)
self.assertTrue(gf.handle_arg('-logevents:newusers,,1'))
gen = gf.getCombinedGenerator()
self.assertIsNotNone(gen)
pages = set(gen)
self.assertLength(pages, 1)
for item in pages:
self.assertIsInstance(item, pywikibot.User)
class PageGeneratorIntersectTestCase(GeneratorIntersectTestCase,
RecentChangesTestCase):
"""Page intersect_generators test cases."""
def test_intersect_newpages_twice(self):
"""Test newpages intersection."""
site = self.get_site()
self.assertEqualItertools(
[pagegenerators.NewpagesPageGenerator(site=site, total=10),
pagegenerators.NewpagesPageGenerator(site=site, total=10)])
def test_intersect_newpages_and_recentchanges(self):
"""Test intersection between newpages and recentchanges."""
site = self.get_site()
self.assertEqualItertools(
[pagegenerators.NewpagesPageGenerator(site=site, total=50),
pagegenerators.RecentChangesPageGenerator(site=site, total=200)])
class EnWikipediaPageGeneratorIntersectTestCase(GeneratorIntersectTestCase,
RecentChangesTestCase):
"""Page intersect_generators test cases."""
family = 'wikipedia'
code = 'en'
def test_intersect_newpages_csd(self):
"""Test intersection between newpages and sd candidates."""
site = self.get_site()
self.assertEqualItertools([
pagegenerators.NewpagesPageGenerator(site=site, total=10),
pagegenerators.CategorizedPageGenerator(
pywikibot.Category(
site, 'Category:Candidates_for_speedy_deletion'))]
)
class EventStreamsPageGeneratorTestCase(RecentChangesTestCase):
"""Test case for Live Recent Changes pagegenerator."""
@classmethod
def setUpClass(cls):
"""Setup test class."""
super().setUpClass()
cls.client = 'sseclient'
if not has_module(cls.client):
raise unittest.SkipTest(f'{cls.client} is not available')
def test_RC_pagegenerator_result(self):
"""Test RC pagegenerator."""
lgr = logging.getLogger(self.client)
lgr.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
lgr.addHandler(ch)
site = self.get_site()
pagegenerator = pagegenerators.LiveRCPageGenerator(site,
total=self.length)
entries = list(pagegenerator)
self.assertLength(entries, self.length)
testentry = entries[0]
self.assertEqual(testentry.site, site)
self.assertTrue(hasattr(testentry, '_rcinfo'))
rcinfo = testentry._rcinfo
self.assertEqual(rcinfo['server_name'], site.hostname())
self.assertEqual(rcinfo['wiki'], site.dbName())
for key in ['type', 'namespace', 'title', 'comment',
'timestamp', 'user', 'bot']:
self.assertIn(key, rcinfo.keys())
class TestUnconnectedPageGenerator(DefaultSiteTestCase):
"""Test UnconnectedPageGenerator."""
def test_unconnected_with_repo(self):
"""Test UnconnectedPageGenerator."""
site = self.site.data_repository()
if not site:
self.skipTest('Site is not using a Wikibase repository')
pages = list(pagegenerators.UnconnectedPageGenerator(self.site, 3))
self.assertLessEqual(len(pages), 3)
pattern = (fr'Page \[\[({site.sitename}:|{site.code}:)-1\]\]'
r" doesn't exist\.")
found = []
for page in pages:
with self.subTest(page=page):
try:
page.data_item()
except NoPageError as e:
self.assertRegex(str(e), pattern)
else:
found.append(page)
if found:
unittest_print('connection found for ',
', '.join(str(p) for p in found))
# assume that we have at least one unconnected page
self.assertLess(len(found), 3)
def test_unconnected_without_repo(self):
"""Test that it raises a ValueError on sites without repository."""
if self.site.data_repository():
self.skipTest('Site is using a Wikibase repository')
with self.assertRaises(ValueError):
for _ in pagegenerators.UnconnectedPageGenerator(self.site,
total=5):
raise AssertionError(
"this shouldn't be reached") # pragma: no cover
class TestLinksearchPageGenerator(TestCase):
"""Tests for pagegenerators.LinksearchPageGenerator."""
family = 'wikipedia'
code = 'en'
@unittest.skip('Needs to be adapted for T14810')
def test_weblink(self):
"""Test -weblink."""
cases = (('wikipedia.org', 'http://wikipedia.org'),
('en.wikipedia.org', 'http://en.wikipedia.org'),
('https://fr.wikipedia.org', 'https://fr.wikipedia.org'),
('ftp://*', 'ftp://'))
for search, expected in cases:
with self.subTest(search=search, expected=expected):
gf = pagegenerators.GeneratorFactory(site=self.site)
gf.handle_arg(f'-weblink:{search}')
gf.handle_arg('-ns:2')
gf.handle_arg('-limit:1')
gen = gf.getCombinedGenerator()
genlist = list(gen)
self.assertLength(genlist, 1)
page = genlist[0]
self.assertIsInstance(page, pywikibot.Page)
self.assertTrue(page.exists())
self.assertEqual(page.namespace(), 2)
self.assertIn(expected, page.text)
def test_double_opposite_protocols(self):
"""Test LinksearchPageGenerator with two opposite protocols."""
with self.assertRaises(ValueError):
pagegenerators.LinksearchPageGenerator('http://w.wiki',
protocol='https',
site=self.site)
@unittest.skip('Needs to be adapted for T14810')
def test_double_same_protocols(self):
"""Test LinksearchPageGenerator with two same protocols."""
gen = pagegenerators.LinksearchPageGenerator('https://w.wiki',
protocol='https',
site=self.site,
total=1)
self.assertIsInstance(gen, pywikibot.data.api.PageGenerator)
self.assertLength(list(gen), 1)
if __name__ == '__main__':
with suppress(SystemExit):
unittest.main()