import random
import string

import pytest

from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)
node = cluster.add_instance(
    "node",
    main_configs=[
        "config/enable_keeper.xml",
        "config/users.xml",
    ],
    stay_alive=True,
    with_minio=True,
    macros={"shard": 1, "replica": 1},
)


@pytest.fixture(scope="module")
def start_cluster():
    try:
        cluster.start()
        yield cluster
    finally:
        cluster.shutdown()


def randomize_table_name(table_name, random_suffix_length=10):
    letters = string.ascii_letters + string.digits
    return f"{table_name}_{''.join(random.choice(letters) for _ in range(random_suffix_length))}"


@pytest.mark.parametrize("engine", ["ReplicatedMergeTree"])
def test_aliases_in_default_expr_not_break_table_structure(start_cluster, engine):
    """
    Making sure that using aliases in columns' default expressions does not lead to having different columns metadata in ZooKeeper and on disk.
    Issue: https://github.com/ClickHouse/clickhouse-private/issues/5150
    """

    data = '{"event": {"col1-key": "col1-val", "col2-key": "col2-val"}}'

    table_name = randomize_table_name("t")

    node.query(
        f"""
        DROP TABLE IF EXISTS {table_name};
        CREATE TABLE {table_name}
        (
            `data` String,
            `col1` String DEFAULT JSONExtractString(JSONExtractString(data, 'event') AS event, 'col1-key'),
            `col2` String MATERIALIZED JSONExtractString(JSONExtractString(data, 'event') AS event, 'col2-key')
        )
        ENGINE = {engine}('/test/{table_name}', '{{replica}}')
        ORDER BY col1
        """
    )

    node.restart_clickhouse()

    node.query(
        f"""
        INSERT INTO {table_name} (data) VALUES ('{data}');
        """
    )
    assert node.query(f"SELECT data FROM {table_name}").strip() == data
    assert node.query(f"SELECT col1 FROM {table_name}").strip() == "col1-val"
    assert node.query(f"SELECT col2 FROM {table_name}").strip() == "col2-val"

    node.query(f"DROP TABLE {table_name}")