Source code for tests.noreferences_tests
#!/usr/bin/env python3
"""Test noreferences bot module."""
#
# (C) Pywikibot team, 2018-2022
#
# Distributed under the terms of the MIT license.
#
from __future__ import annotations
import pywikibot
from scripts.noreferences import NoReferencesBot
from tests.aspects import TestCase, unittest
[docs]
class TestAddingReferences(TestCase):
"""Test adding references to section."""
family = 'wikipedia'
code = 'cs'
[docs]
def test_add(self):
"""Test adding references section."""
page = pywikibot.Page(self.site, 'foo')
bot = NoReferencesBot()
bot.site = self.site
page.text = '\n== Reference ==\n* [http://www.baz.org Baz]'
new_text = bot.addReferences(page.text)
expected = ('\n== Reference ==\n<references />'
'\n\n* [http://www.baz.org Baz]')
self.assertEqual(new_text, expected)
[docs]
def test_add_under_templates(self):
"""Test adding references section under templates in section."""
page = pywikibot.Page(self.site, 'foo')
bot = NoReferencesBot()
bot.site = self.site
page.text = '\n== Reference ==\n{{Překlad|en|Baz|123456}}'
new_text = bot.addReferences(page.text)
expected = page.text + '\n<references />\n'
self.assertEqual(new_text, expected)
if __name__ == '__main__':
unittest.main()