From c8769fb36a5872fcf2f4f5ff92e1e4d638ec83ae Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alperen=20K=C3=B6m=C3=BCrc=C3=BC?= Date: Thu, 3 Sep 2026 10:12:34 +0200 Subject: [PATCH] fix(test): fix batch service e2e test --- .../batch_service/test_base.py | 45 ++++++++++++++----- 1 file changed, 33 insertions(+), 12 deletions(-) diff --git a/packages/gen/integration_tests/batch_service/test_base.py b/packages/gen/integration_tests/batch_service/test_base.py index 282aa303..52d500c1 100644 --- a/packages/gen/integration_tests/batch_service/test_base.py +++ b/packages/gen/integration_tests/batch_service/test_base.py @@ -11,6 +11,7 @@ from time import sleep +from ai_api_client_sdk.exception import AIAPIServerException from gen_ai_hub import GenAIHubProxyClient from gen_ai_hub.proxy import get_proxy_client from gen_ai_hub.batch_service.models.response import BatchStatus @@ -49,16 +50,36 @@ def wait_for_batch_to_be_deletable(self, batch_id: str, timeout=300, polling_int def create_object_store_secret(cls): aws_access_key_id = os.environ.get(AWS_ACCESS_KEY_ID) aws_secret_access_key = os.environ.get(AWS_SECRET_ACCESS_KEY) - response = cls.proxy_client.ai_core_client.object_store_secrets.create( - name=cls.secret_name, - type="S3", - data={ - AWS_ACCESS_KEY_ID: aws_access_key_id, - AWS_SECRET_ACCESS_KEY: aws_secret_access_key, - }, - bucket=os.environ.get(AWS_BUCKET_NAME), - endpoint=os.environ.get(AWS_HOST), - region=os.environ.get(AWS_REGION), - resource_group=os.environ.get(AICORE_RESOURCE_GROUP), - ) + try: + response = cls.proxy_client.ai_core_client.object_store_secrets.create( + name=cls.secret_name, + type="S3", + data={ + AWS_ACCESS_KEY_ID: aws_access_key_id, + AWS_SECRET_ACCESS_KEY: aws_secret_access_key, + }, + bucket=os.environ.get(AWS_BUCKET_NAME), + endpoint=os.environ.get(AWS_HOST), + region=os.environ.get(AWS_REGION), + resource_group=os.environ.get(AICORE_RESOURCE_GROUP), + ) + except AIAPIServerException as e: + if e.status_code == 409: + if 'already exists with different data' in e.error_message: + cls.proxy_client.ai_core_client.object_store_secrets.modify( + name=cls.secret_name, + type="S3", + data={ + AWS_ACCESS_KEY_ID: aws_access_key_id, + AWS_SECRET_ACCESS_KEY: aws_secret_access_key, + }, + bucket=os.environ.get(AWS_BUCKET_NAME), + endpoint=os.environ.get(AWS_HOST), + region=os.environ.get(AWS_REGION), + resource_group=os.environ.get(AICORE_RESOURCE_GROUP), + ) + else: + raise Exception( + f"Error while creating the Object Store secret. Request failed with error of {e}" + ) from e