import pytest

from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)
node = cluster.add_instance("node", main_configs=["configs/config.xml"])
path_to_userfiles_from_defaut_config = "user_files"


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


def test_filepath(start_cluster):
    # 2 rows data
    some_data = "Test\t111.222\nData\t333.444"

    node.exec_in_container(
        ["bash", "-c", "mkdir -p {}".format(path_to_userfiles_from_defaut_config)],
        privileged=True,
        user="root",
    )

    node.exec_in_container(
        [
            "bash",
            "-c",
            'echo "{}" > {}'.format(
                some_data,
                path_to_userfiles_from_defaut_config + "/relative_user_file_test",
            ),
        ],
        privileged=True,
        user="root",
    )

    test_requests = [
        ("relative_user_file_test", "2"),
        (
            "../" + path_to_userfiles_from_defaut_config + "/relative_user_file_test",
            "2",
        ),
    ]

    for pattern, value in test_requests:
        assert (
            node.query(
                """
            select count() from file('{}', 'TSV', 'text String, number Float64')
        """.format(
                    pattern
                )
            )
            == "{}\n".format(value)
        )