mirror of
https://github.com/SigNoz/signoz.git
synced 2025-12-17 15:36:48 +00:00
eb74adad44
* test(integration): set the base for integration tests * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test * ci: add ci pipeline for integration test
112 lines
3.3 KiB
Python
112 lines
3.3 KiB
Python
import os
|
|
from typing import Any, Generator
|
|
|
|
import clickhouse_driver
|
|
import pytest
|
|
from testcontainers.clickhouse import ClickHouseContainer
|
|
from testcontainers.core.container import Network
|
|
|
|
from fixtures import types
|
|
|
|
|
|
@pytest.fixture(name="clickhouse", scope="package")
|
|
def clickhouse(
|
|
tmpfs: Generator[types.LegacyPath, Any, None],
|
|
network: Network,
|
|
zookeeper: types.TestContainerDocker,
|
|
request: pytest.FixtureRequest,
|
|
) -> types.TestContainerClickhouse:
|
|
"""
|
|
Package-scoped fixture for Clickhouse TestContainer.
|
|
"""
|
|
version = request.config.getoption("--clickhouse-version")
|
|
|
|
container = ClickHouseContainer(
|
|
image=f"clickhouse/clickhouse-server:{version}",
|
|
port=9000,
|
|
username="signoz",
|
|
password="password",
|
|
)
|
|
|
|
cluster_config = f"""
|
|
<clickhouse>
|
|
<logger>
|
|
<level>information</level>
|
|
<formatting>
|
|
<type>json</type>
|
|
</formatting>
|
|
<log>/var/log/clickhouse-server/clickhouse-server.log</log>
|
|
<errorlog>/var/log/clickhouse-server/clickhouse-server.err.log</errorlog>
|
|
<size>1000M</size>
|
|
<count>3</count>
|
|
<console>1</console>
|
|
</logger>
|
|
|
|
<macros>
|
|
<shard>01</shard>
|
|
<replica>01</replica>
|
|
</macros>
|
|
|
|
<zookeeper>
|
|
<node>
|
|
<host>{zookeeper.container_config.address}</host>
|
|
<port>{zookeeper.container_config.port}</port>
|
|
</node>
|
|
</zookeeper>
|
|
|
|
<remote_servers>
|
|
<cluster>
|
|
<shard>
|
|
<replica>
|
|
<host>127.0.0.1</host>
|
|
<port>9000</port>
|
|
</replica>
|
|
</shard>
|
|
</cluster>
|
|
</remote_servers>
|
|
|
|
<distributed_ddl>
|
|
<path>/clickhouse/task_queue/ddl</path>
|
|
<profile>default</profile>
|
|
</distributed_ddl>
|
|
</clickhouse>
|
|
"""
|
|
|
|
tmp_dir = tmpfs("clickhouse")
|
|
cluster_config_file_path = os.path.join(tmp_dir, "cluster.xml")
|
|
with open(cluster_config_file_path, "w", encoding="utf-8") as f:
|
|
f.write(cluster_config)
|
|
|
|
container.with_volume_mapping(
|
|
cluster_config_file_path, "/etc/clickhouse-server/config.d/cluster.xml"
|
|
)
|
|
container.with_network(network)
|
|
container.start()
|
|
|
|
connection = clickhouse_driver.connect(
|
|
user=container.username,
|
|
password=container.password,
|
|
host=container.get_container_host_ip(),
|
|
port=container.get_exposed_port(9000),
|
|
)
|
|
|
|
def stop():
|
|
connection.close()
|
|
container.stop(delete_volume=True)
|
|
|
|
request.addfinalizer(stop)
|
|
|
|
return types.TestContainerClickhouse(
|
|
container=container,
|
|
host_config=types.TestContainerUrlConfig(
|
|
"tcp", container.get_container_host_ip(), container.get_exposed_port(9000)
|
|
),
|
|
container_config=types.TestContainerUrlConfig(
|
|
"tcp", container.get_wrapped_container().name, 9000
|
|
),
|
|
conn=connection,
|
|
env={
|
|
"SIGNOZ_TELEMETRYSTORE_CLICKHOUSE_DSN": f"tcp://{container.username}:{container.password}@{container.get_wrapped_container().name}:{9000}" # pylint: disable=line-too-long
|
|
},
|
|
)
|