mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2025-02-25 18:55:31 -06:00
Added support for OAuth 2 authentication. Fixes #5940
Initial patch sent by: Florian Sabonchi
This commit is contained in:
committed by
Akshay Joshi
parent
fff4060b31
commit
48ca83f31d
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user