Skip to content

Commit 897da28

Browse files
committed
refactor: use helper
1 parent c0490dd commit 897da28

1 file changed

Lines changed: 13 additions & 17 deletions

File tree

tests/rest/test_connection.py

Lines changed: 13 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -790,6 +790,16 @@ def test_create_connection_lazy_refresh_token_store(requests_mock):
790790
)
791791

792792

793+
def assert_provider(providers, provider_id, expected_type, expected_issuer, expected_title):
794+
# helper for test_list_auth_providers
795+
provider = next(filter(lambda x: x["id"] == provider_id, providers), None)
796+
assert isinstance(provider, dict)
797+
assert provider["type"] == expected_type
798+
assert provider["issuer"] == expected_issuer
799+
assert provider["title"] == expected_title
800+
return provider
801+
802+
793803
def test_list_auth_providers(requests_mock, api_version, token_type):
794804
requests_mock.get(API_URL, json=build_capabilities(api_version=api_version, token_type=token_type))
795805
requests_mock.get(
@@ -806,23 +816,9 @@ def test_list_auth_providers(requests_mock, api_version, token_type):
806816
providers = conn.list_auth_providers()
807817
assert len(providers) == 3
808818

809-
p1 = next(filter(lambda x: x["id"] == "p1", providers), None)
810-
assert isinstance(p1, dict)
811-
assert p1["type"] == "oidc"
812-
assert p1["issuer"] == "https://openeo.example"
813-
assert p1["title"] == "openEO"
814-
815-
p2 = next(filter(lambda x: x["id"] == "p2", providers), None)
816-
assert isinstance(p2, dict)
817-
assert p2["type"] == "oidc"
818-
assert p2["issuer"] == "https://other.example"
819-
assert p2["title"] == "Other"
820-
821-
basic = next(filter(lambda x: x["id"] == "/credentials/basic", providers), None)
822-
assert isinstance(basic, dict)
823-
assert basic["type"] == "basic"
824-
assert basic["issuer"] == API_URL + "credentials/basic"
825-
assert basic["title"] == "Internal"
819+
assert_provider(providers, "p1", "oidc", "https://openeo.example", "openEO")
820+
assert_provider(providers, "p2", "oidc", "https://other.example", "Other")
821+
assert_provider(providers, "/credentials/basic", "basic", API_URL + "credentials/basic", "Internal")
826822

827823

828824
def test_list_auth_providers_empty(requests_mock, api_version, token_type):

0 commit comments

Comments
 (0)