source: products/quintagroup.plonegooglesitemaps/branches/test_refactoring/quintagroup/plonegooglesitemaps/tests/testConfiglet.py @ 2531

#
# Tests for quintagroup.plonegooglesitemaps
#

from base import *
from cgi import FieldStorage
from tempfile import TemporaryFile, NamedTemporaryFile

from OFS.Image import cookId
from Products.CMFPlone.utils import _createObjectByType
from ZPublisher.HTTPRequest import FileUpload


class TestConfigletSettings(FunctionalTestCase):

    def afterSetUp(self):
        super(TestConfigletSettings, self).afterSetUp()
        self.auth = "%s:%s" % (portal_owner, default_password)
        self.settingsURL = '/'+self.portal.absolute_url(1) + '/prefs_gsm_settings'

    def submitForm(self, fdata, fextra={}):
        form = {'form.submitted': 1}
        form.update(fdata)
        return self.publish(self.settingsURL, request_method='POST',
                            stdin=StringIO(urlencode(form)),
                            basic=self.auth, extra=fextra)

    def testInitButtons(self):
        settings = self.publish(self.settingsURL, self.auth).getBody()
        self.assert_("Add Content Sitemap" in settings)
        self.assert_("Add News Sitemap" in settings)
        self.assert_("Add Mobile Sitemap" in settings)

    def testAddContentSitemap(self):
        resp = self.submitForm({'form.button.AddContent': "Add Content Sitemap"})
        self.assertEqual(resp.getStatus() / 100, 3)
        self.assertEqual(resp.getHeader("Location").endswith("sitemap.xml/edit"), True)
        # Add SM
        sm = self.portal["sitemap.xml"]
        #sm.setPortalTypes(("Documents",))
        newform = self.publish(self.settingsURL, basic=self.auth).getBody()
        self.assertEqual('href="http://nohost/plone/sitemap.xml/edit"' in newform, True)
        self.assertEqual("form.button.AddContent" in newform, False)

    def testAddNewsSitemap(self):
        resp = self.submitForm({'form.button.AddNews': "Add News Sitemap"})
        self.assertEqual(resp.getStatus() / 100, 3)
        self.assertEqual(resp.getHeader("Location").endswith("news-sitemap.xml/edit"), True)
        # Add SM
        sm = self.portal["news-sitemap.xml"]
        #sm.setPortalTypes(("News Item",))
        newform = self.publish(self.settingsURL, basic=self.auth).getBody()
        self.assertEqual('href="http://nohost/plone/news-sitemap.xml/edit"' in newform, True)
        self.assertEqual("form.button.AddNews" in newform, False)

    def testAddMobileSitemap(self):
        resp = self.submitForm({'form.button.AddMobile': "Add Mobile Sitemap"})
        self.assertEqual(resp.getStatus() / 100, 3)
        self.assertEqual(resp.getHeader("Location").endswith("mobile-sitemap.xml/edit"), True)
        # Add SM
        sm = self.portal["mobile-sitemap.xml"]
        #sm.setPortalTypes(("Documents",))
        newform = self.publish(self.settingsURL, basic=self.auth).getBody()
        self.assertEqual('href="http://nohost/plone/mobile-sitemap.xml/edit"' in newform, True)
        self.assertEqual("form.button.AddMobile" in newform, False)


class TestConfigletOverview(FunctionalTestCase):

    def afterSetUp(self):
        super(TestConfigletOverview, self).afterSetUp()
        self.auth = "%s:%s" % (portal_owner, default_password)
        self.overviewURL = '/'+self.portal.absolute_url(1) + '/prefs_gsm_overview'
        self.smURL = "%s/sitemap.xml" % self.portal.absolute_url()
        
    def testInitial(self):
        overview = self.publish(self.overviewURL, self.auth).getBody()
        self.assert_(not self.smURL in overview)

    def testPresentedSM(self):
        _createObjectByType("Sitemap", self.portal, id="sitemap.xml",
                            sitemapType="content")
        overview = self.publish(self.overviewURL, self.auth).getBody()
        self.assert_(self.smURL in overview)


def test_suite():
    from unittest import TestSuite, makeSuite
    suite = TestSuite()
    suite.addTest(makeSuite(TestConfigletSettings))
    suite.addTest(makeSuite(TestConfigletOverview))
    return suite

if __name__ == '__main__':
    unittest.main(defaultTest='test_suite')
#    framework()