Add test for upgrade_channel_to_proxy_authenticated

flyteorg · Aug 11, 2023 · 7987098 · 7987098
1 parent 4038eb3
commit 7987098
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/flytekit/unit/clients/test_auth_helper.py b/tests/flytekit/unit/clients/test_auth_helper.py
@@ -21,6 +21,7 @@
     get_session,
     load_cert,
     upgrade_channel_to_authenticated,
+    upgrade_channel_to_proxy_authenticated,
     wrap_exceptions_channel,
 )
 from flytekit.clients.grpc_utils.auth_interceptor import AuthUnaryInterceptor
@@ -163,6 +164,19 @@ def test_upgrade_channel_to_auth():
     assert isinstance(out_ch._interceptor, AuthUnaryInterceptor)  # noqa
 
 
+def test_upgrade_channel_to_proxy_auth():
+    ch = MagicMock()
+    out_ch = upgrade_channel_to_proxy_authenticated(
+        PlatformConfig(
+            auth_mode="Pkce",
+            proxy_command=["echo", "foo-bar"],
+        ),
+        ch,
+    )
+    assert isinstance(out_ch._interceptor, AuthUnaryInterceptor)
+    assert isinstance(out_ch._interceptor._authenticator, CommandAuthenticator)
+
+
 def test_load_cert():
     cert_file = os.path.join(os.path.dirname(__file__), "testdata", "rootCACert.pem")
     f = load_cert(cert_file)