Changeset 14670 for main/waeup.kofa/trunk/src/waeup/kofa/tests

Timestamp:

6 Apr 2017, 12:20:41 (8 years ago)

Author:

uli

Message:

Add IKofaPluggable to update local PAU.

The new plugin enables updating of sites that have
yet no XMLRPC authentication enabled.

The plugin can be removed after updating all sites.

New sites (university-instances) do not need this
plugin at all.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/tests/test_authentication.py (modified) (5 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/tests/test_authentication.py

@@ -31 +31 @@
 from zope.securitypolicy.interfaces import IPrincipalRoleManager
 from waeup.kofa.testing import FunctionalTestCase, FunctionalLayer
+from waeup.kofa.app import University
 from waeup.kofa.authentication import (
     UserAuthenticatorPlugin, Account, KofaPrincipalInfo, FailedLoginInfo,
     get_principal_role_manager, UsersPlugin, KofaXMLRPCCredentialsPlugin,
-    setup_authentication)
+    setup_authentication, UpdatePAUPlugin)
 from waeup.kofa.interfaces import (
     IAuthPluginUtility, IUserAccount, IFailedLoginInfo, IKofaPrincipalInfo,
@@ -368 +369 @@
     pass
 
+
+def get_logger():
+    logger = logging.getLogger('waeup.test')
+    stream = StringIO()
+    handler = logging.StreamHandler(stream)
+    logger.setLevel(logging.DEBUG)
+    logger.propagate = False
+    logger.addHandler(handler)
+    return logger, stream
+
+
 class UsersPluginTests(unittest.TestCase):
 
@@ -375 +387 @@
         self.site['users'] = grok.Container()
         return
-
-    def get_logger(self):
-        logger = logging.getLogger('waeup.test')
-        stream = StringIO()
-        handler = logging.StreamHandler(stream)
-        logger.setLevel(logging.DEBUG)
-        logger.propagate = False
-        logger.addHandler(handler)
-        return logger, stream
 
     def test_ifaces(self):
@@ -395 +398 @@
         # make sure user accounts are updated properly.
         plugin = UsersPlugin()
-        logger, stream = self.get_logger()
+        logger, stream = get_logger()
         plugin.update(self.site, 'app', logger)
         stream.seek(0)
         self.assertEqual(stream.read(), '')
         self.site['users']['bob'] = FakeUserAccount()
-        logger, stream = self.get_logger()
+        logger, stream = get_logger()
         plugin.update(self.site, 'app', logger)
         stream.seek(0)
@@ -412 +415 @@
         self.assertTrue('attribute failed_logins added' in log_content)
         return
+
+
+class TestUpdatePAUPlugin(FunctionalTestCase):
+
+    layer = FunctionalLayer
+
+    def setUp(self):
+        super(TestUpdatePAUPlugin, self).setUp()
+        self.getRootFolder()['app'] = University()
+        self.site = self.getRootFolder()['app']
+
+    def tearDown(self):
+        clearSite()
+        super(TestUpdatePAUPlugin, self).tearDown()
+
+    def get_pau(self):
+        # the PAU is registered as a local utility in local site manager.
+        # the name is derived from class name.
+        pau = self.site.getSiteManager()['PluggableAuthentication']
+        assert pau is not None
+        return pau
+
+    def test_update_outdated(self):
+        # we can update outdated sites.
+        plugin = UpdatePAUPlugin()
+        logger, stream = get_logger()
+        pau = self.get_pau()
+        pau.credentialsPlugins = ('foo', 'credentials', 'bar')
+        plugin.update(self.site, 'xmlrpc-credentials', logger)
+        assert 'xmlrpc-credentials' in pau.credentialsPlugins
+        assert pau.credentialsPlugins.index('xmlrpc-credentials') == 1
+
+    def test_update_uptodate(self):
+        # we cope with already updated sites.
+        plugin = UpdatePAUPlugin()
+        logger, stream = get_logger()
+        pau = self.get_pau()
+        pau.credentialsPlugins = ('foo', 'xmlrpc-credentials', 'bar')
+        plugin.update(self.site, 'xmlrpc-credentials', logger)
+        assert pau.credentialsPlugins.count('xmlrpc-credentials') == 1