diff --git a/tests/interwikidata_tests.py b/tests/interwikidata_tests.py index 4780a65fdf..51b0f1cd91 100755 --- a/tests/interwikidata_tests.py +++ b/tests/interwikidata_tests.py @@ -13,7 +13,7 @@ import pywikibot from pywikibot import Link from scripts import interwikidata -from tests.aspects import SiteAttributeTestCase +from tests.aspects import AlteredDefaultSiteTestCase, SiteAttributeTestCase from tests.utils import empty_sites @@ -34,7 +34,7 @@ def try_to_add(self): return -class TestInterwikidataBot(SiteAttributeTestCase): +class TestInterwikidataBot(AlteredDefaultSiteTestCase, SiteAttributeTestCase): """Test Interwikidata.""" @@ -53,10 +53,13 @@ class TestInterwikidataBot(SiteAttributeTestCase): }, } - def test_main(self): + def test_main(self, key): """Test main function interwikidata.py.""" - # The default site is used here - if pywikibot.Site().has_data_repository: + site = self.get_site(key) + pywikibot.config.family = site.family + pywikibot.config.mylang = site.code + + if site.has_data_repository: with empty_sites(): # The main function return None. self.assertIsNone(interwikidata.main())