Added support for OAuth 2 authentication. Fixes #5940

Initial patch sent by: Florian Sabonchi
This commit is contained in:
Khushboo Vashi
2021-07-06 13:22:58 +05:30
committed by Akshay Joshi
parent fff4060b31
commit 48ca83f31d
35 changed files with 750 additions and 227 deletions

View File

@@ -13,6 +13,7 @@ from regression.python_test_utils import test_utils as utils
from regression.test_setup import config_data
from pgadmin.authenticate.registry import AuthSourceRegistry
from unittest.mock import patch
from pgadmin.utils.constants import LDAP, INTERNAL
class LDAPLoginMockTestCase(BaseTestGenerator):
@@ -23,17 +24,17 @@ class LDAPLoginMockTestCase(BaseTestGenerator):
scenarios = [
('LDAP Authentication with Auto Create User', dict(
auth_source=['ldap'],
auth_source=[LDAP],
auto_create_user=True,
username='ldap_user',
password='ldap_pass')),
('LDAP Authentication without Auto Create User', dict(
auth_source=['ldap'],
auth_source=[LDAP],
auto_create_user=False,
username='ldap_user',
password='ldap_pass')),
('LDAP + Internal Authentication', dict(
auth_source=['ldap', 'internal'],
auth_source=[LDAP, INTERNAL],
auto_create_user=False,
username=config_data[
'pgAdmin4_login_credentials']['login_username'],
@@ -56,14 +57,15 @@ class LDAPLoginMockTestCase(BaseTestGenerator):
app_config.LDAP_ANONYMOUS_BIND = False
app_config.LDAP_BIND_USER = None
app_config.LDAP_BIND_PASSWORD = None
self.app.PGADMIN_EXTERNAL_AUTH_SOURCE = LDAP
@patch.object(AuthSourceRegistry._registry['ldap'], 'connect',
@patch.object(AuthSourceRegistry._registry[LDAP], 'connect',
return_value=[True, "Done"])
@patch.object(AuthSourceRegistry._registry['ldap'], 'search_ldap_user',
@patch.object(AuthSourceRegistry._registry[LDAP], 'search_ldap_user',
return_value=[True, ''])
def runTest(self, conn_mock_obj, search_mock_obj):
"""This function checks ldap login functionality."""
AuthSourceRegistry._registry['ldap'].dedicated_user = False
AuthSourceRegistry._registry[LDAP].dedicated_user = False
res = self.tester.login(self.username, self.password, True)
respdata = 'Gravatar image for %s' % self.username
self.assertTrue(respdata in res.data.decode('utf8'))
@@ -78,5 +80,5 @@ class LDAPLoginMockTestCase(BaseTestGenerator):
finishes.
"""
cls.tester.logout()
app_config.AUTHENTICATION_SOURCES = ['internal']
app_config.AUTHENTICATION_SOURCES = [INTERNAL]
utils.login_tester_account(cls.tester)