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)