feat: add auth test to upgrade

feat: add auth test to upgrade

Author: rnetser

tests/model_serving/model_server/upgrade/conftest.py

@@ -394,6 +394,7 @@ def model_service_account_scope_session(
             namespace=ci_endpoint_s3_secret_scope_session.namespace,
             name="upgrade-models-bucket-sa",
             secrets=[{"name": ci_endpoint_s3_secret_scope_session.name}],
+            teardown=teardown_resources,
         ) as sa:
             yield sa
 
@@ -457,6 +458,7 @@ def http_role_binding_scope_session(
             role_ref_kind=http_view_role_scope_session.kind,
             subjects_kind=model_service_account_scope_session.kind,
             subjects_name=model_service_account_scope_session.name,
+            teardown=teardown_resources,
         ) as rb:
             yield rb
 

utilities/infra.py

@@ -114,14 +114,16 @@ def create_ns(
             project.wait_for_status(status=project.Status.ACTIVE, timeout=Timeout.TIMEOUT_2MIN)
             yield project
 
-            wait_for_serverless_pods_deletion(resource=project, admin_client=admin_client)
+            if teardown:
+                wait_for_serverless_pods_deletion(resource=project, admin_client=admin_client)
 
     else:
         with Namespace(**namespace_kwargs) as ns:
             ns.wait_for_status(status=Namespace.Status.ACTIVE, timeout=Timeout.TIMEOUT_2MIN)
             yield ns
 
-            wait_for_serverless_pods_deletion(resource=ns, admin_client=admin_client)
+            if teardown:
+                wait_for_serverless_pods_deletion(resource=ns, admin_client=admin_client)
 
 
 def wait_for_replicas_in_deployment(deployment: Deployment, replicas: int) -> None: