def test_start_prefix(self):
"""Test with a prefix as a start page title."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake',
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPageTitlesEqual(pages, ('Fake page', 'Talk:Fake page'),
site=self.get_site())
def test_simple_bare_refs(self):
"""Test simple bare references in multiple namespaces."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake page',
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPageTitlesEqual(pages, ('Fake page', 'Talk:Fake page'),
site=self.get_site())
def test_namespace_names(self):
"""Test namespaces with namespace names."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake page',
namespaces=['Talk'],
site=self.get_site())
pages = list(gen)
self.assertPageTitlesEqual(pages, ('Talk:Fake page', ),
site=self.get_site())
def test_namespace_names(self):
"""Test namespaces with namespace names."""
gen = XmlDumpPageGenerator(xmlFilename=os.path.join(
_xml_data_dir, 'dummy-reflinks.xml'),
xmlStart=u'Fake page',
namespaces=["Talk"],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Talk:Fake page', ),
site=self.get_site())
def test_start_with_underscore(self):
"""Test with underscore in start page title."""
gen = XmlDumpPageGenerator(xmlFilename=os.path.join(
_xml_data_dir, 'dummy-reflinks.xml'),
xmlStart=u'Fake_page',
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Fake page', u'Talk:Fake page'),
site=self.get_site())
def test_simple_bare_refs(self):
"""Test simple bare references in multiple namespaces."""
gen = XmlDumpPageGenerator(xmlFilename=os.path.join(
_xml_data_dir, 'dummy-reflinks.xml'),
xmlStart=u'Fake page',
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Fake page', u'Talk:Fake page'),
site=self.get_site())
def test_without_start(self):
"""Test without a start page title."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start=None,
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Fake page', u'Talk:Fake page'),
site=self.get_site())
def test_namespace_string_ids(self):
"""Test namespaces with ids as string."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake page',
namespaces=["0", "1"],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Fake page', u'Talk:Fake page'),
site=self.get_site())
def test_namespace_None(self):
"""Test namespaces=None processes all namespaces."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake page',
namespaces=None,
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, (u'Fake page', u'Talk:Fake page'),
site=self.get_site())
def test_start_with_underscore(self):
"""Test with underscore in start page title."""
gen = XmlDumpPageGenerator(
filename=join_xml_data_path('dummy-reflinks.xml'),
start='Fake_page',
namespaces=[0, 1],
site=self.get_site())
pages = list(gen)
self.assertPagelistTitles(pages, ('Fake page', 'Talk:Fake page'),
site=self.get_site())
def test_simple_bare_refs(self):
"""Test simple bare references with several namespaces options."""
namespace_variants = (None, [], [0, 1], ['0', '1'])
filename = join_xml_data_path('dummy-reflinks.xml')
for namespaces in namespace_variants:
with self.subTest(namespaces=namespaces):
gen = XmlDumpPageGenerator(filename=filename,
start='Fake page',
namespaces=namespaces,
site=self.site)
pages = list(gen)
self.assertPageTitlesEqual(pages,
('Fake page', 'Talk:Fake page'),
site=self.site)
def test_start_variants(self):
"""Test with several page title options."""
start_variants = (
None, # None
'Fake', # prefix
'Fake_page', # underscore
)
filename = join_xml_data_path('dummy-reflinks.xml')
for start in start_variants:
with self.subTest(start=start):
gen = XmlDumpPageGenerator(filename=filename,
start=start,
namespaces=[0, 1],
site=self.site)
pages = list(gen)
self.assertPageTitlesEqual(pages,
('Fake page', 'Talk:Fake page'),
site=self.site)
