def test_add_under_templates(self):
"""Test adding references section under templates in section."""
page = pywikibot.Page(self.site, 'foo')
bot = NoReferencesBot(None)
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)
def test_add(self):
"""Test adding references section."""
page = pywikibot.Page(self.site, 'foo')
bot = NoReferencesBot(None)
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)