mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 00:22:29 +00:00
58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
import os
|
|
import math
|
|
import subprocess
|
|
from tempfile import NamedTemporaryFile
|
|
import pytest
|
|
|
|
|
|
def run_command_in_container(cmd, *args):
|
|
# /clickhouse is mounted by integration tests runner
|
|
alternative_binary = os.getenv("CLICKHOUSE_BINARY", "/clickhouse")
|
|
if alternative_binary:
|
|
args += (
|
|
"--volume",
|
|
f"{alternative_binary}:/usr/bin/clickhouse",
|
|
)
|
|
|
|
command = [
|
|
"docker",
|
|
"run",
|
|
"--rm",
|
|
*args,
|
|
"ubuntu:22.04",
|
|
"sh",
|
|
"-c",
|
|
cmd,
|
|
]
|
|
|
|
logging.debug("Command: %s", " ".join(command))
|
|
return subprocess.check_output(command)
|
|
|
|
|
|
def run_with_cpu_limit(cmd, num_cpus, *args):
|
|
args += (
|
|
"--cpus",
|
|
f"{num_cpus}",
|
|
)
|
|
return run_command_in_container(cmd, *args)
|
|
|
|
|
|
def test_cgroup_cpu_limit():
|
|
for num_cpus in (1, 2, 4, 2.8):
|
|
result = run_with_cpu_limit(
|
|
"clickhouse local -q \"select value from system.settings where name='max_threads'\"",
|
|
num_cpus,
|
|
)
|
|
expect_output = (r"\'auto({})\'".format(math.ceil(num_cpus))).encode()
|
|
assert (
|
|
result.strip() == expect_output
|
|
), f"fail for cpu limit={num_cpus}, result={result.strip()}, expect={expect_output}"
|
|
|
|
|
|
# For manual run
|
|
if __name__ == "__main__":
|
|
test_cgroup_cpu_limit()
|