Revision: 5861 Author: russblau Date: 2008-08-28 19:37:33 +0000 (Thu, 28 Aug 2008) Log Message: ----------- many new site method tests Modified Paths: -------------- branches/rewrite/pywikibot/tests/site_tests.py Modified: branches/rewrite/pywikibot/tests/site_tests.py =================================================================== --- branches/rewrite/pywikibot/tests/site_tests.py 2008-08-28 19:11:10 UTC (rev 5860) +++ branches/rewrite/pywikibot/tests/site_tests.py 2008-08-28 19:37:33 UTC (rev 5861) @@ -13,22 +13,149 @@ import unittest import pywikibot -mysite = pywikibot.Site("en", "wikipedia") +mysite = pywikibot.Site() +mainpage = pywikibot.Page(pywikibot.Link("Main Page", mysite)) class TestSiteObject(unittest.TestCase): """Test cases for Site methods.""" def testBaseMethods(self): """Test cases for BaseSite methods""" - self.assertEqual(mysite.family, pywikibot.site.Family("wikipedia")) - self.assertEqual(mysite.language(), "en") - self.assert_(isinstance(mysite.user(), (basestring, type(None)))) - self.assertEqual(mysite.sitename(), "wikipedia:en") - self.assertEqual(mysite.ns_normalize("project"), "Wikipedia") - self.assertEqual(mysite.redirect(), "REDIRECT") + self.assertEqual(mysite.family.name, pywikibot.config.family) + self.assertEqual(mysite.code, pywikibot.config.mylang) + self.assertTrue(isinstance(mysite.user(), (basestring, type(None)))) + self.assertEqual(mysite.sitename(), + "%s:%s" % (pywikibot.config.family, pywikibot.config.mylang)) + self.assertTrue(isinstance(mysite.ns_normalize("project"), basestring)) + self.assertTrue(isinstance(mysite.redirect(), basestring)) + def testApiMethods(self): + """Test generic ApiSite methods""" + + self.assertTrue(isinstance(mysite.logged_in(), bool)) + self.assertTrue(isinstance(mysite.getuserinfo(), dict)) + self.assertTrue(isinstance(mysite.getcurrenttimestamp(), basestring)) + self.assertTrue(isinstance(mysite.siteinfo, dict)) + self.assertTrue(isinstance(mysite.case(), basestring)) + self.assertTrue(isinstance(mysite.namespaces(), dict)) + self.assertTrue(all(isinstance(key, int) + for key in mysite.namespaces())) + self.assertTrue(all(isinstance(val, list) + for val in mysite.namespaces().values())) + self.assertTrue(all(isinstance(name, basestring) + for val in mysite.namespaces().values() + for name in val)) + self.assertTrue(all(isinstance(mysite.namespace(key), basestring) + for key in mysite.namespaces())) + self.assertTrue(all(isinstance(mysite.namespace(key, True), list) + for key in mysite.namespaces())) + self.assertTrue(all(isinstance(item, basestring) + for key in mysite.namespaces() + for item in mysite.namespace(key, True))) + def testPageMethods(self): + """Test ApiSite methods for getting page-specific info""" + + self.assertTrue(isinstance(mysite.page_exists(mainpage), bool)) + self.assertTrue(isinstance(mysite.page_restrictions(mainpage), dict)) + self.assertTrue(isinstance(mysite.page_can_be_edited(mainpage), bool)) + self.assertTrue(isinstance(mysite.page_isredirect(mainpage), bool)) + if mysite.page_isredirect(mainpage): + self.assertTrue(isinstance(mysite.getredirtarget(mainpage), + pywikibot.Page)) + else: + self.assertRaises(pywikibot.IsNotRedirectPage, + mysite.getredirtarget, mainpage) + a = list(mysite.preloadpages([mainpage])) + self.assertEqual(len(a), int(mysite.page_exists(mainpage))) + if a: + self.assertEqual(a[0], mainpage) + + def testTokens(self): + """Test ability to get page tokens""" + + for ttype in ("edit", "move"): # token types for non-sysops + self.assertTrue(isinstance(mysite.token(mainpage, ttype), + basestring)) + + def testLinkMethods(self): + """Test site methods for getting links to and from a page""" + + backlinks = set(mysite.pagebacklinks(mainpage)) + embedded = set(mysite.page_embeddedin(mainpage)) + refs = set(mysite.pagereferences(mainpage)) + for bl in backlinks: + self.assertTrue(isinstance(bl, pywikibot.Page)) + self.assertTrue(bl in refs) + for ei in embedded: + self.assertTrue(isinstance(ei, pywikibot.Page)) + self.assertTrue(ei in refs) + for ref in refs: + self.assertTrue(ref in backlinks or ref in embedded) + for pl in mysite.pagelinks(mainpage): + self.assertTrue(isinstance(pl, pywikibot.Page)) + for cat in mysite.pagecategories(mainpage): + self.assertTrue(isinstance(cat, pywikibot.Category)) + for cm in mysite.categorymembers(cat): + self.assertTrue(isinstance(cat, pywikibot.Page)) + self.assertTrue(all(isinstance(im, pywikibot.ImagePage) + for im in mysite.pageimages(mainpage))) + self.assertTrue(all(isinstance(te, pywikibot.Page) + for te in mysite.pagetemplates(mainpage))) + for ll in mysite.pagelanglinks(mainpage): + self.assertTrue(isinstance(ll, pywikibot.Link)) + self.assertTrue(all(isinstance(el, basestring) + for el in mysite.page_extlinks(mainpage))) + + def testLoadRevisions(self): + """Test the site.loadrevisions() method""" + + mysite.loadrevisions(mainpage) + self.assertTrue(hasattr(mainpage, "_revid")) + self.assertTrue(hasattr(mainpage, "_revisions")) + self.assertTrue(mainpage._revisions.has_key(mainpage._revid)) + + def testAllPages(self): + """Test the site.allpages() method""" + + ap = list(mysite.allpages(limit=10)) + self.assertTrue(len(ap) <= 10) + for page in ap: + self.assertTrue(isinstance(page, pywikibot.Page)) + self.assertTrue(mysite.page_exists(page)) + # TODO: test various optional arguments to allpages + + def testAllLinks(self): + """Test the site.alllinks() method""" + + al = list(mysite.alllinks(limit=10)) + self.assertTrue(len(al) <= 10) + self.assertTrue(all(isinstance(link, pywikibot.Page) for link in al)) + # TODO: test various optional arguments to alllinks + + def testAllCategories(self): + """Test the site.allcategories() method""" + + ac = list(mysite.allcategories(limit=10)) + self.assertTrue(len(ac) <= 10) + self.assertTrue(all(isinstance(cat, pywikibot.Category) + for cat in ac)) + # TODO: test various optional arguments to allcategories + + def testAllUsers(self): + """Test the site.allusers() method""" + + au = list(mysite.allusers(limit=10)) + self.assertTrue(len(au) <= 10) + for user in au: + self.assertTrue(isinstance(user, dict)) + self.assertTrue(user.has_key("name")) + self.assertTrue(user.has_key("editcount")) + self.assertTrue(user.has_key("registration")) + + if __name__ == '__main__': +# pywikibot.logging.getLogger().setLevel(pywikibot.logging.DEBUG) try: try: unittest.main()