mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-16 03:12:43 +00:00
1bbf5acd47
* Pass headers from StorageURL to WriteBufferFromHTTP * Add a test * Lint * `time.sleep(1)` * Start echo server earlier * Add proper handling for mock server start * Automatic style fix --------- Co-authored-by: robot-clickhouse <robot-clickhouse@users.noreply.github.com>
67 lines
1.5 KiB
Python
67 lines
1.5 KiB
Python
import pytest
|
|
import os
|
|
import time
|
|
|
|
from . import http_headers_echo_server
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
server = cluster.add_instance("node")
|
|
|
|
|
|
def run_echo_server():
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
server.copy_file_to_container(
|
|
os.path.join(script_dir, "http_headers_echo_server.py"),
|
|
"/http_headers_echo_server.py",
|
|
)
|
|
|
|
server.exec_in_container(
|
|
[
|
|
"bash",
|
|
"-c",
|
|
"python3 /http_headers_echo_server.py > /http_headers_echo.server.log 2>&1",
|
|
],
|
|
detach=True,
|
|
user="root",
|
|
)
|
|
|
|
for _ in range(0, 10):
|
|
ping_response = server.exec_in_container(
|
|
["curl", "-s", f"http://localhost:8000/"],
|
|
nothrow=True,
|
|
)
|
|
|
|
if "html" in ping_response:
|
|
return
|
|
|
|
print(ping_response)
|
|
|
|
raise Exception("Echo server is not responding")
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
run_echo_server()
|
|
yield cluster
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_storage_url_http_headers(started_cluster):
|
|
query = "INSERT INTO TABLE FUNCTION url('http://localhost:8000/', JSON, 'a UInt64', headers('X-My-Custom-Header'='test-header')) VALUES (1)"
|
|
|
|
server.query(query)
|
|
|
|
result = server.exec_in_container(
|
|
["cat", http_headers_echo_server.RESULT_PATH], user="root"
|
|
)
|
|
|
|
print(result)
|
|
|
|
assert "X-My-Custom-Header: test-header" in result
|