2626from exasol_integration_test_docker_environment .lib .test_environment .parameter .docker_db_test_environment_parameter import ( # pylint: disable=line-too-long
2727 DbOsAccess ,
2828)
29+ from exasol_integration_test_docker_environment .lib .test_environment .ports import Ports
2930from exasol_integration_test_docker_environment .lib .utils .api_function_decorators import (
3031 cli_function ,
3132)
@@ -60,8 +61,8 @@ def run_db_test(
6061 additional_db_parameter : tuple [str , ...] = tuple (),
6162 docker_environment_variable : tuple [str , ...] = tuple (),
6263 external_exasol_db_host : Optional [str ] = None ,
63- external_exasol_db_port : int = 8563 ,
64- external_exasol_bucketfs_port : int = 2580 ,
64+ external_exasol_db_port : int = Ports . external . database ,
65+ external_exasol_bucketfs_port : int = Ports . external . bucketfs_http ,
6566 external_exasol_ssh_port : Optional [int ] = None ,
6667 external_exasol_db_user : Optional [str ] = None ,
6768 external_exasol_db_password : Optional [str ] = None ,
@@ -104,6 +105,7 @@ def run_db_test(
104105 use_job_specific_log_file : bool = True ,
105106 compression_strategy : CompressionStrategy = defaultCompressionStrategy (),
106107 accelerator : Accelerator = defaultAccelerator (),
108+ external_exasol_bucketfs_https_port : int = Ports .external .bucketfs_https ,
107109) -> AllTestsResult :
108110 """
109111 This command runs the integration tests in local docker-db.
@@ -157,10 +159,12 @@ def run_db_test(
157159 raise api_errors .MissingArgumentError ("external_exasol_ssh_port" )
158160
159161 docker_runtime = None
162+ itde_accelerator : tuple [str , ...] = ()
160163 if accelerator == Accelerator .NVIDA :
161- additional_db_parameter += ("-enableAcceleratorDeviceDetection=1" ,)
162- docker_runtime = "nvidia"
163- docker_environment_variable += ("NVIDIA_VISIBLE_DEVICES=all" ,)
164+ itde_accelerator = ("nvidia=all" ,)
165+ add_db_param_accel_detection = "-enableAcceleratorDeviceDetection=1"
166+ if add_db_param_accel_detection not in additional_db_parameter :
167+ additional_db_parameter += (add_db_param_accel_detection ,)
164168
165169 def root_task_generator () -> DependencyLoggerBaseTask :
166170 return generate_root_task (
@@ -191,7 +195,8 @@ def root_task_generator() -> DependencyLoggerBaseTask:
191195 max_start_attempts = max_start_attempts ,
192196 external_exasol_db_host = external_exasol_db_host ,
193197 external_exasol_db_port = external_exasol_db_port ,
194- external_exasol_bucketfs_port = external_exasol_bucketfs_port ,
198+ external_exasol_bucketfs_http_port = external_exasol_bucketfs_port ,
199+ external_exasol_bucketfs_https_port = external_exasol_bucketfs_https_port ,
195200 external_exasol_db_user = external_exasol_db_user ,
196201 external_exasol_db_password = external_exasol_db_password ,
197202 external_exasol_ssh_port = external_exasol_ssh_port ,
@@ -207,6 +212,7 @@ def root_task_generator() -> DependencyLoggerBaseTask:
207212 compression_strategy = compression_strategy ,
208213 docker_runtime = docker_runtime ,
209214 docker_environment_variables = docker_environment_variable ,
215+ accelerator = itde_accelerator ,
210216 )
211217
212218 return run_task (
0 commit comments