Skip to content

Commit 048b0c2

Browse files
committed
Refactored existing unit tests
1 parent 0e673f0 commit 048b0c2

1 file changed

Lines changed: 68 additions & 8 deletions

File tree

tests/unit/models/config/test_authentication_configuration.py

Lines changed: 68 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ def test_authentication_configuration_module_unsupported() -> None:
229229
)
230230

231231

232-
def test_authentication_configuration_in_config() -> None:
232+
def test_authentication_configuration_in_config_noop() -> None:
233233
"""Test the authentication configuration in main config."""
234234
# pylint: disable=no-member
235235
cfg = Configuration(
@@ -250,7 +250,11 @@ def test_authentication_configuration_in_config() -> None:
250250
assert cfg.authentication.k8s_ca_cert_path is None
251251
assert cfg.authentication.k8s_cluster_api is None
252252

253-
cfg2 = Configuration(
253+
254+
def test_authentication_configuration_in_config_k8s() -> None:
255+
"""Test the authentication configuration in main config."""
256+
# pylint: disable=no-member
257+
cfg = Configuration(
254258
name="test_name",
255259
service=ServiceConfiguration(),
256260
llama_stack=LlamaStackConfiguration(
@@ -268,10 +272,66 @@ def test_authentication_configuration_in_config() -> None:
268272
k8s_cluster_api=None,
269273
),
270274
)
271-
assert cfg2.authentication is not None
272-
assert cfg2.authentication.module == AUTH_MOD_K8S
273-
assert cfg2.authentication.skip_tls_verification is True
274-
assert cfg2.authentication.k8s_ca_cert_path == Path(
275-
"tests/configuration/server.crt"
275+
assert cfg.authentication is not None
276+
assert cfg.authentication.module == AUTH_MOD_K8S
277+
assert cfg.authentication.skip_tls_verification is True
278+
assert cfg.authentication.k8s_ca_cert_path == Path("tests/configuration/server.crt")
279+
assert cfg.authentication.k8s_cluster_api is None
280+
281+
282+
def test_authentication_configuration_in_config_rh_identity() -> None:
283+
"""Test the authentication configuration in main config."""
284+
# pylint: disable=no-member
285+
cfg = Configuration(
286+
name="test_name",
287+
service=ServiceConfiguration(),
288+
llama_stack=LlamaStackConfiguration(
289+
use_as_library_client=True,
290+
library_client_config_path="tests/configuration/run.yaml",
291+
),
292+
user_data_collection=UserDataCollection(
293+
feedback_enabled=False, feedback_storage=None
294+
),
295+
mcp_servers=[],
296+
authentication=AuthenticationConfiguration(
297+
module=AUTH_MOD_RH_IDENTITY,
298+
skip_tls_verification=True,
299+
k8s_ca_cert_path="tests/configuration/server.crt",
300+
k8s_cluster_api=None,
301+
rh_identity_config=RHIdentityConfiguration(required_entitlements=[]),
302+
),
276303
)
277-
assert cfg2.authentication.k8s_cluster_api is None
304+
assert cfg.authentication is not None
305+
assert cfg.authentication.module == AUTH_MOD_RH_IDENTITY
306+
assert cfg.authentication.skip_tls_verification is True
307+
assert cfg.authentication.k8s_ca_cert_path == Path("tests/configuration/server.crt")
308+
assert cfg.authentication.k8s_cluster_api is None
309+
310+
311+
def test_authentication_configuration_in_config_jwktoken() -> None:
312+
"""Test the authentication configuration in main config."""
313+
# pylint: disable=no-member
314+
cfg = Configuration(
315+
name="test_name",
316+
service=ServiceConfiguration(),
317+
llama_stack=LlamaStackConfiguration(
318+
use_as_library_client=True,
319+
library_client_config_path="tests/configuration/run.yaml",
320+
),
321+
user_data_collection=UserDataCollection(
322+
feedback_enabled=False, feedback_storage=None
323+
),
324+
mcp_servers=[],
325+
authentication=AuthenticationConfiguration(
326+
module=AUTH_MOD_JWK_TOKEN,
327+
skip_tls_verification=True,
328+
k8s_ca_cert_path="tests/configuration/server.crt",
329+
k8s_cluster_api=None,
330+
jwk_config=JwkConfiguration(url="http://foo.bar.baz"),
331+
),
332+
)
333+
assert cfg.authentication is not None
334+
assert cfg.authentication.module == AUTH_MOD_JWK_TOKEN
335+
assert cfg.authentication.skip_tls_verification is True
336+
assert cfg.authentication.k8s_ca_cert_path == Path("tests/configuration/server.crt")
337+
assert cfg.authentication.k8s_cluster_api is None

0 commit comments

Comments
 (0)