Allow passing SSO public key as parameter (for testing)

ai_web_common/flask_sso/ext.py

@@ -22,6 +22,11 @@ def init_sso(app, talisman):
     if 'SSO_DOMAIN' not in app.config:
         raise Exception('Must configure SSO_DOMAIN')
 
+    # Allow passing the public key as a parameter (useful for testing)
+    # or read it from a file.
+    if 'SSO_PUBLIC_KEY' in app.config:
+        pubkey = app.config['SSO_PUBLIC_KEY']
+    else:
         pubkey_file = app.config.get(
             'SSO_PUBLIC_KEY_FILE', '/etc/sso/public.key')
         with open(pubkey_file, 'rb') as f: