def configure_package(configure_security): test_jobs = [] try: test_jobs = config.get_all_jobs( node_address=config.get_foldered_node_address()) sdk_install.uninstall(config.get_foldered_service_name(), package_name=config.PACKAGE_NAME) sdk_upgrade.test_upgrade( "beta-{}".format(config.PACKAGE_NAME), config.PACKAGE_NAME, config.DEFAULT_TASK_COUNT, service_name=config.get_foldered_service_name(), additional_options={ "service": { "name": config.get_foldered_service_name() } }) tmp_dir = tempfile.mkdtemp(prefix='cassandra-test') for job in test_jobs: sdk_jobs.install_job(job, tmp_dir=tmp_dir) yield # let the test session execute finally: sdk_install.uninstall(config.get_foldered_service_name(), package_name=config.PACKAGE_NAME) for job in test_jobs: sdk_jobs.remove_job(job)
def test_xpack_enabled_update( service_name: str, from_xpack_enabled: bool, to_xpack_enabled: bool, ) -> None: sdk_upgrade.test_upgrade( PACKAGE_NAME, service_name, DEFAULT_TASK_COUNT, additional_options={ "elasticsearch": { "xpack_enabled": from_xpack_enabled } }, test_version_additional_options={ "service": { "update_strategy": "parallel" }, "elasticsearch": { "xpack_enabled": to_xpack_enabled }, }, ) wait_for_expected_nodes_to_exist(service_name=service_name, task_count=DEFAULT_TASK_COUNT)
def configure_package(configure_security): try: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME) if shakedown.dcos_version_less_than("1.9"): # HDFS upgrade in 1.8 is not supported. sdk_install.install(PACKAGE_NAME, DEFAULT_TASK_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={ "service": { "name": FOLDERED_SERVICE_NAME } }, timeout_seconds=30 * 60) else: sdk_upgrade.test_upgrade("beta-{}".format(PACKAGE_NAME), PACKAGE_NAME, DEFAULT_TASK_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={ "service": { "name": FOLDERED_SERVICE_NAME } }, timeout_seconds=30 * 60) yield # let the test session execute finally: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME)
def configure_package(configure_security): try: log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=config.PACKAGE_NAME) sdk_upgrade.test_upgrade("beta-{}".format(config.PACKAGE_NAME), config.PACKAGE_NAME, config.DEFAULT_TASK_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={ "service": { "name": FOLDERED_SERVICE_NAME }, "ingest_nodes": { "count": 1 } }) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=config.PACKAGE_NAME)
def test_update_from_xpack_enabled_to_xpack_security_enabled( service_name: str, xpack_enabled: bool, xpack_security_enabled: bool, ) -> None: assert not ( xpack_enabled is True and xpack_security_enabled is True ), "This function does not handle the 'xpack_enabled: True' to 'xpack_security_enabled: True' upgrade scenario" sdk_upgrade.test_upgrade( PACKAGE_NAME, service_name, DEFAULT_TASK_COUNT, additional_options={"elasticsearch": { "xpack_enabled": xpack_enabled }}, test_version_additional_options={ "service": { "update_strategy": "parallel" }, "elasticsearch": { "xpack_security_enabled": xpack_security_enabled }, }, ) wait_for_expected_nodes_to_exist(service_name=service_name, task_count=DEFAULT_TASK_COUNT)
def install_kafka(use_v0=False): mesos_api_version = "V0" if use_v0 else "V1" foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) if sdk_utils.dcos_version_less_than("1.9"): # Last beta-kafka release (1.1.25-0.10.1.0-beta) excludes 1.8. Skip upgrade tests with 1.8 and just install sdk_install.install(config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={ "service": { "name": foldered_name, "mesos_api_version": mesos_api_version } }) else: sdk_upgrade.test_upgrade(config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={ "service": { "name": foldered_name, "mesos_api_version": mesos_api_version }, "brokers": { "cpus": 0.5 } })
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) if shakedown.dcos_version_less_than("1.9"): # Last beta-kafka release (1.1.25-0.10.1.0-beta) excludes 1.8. Skip upgrade tests with 1.8 and just install sdk_install.install( config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={"service": { "name": foldered_name }}) else: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={"service": { "name": foldered_name }}) # wait for brokers to finish registering before starting tests test_utils.broker_count_check(config.DEFAULT_BROKER_COUNT, service_name=foldered_name) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): test_jobs = [] try: test_jobs = config.get_all_jobs( node_address=config.get_foldered_node_address()) # destroy any leftover jobs first, so that they don't touch the newly installed service: for job in test_jobs: sdk_jobs.remove_job(job) sdk_install.uninstall(config.PACKAGE_NAME, config.get_foldered_service_name()) sdk_upgrade.test_upgrade(config.PACKAGE_NAME, config.get_foldered_service_name(), config.DEFAULT_TASK_COUNT, additional_options={ "service": { "name": config.get_foldered_service_name() } }) tmp_dir = tempfile.mkdtemp(prefix='cassandra-test') for job in test_jobs: sdk_jobs.install_job(job, tmp_dir=tmp_dir) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, config.get_foldered_service_name()) for job in test_jobs: sdk_jobs.remove_job(job)
def configure_package(configure_security): try: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) if sdk_utils.dcos_version_less_than("1.9"): # HDFS upgrade in 1.8 is not supported. sdk_install.install( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": foldered_name}}, timeout_seconds=30 * 60, ) else: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options={"service": {"name": foldered_name}}, timeout_seconds=30 * 60, ) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def test_install_kerberised_hdfs_service(kerberos, service_account): try: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options={ "service": { "name": foldered_name, "service_account": service_account["name"], "service_account_secret": service_account["secret"], "security": { "kerberos": { "enabled": True, "debug": True, "kdc": { "hostname": kerberos.get_host(), "port": int(kerberos.get_port()), }, "realm": kerberos.get_realm(), "keytab_secret": kerberos.get_keytab_path(), } }, } }, timeout_seconds=30 * 60, ) finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade(config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={ "service": { "name": foldered_name }, "ingest_nodes": { "count": 1 } }) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) if sdk_utils.dcos_version_less_than("1.9"): # HDFS upgrade in 1.8 is not supported. sdk_install.install( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={"service": { "name": foldered_name }}, timeout_seconds=30 * 60) else: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={"service": { "name": foldered_name }}, timeout_seconds=30 * 60) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: install.uninstall(FOLDERED_SERVICE_NAME, package_name=config.PACKAGE_NAME) if shakedown.dcos_version_less_than("1.9"): # Last beta-kafka release (1.1.25-0.10.1.0-beta) excludes 1.8. Skip upgrade tests with 1.8 and just install install.install(config.PACKAGE_NAME, config.DEFAULT_BROKER_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={ "service": { "name": FOLDERED_SERVICE_NAME } }) else: sdk_upgrade.test_upgrade("beta-{}".format(config.PACKAGE_NAME), config.PACKAGE_NAME, config.DEFAULT_BROKER_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={ "service": { "name": FOLDERED_SERVICE_NAME } }) yield # let the test session execute finally: install.uninstall(FOLDERED_SERVICE_NAME, package_name=config.PACKAGE_NAME)
def configure_package(configure_security): test_jobs = [] try: test_jobs = config.get_all_jobs( node_address=config.get_foldered_node_address()) # destroy/reinstall any prior leftover jobs, so that they don't touch the newly installed service: for job in test_jobs: sdk_jobs.install_job(job) sdk_install.uninstall(config.PACKAGE_NAME, config.get_foldered_service_name()) sdk_upgrade.test_upgrade(config.PACKAGE_NAME, config.get_foldered_service_name(), config.DEFAULT_TASK_COUNT, additional_options={ "service": { "name": config.get_foldered_service_name() } }) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, config.get_foldered_service_name()) for job in test_jobs: sdk_jobs.remove_job(job)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={ "service": { "name": foldered_name }, "brokers": { "cpus": 0.5 } }, ) # wait for brokers to finish registering before starting tests test_utils.broker_count_check(config.DEFAULT_BROKER_COUNT, service_name=foldered_name) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def test_xpack_enabled_update( service_name: str, from_xpack_enabled: bool, to_xpack_enabled: bool, from_version: str, to_version: str = "stub-universe", ) -> None: sdk_upgrade.test_upgrade( PACKAGE_NAME, service_name, DEFAULT_TASK_COUNT, from_version=from_version, from_options={"elasticsearch": { "xpack_enabled": from_xpack_enabled }}, to_version=to_version, to_options={ "service": { "update_strategy": "parallel" }, "elasticsearch": { "xpack_enabled": to_xpack_enabled }, }, ) wait_for_expected_nodes_to_exist(service_name=service_name)
def test_custom_rack_upgrade(): foldered_service_name = config.get_foldered_service_name() service_options = {"service": {"name": foldered_service_name, "rack": "not-rack1"}} sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_service_name, config.DEFAULT_TASK_COUNT, additional_options=service_options, )
def test_custom_rack_upgrade() -> None: foldered_service_name = config.get_foldered_service_name() service_options = {"service": {"name": foldered_service_name, "rack": "not-rack1"}} sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_service_name, config.DEFAULT_TASK_COUNT, from_options=service_options, )
def configure_package(configure_security): try: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": FOLDERED_SERVICE_NAME, "user": "******"}}) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": foldered_name}}) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": foldered_name}}) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: service_options = {"service": {"name": foldered_name}} sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options=service_options, ) yield {"package_name": config.PACKAGE_NAME, **service_options} finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME) sdk_upgrade.test_upgrade( PACKAGE_NAME, PACKAGE_NAME, DEFAULT_TASK_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={"service": {"name": FOLDERED_SERVICE_NAME}}) yield # let the test session execute finally: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME)
def test_hdfs_upgrade(): try: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options={"service": {"name": foldered_name}}, timeout_seconds=30 * 60, ) finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: service_options = {"service": {"name": foldered_name}} sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options=service_options, ) yield {"package_name": config.PACKAGE_NAME, **service_options} finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def test_xpack_security_enabled_update( service_name: str, from_xpack_security_enabled: bool, to_xpack_security_enabled: bool ) -> None: sdk_upgrade.test_upgrade( PACKAGE_NAME, service_name, DEFAULT_TASK_COUNT, from_options={"elasticsearch": {"xpack_security_enabled": from_xpack_security_enabled}}, to_options={ "service": {"update_strategy": "parallel"}, "elasticsearch": {"xpack_security_enabled": to_xpack_security_enabled}, }, ) wait_for_expected_nodes_to_exist(service_name=service_name, task_count=DEFAULT_TASK_COUNT)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) service_options = {"service": {"name": foldered_name, "scenario": "CUSTOM_DECOMMISSION"}} sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_TASK_COUNT, from_options=service_options, to_options=service_options, ) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security): try: log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, current_expected_task_count, additional_options={ "service": {"name": foldered_name} }) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def configure_package(configure_security: None) -> Iterator[None]: try: log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(kibana_package_name, kibana_package_name) sdk_install.uninstall(package_name, service_name) sdk_upgrade.test_upgrade( package_name, service_name, current_expected_task_count, from_options={"service": {"name": service_name}}, ) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(kibana_package_name, kibana_package_name) sdk_install.uninstall(package_name, service_name)
def configure_package(configure_security): try: foldered_name = sdk_utils.get_foldered_name(config.SERVICE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, foldered_name) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, foldered_name, config.DEFAULT_BROKER_COUNT, additional_options={"service": {"name": foldered_name}, "brokers": {"cpus": 0.5}}) # wait for brokers to finish registering before starting tests test_utils.broker_count_check(config.DEFAULT_BROKER_COUNT, service_name=foldered_name) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, foldered_name)
def test_initial_upgrade(): options = {"service": {"name": SERVICE_NAME}} sdk_upgrade.test_upgrade( config.PACKAGE_NAME, SERVICE_NAME, 3, from_version=UPGRADE_FROM, from_options=options ) # Get the current service state to verify roles have applied. service_roles = sdk_utils.get_service_roles(SERVICE_NAME) current_task_roles = service_roles["task-roles"] # We must have some role! assert len(current_task_roles) > 0 assert LEGACY_ROLE in current_task_roles.values() assert ENFORCED_ROLE not in current_task_roles.values() assert service_roles["framework-roles"] is None assert service_roles["framework-role"] == ENFORCED_ROLE
def configure_package(configure_security): try: log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_TASK_COUNT, additional_options={ "service": {"name": FOLDERED_SERVICE_NAME}, "ingest_nodes": {"count": 1} }) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(config.KIBANA_PACKAGE_NAME, config.KIBANA_PACKAGE_NAME) sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME)
def configure_package(configure_security: None) -> Iterator[None]: try: log.info("Ensure elasticsearch and kibana are uninstalled...") sdk_install.uninstall(kibana_package_name, kibana_package_name) sdk_install.uninstall(package_name, service_name) sdk_upgrade.test_upgrade( package_name, service_name, current_expected_task_count, from_options={"service": { "name": service_name }}, ) yield # let the test session execute finally: log.info("Clean up elasticsearch and kibana...") sdk_install.uninstall(kibana_package_name, kibana_package_name) sdk_install.uninstall(package_name, service_name)
def configure_package(configure_security): try: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME) sdk_upgrade.test_upgrade( "beta-{}".format(PACKAGE_NAME), PACKAGE_NAME, DEFAULT_TASK_COUNT, service_name=FOLDERED_SERVICE_NAME, additional_options={"service": {"name": FOLDERED_SERVICE_NAME} }) tmp_dir = tempfile.mkdtemp(prefix='cassandra-test') for job in TEST_JOBS: sdk_jobs.install_job(job, tmp_dir=tmp_dir) yield # let the test session execute finally: sdk_install.uninstall(FOLDERED_SERVICE_NAME, package_name=PACKAGE_NAME) for job in TEST_JOBS: sdk_jobs.remove_job(job)
def configure_package(configure_security): try: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME) if shakedown.dcos_version_less_than("1.9"): # HDFS upgrade in 1.8 is not supported. sdk_install.install( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": FOLDERED_SERVICE_NAME}}, timeout_seconds=30 * 60) else: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_TASK_COUNT, additional_options={"service": {"name": FOLDERED_SERVICE_NAME}}, timeout_seconds=30 * 60) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME)
def configure_package(configure_security): try: install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME) if shakedown.dcos_version_less_than("1.9"): # Last beta-kafka release (1.1.25-0.10.1.0-beta) excludes 1.8. Skip upgrade tests with 1.8 and just install install.install( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_BROKER_COUNT, additional_options={"service": {"name": FOLDERED_SERVICE_NAME}}) else: sdk_upgrade.test_upgrade( config.PACKAGE_NAME, FOLDERED_SERVICE_NAME, config.DEFAULT_BROKER_COUNT, additional_options={"service": {"name": FOLDERED_SERVICE_NAME}}) # wait for brokers to finish registering before starting tests test_utils.broker_count_check(config.DEFAULT_BROKER_COUNT, service_name=FOLDERED_SERVICE_NAME) yield # let the test session execute finally: install.uninstall(config.PACKAGE_NAME, FOLDERED_SERVICE_NAME)
def configure_package(configure_security, kafka_client: client.KafkaClient): try: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_NAME) sdk_upgrade.test_upgrade( config.PACKAGE_NAME, FOLDERED_NAME, config.DEFAULT_BROKER_COUNT, additional_options={ "service": { "name": FOLDERED_NAME }, "brokers": { "cpus": 0.5 } }, ) # wait for brokers to finish registering before starting tests kafka_client.connect(config.DEFAULT_BROKER_COUNT) yield # let the test session execute finally: sdk_install.uninstall(config.PACKAGE_NAME, FOLDERED_NAME)