from helpers.cluster import ClickHouseCluster


def test_extra_yaml_mix():
    # some configs are written in XML, others are written in YAML
    cluster = ClickHouseCluster(
        __file__, zookeeper_config_path="configs/config.d/zookeeper.xml"
    )

    all_confd = [
        "configs/config.d/0_common_instance_config.yaml",
        "configs/config.d/access_control.yaml",
        "configs/config.d/error_log.xml",
        "configs/config.d/keeper_port.xml",
        "configs/config.d/logging_no_rotate.xml",
        "configs/config.d/log_to_console.yaml",
        "configs/config.d/macros.yaml",
        "configs/config.d/mark_cache_size.yaml",
        "configs/config.d/metric_log.xml",
        "configs/config.d/more_clusters.yaml",
        "configs/config.d/part_log.xml",
        "configs/config.d/path.yaml",
        "configs/config.d/query_masking_rules.xml",
        "configs/config.d/tcp_with_proxy.yaml",
        "configs/config.d/test_cluster_with_incorrect_pw.xml",
        "configs/config.d/text_log.yaml",
        "configs/config.d/zookeeper.xml",
    ]

    all_userd = [
        "configs/users.d/allow_introspection_functions.xml",
        "configs/users.d/log_queries.yaml",
    ]

    node = cluster.add_instance(
        "node",
        base_config_dir="configs",
        main_configs=all_confd,
        user_configs=all_userd,
        with_zookeeper=False,
        users_config_name="users.yaml",
        copy_common_configs=False,
        config_root_name="clickhouse",
        env_variables={"CONFIG_TEST_ENV": "8956"},
    )

    try:
        cluster.start()
        assert (
            node.query(
                "select value from system.settings where name = 'max_memory_usage'"
            )
            == "10000000000\n"
        )
        assert (
            node.query(
                "select value from system.settings where name = 'max_block_size'"
            )
            == "64999\n"
        )
        assert (
            node.query(
                "select value from system.server_settings where name = 'mark_cache_size'"
            )
            == "8956\n"
        )

    finally:
        cluster.shutdown()