QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-22 01:24:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(OAuth2): refactor for custom OAuth2 clients (#27880)

Browse Source
This commit is contained in:
Beto Dealmeida
2024-04-05 12:17:39 -04:00
committed by GitHub
parent 62433c14a7
commit 9377227e06
22 changed files with 385 additions and 297 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/unit_tests/databases/api_test.py
View File

@@ -668,6 +668,11 @@ def test_oauth2_happy_path(
)
db.session.commit()
mocker.patch.object(
SqliteEngineSpec,
"get_oauth2_config",
return_value={"id": "one", "secret": "two"},
)
get_oauth2_token = mocker.patch.object(SqliteEngineSpec, "get_oauth2_token")
get_oauth2_token.return_value = {
"access_token": "YYY",
@@ -696,7 +701,7 @@ def test_oauth2_happy_path(
assert response.status_code == 200
decode_oauth2_state.assert_called_with("some%2Estate")
get_oauth2_token.assert_called_with("XXX", state)
get_oauth2_token.assert_called_with({"id": "one", "secret": "two"}, "XXX")
token = db.session.query(DatabaseUserOAuth2Tokens).one()
assert token.user_id == 1
@@ -731,6 +736,11 @@ def test_oauth2_multiple_tokens(
)
db.session.commit()
mocker.patch.object(
SqliteEngineSpec,
"get_oauth2_config",
return_value={"id": "one", "secret": "two"},
)
get_oauth2_token = mocker.patch.object(SqliteEngineSpec, "get_oauth2_token")
get_oauth2_token.side_effect = [
{