[3296] | 1 | import unittest |
---|
| 2 | |
---|
| 3 | from base import CacheFuTestCase |
---|
| 4 | |
---|
| 5 | from Products.CMFCore.utils import getToolByName |
---|
| 6 | import Products.CacheSetup.config as config |
---|
| 7 | |
---|
| 8 | # util for making content in a container |
---|
| 9 | def makeContent(container, id, portal_type, title=None): |
---|
| 10 | container.invokeFactory(id=id, type_name=portal_type) |
---|
| 11 | o = getattr(container, id) |
---|
| 12 | if title is not None: |
---|
| 13 | o.setTitle(title) |
---|
| 14 | return o |
---|
| 15 | |
---|
| 16 | |
---|
| 17 | # This is the test case. You will have to add test_<methods> to your |
---|
| 18 | # class inorder to assert things about your Product. |
---|
| 19 | class CacheManagerTest(CacheFuTestCase): |
---|
| 20 | USER1 = 'user1' |
---|
| 21 | |
---|
| 22 | def afterSetUp(self): |
---|
| 23 | CacheFuTestCase.afterSetUp(self) |
---|
| 24 | |
---|
| 25 | # Add a couple of users |
---|
| 26 | self.portal.acl_users._doAddUser('manager', 'secret', ['Manager'], []) |
---|
| 27 | self.portal.acl_users._doAddUser(self.USER1, 'secret', ['Member'], []) |
---|
| 28 | self.login('manager') |
---|
| 29 | |
---|
| 30 | self.portal.portal_quickinstaller.installProducts(['CacheSetup']) |
---|
| 31 | |
---|
| 32 | # We have added a skin so we need to rebuild the skin object |
---|
| 33 | # (since the object is cached in the current request) |
---|
| 34 | self._refreshSkinData() |
---|
| 35 | |
---|
| 36 | pcs = self.portal.portal_cache_settings |
---|
| 37 | pcs.setEnabled(True) |
---|
| 38 | |
---|
| 39 | headers = pcs.getHeaderSets() |
---|
| 40 | headers.manage_delObjects(headers.objectIds()) |
---|
| 41 | headers.invokeFactory(id='hs1', type_name='HeaderSet') |
---|
| 42 | headers.invokeFactory(id='hs2', type_name='HeaderSet') |
---|
| 43 | rules = pcs.getRules() |
---|
| 44 | rules.manage_delObjects(rules.objectIds()) |
---|
| 45 | |
---|
| 46 | def test_phcm_do_cache(self): |
---|
| 47 | pcs = self.portal.portal_cache_settings |
---|
| 48 | rules = pcs.getRules() |
---|
| 49 | rules.invokeFactory(id='my_rule', type_name='PolicyHTTPCacheManagerCacheRule') |
---|
| 50 | rule = getattr(pcs.getRules(), 'my_rule') |
---|
| 51 | |
---|
| 52 | rule.setTitle('PHCM') |
---|
| 53 | self.assertEqual(rule.Title(), 'PHCM') |
---|
| 54 | rule.setCacheManager('HTTPCache') |
---|
| 55 | self.assertEqual(rule.getCacheManager(), 'HTTPCache') |
---|
| 56 | rule.setTypes(['Filesystem Image']) |
---|
| 57 | self.assertEqual(rule.getTypes(), ('Filesystem Image',)) |
---|
| 58 | rule.setIds([]) |
---|
| 59 | self.assertEqual(rule.getIds(), ()) |
---|
| 60 | rule.setHeaderSetIdAnon('hs1') |
---|
| 61 | self.assertEqual(rule.getHeaderSetIdAnon(), 'hs1') |
---|
| 62 | rule.setHeaderSetIdAuth('hs2') |
---|
| 63 | self.assertEqual(rule.getHeaderSetIdAuth(), 'hs2') |
---|
| 64 | rule.setCacheStop(['portal_status_message']) |
---|
| 65 | self.assertEqual(rule.getCacheStop(),('portal_status_message',)) |
---|
| 66 | |
---|
| 67 | rule.reindexObject() |
---|
| 68 | |
---|
| 69 | member = self.portal.portal_membership.getAuthenticatedMember() |
---|
| 70 | image = getattr(self.portal, 'addFavorite.gif') # is associated with HTTPCache in Plone 2.1.2 |
---|
| 71 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, member).getId(), 'hs2') |
---|
| 72 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, None).getId(), 'hs1') |
---|
| 73 | nocache_image = getattr(self.portal, 'action_icon.gif') # is not associated with HTTPCache as of Plone 2.1.2 |
---|
| 74 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, nocache_image, None, member), None) |
---|
| 75 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, nocache_image, None, None), None) |
---|
| 76 | |
---|
| 77 | rule.setTypes([]) |
---|
| 78 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, member).getId(), 'hs2') |
---|
| 79 | rule.setTypes(['File']) |
---|
| 80 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, member), None) |
---|
| 81 | rule.setTypes(['Filesystem Image']) |
---|
| 82 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, member).getId(), 'hs2') |
---|
| 83 | |
---|
| 84 | # make sure stop words stop |
---|
| 85 | self.assertEqual(rule.getHeaderSet({}, image, None, member).getId(), 'hs2') |
---|
| 86 | self.assertEqual(rule.getHeaderSet({'portal_status_message':'foo'}, image, None, member), None) |
---|
| 87 | |
---|
| 88 | # make sure we can check getHeaderSetId by script |
---|
| 89 | self.portal.manage_addProduct['PythonScripts'].manage_addPythonScript('test_script') |
---|
| 90 | self.portal.test_script.ZPythonScript_edit('view', 'return \'hs1\'') |
---|
| 91 | rule.setHeaderSetIdAuth('expression') |
---|
| 92 | rule.setHeaderSetIdExpression('python:object.test_script(view)') |
---|
| 93 | self.assertEqual(rule.getHeaderSet(self.portal.REQUEST, image, None, member).getId(), 'hs1') |
---|
| 94 | |
---|
| 95 | def test_suite(): |
---|
| 96 | suite = unittest.TestSuite() |
---|
| 97 | suite.addTest(unittest.makeSuite(CacheManagerTest)) |
---|
| 98 | return suite |
---|