mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-04 00:51:06 -05:00
f9f0529e8e
The test waited a fixed 0.5s after each PUT/DELETE before reading from every peer, which raced with raft apply on followers under CI load. Replace the fixed sleeps with wait_for-based polling so each per-peer read retries until the expected value is observed. Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
import pathlib
|
|
from typing import Any
|
|
|
|
from .utils import *
|
|
|
|
def test_cluster_metadata(tmp_path: pathlib.Path):
|
|
assert_project_root()
|
|
|
|
peer_api_uris, _peer_dirs, _bootstrap_uri = start_cluster(tmp_path, 3, None)
|
|
|
|
# Get (empty) metadata key list
|
|
get_metadata_keys(peer_api_uris, [])
|
|
|
|
# Put some metadata keys
|
|
put_metadata_key(peer_api_uris, 'string', 'string')
|
|
|
|
put_metadata_key(peer_api_uris, 'array', [
|
|
None,
|
|
1337,
|
|
42.69,
|
|
"string",
|
|
])
|
|
|
|
put_metadata_key(peer_api_uris, 'object', {
|
|
'null': None,
|
|
'integer': 1337,
|
|
'float': 42.69,
|
|
'string': 'string',
|
|
})
|
|
|
|
# Get metadata key list
|
|
get_metadata_keys(peer_api_uris, ['string', 'array', 'object'])
|
|
|
|
# Delete metadata keys
|
|
for key in ['string', 'array', 'object']:
|
|
resp = requests.delete(f"{peer_api_uris[0]}/cluster/metadata/keys/{key}")
|
|
assert_http_ok(resp)
|
|
get_metadata_key(peer_api_uris, key, None)
|
|
|
|
# Get (empty) metadata key list
|
|
get_metadata_keys(peer_api_uris, [])
|
|
|
|
def get_metadata_keys(peer_uris: list[str], keys: list[str]):
|
|
expected = set(keys)
|
|
for peer_uri in peer_uris:
|
|
wait_for(_metadata_keys_match, peer_uri, expected)
|
|
|
|
def _metadata_keys_match(peer_uri: str, expected: set) -> bool:
|
|
resp = requests.get(f"{peer_uri}/cluster/metadata/keys")
|
|
assert_http_ok(resp)
|
|
return set(resp.json()['result']) == expected
|
|
|
|
def put_metadata_key(peer_uris: list[str], key: str, value: Any):
|
|
resp = requests.put(f"{peer_uris[0]}/cluster/metadata/keys/{key}", json=value)
|
|
assert_http_ok(resp)
|
|
get_metadata_key(peer_uris, key, value)
|
|
|
|
def get_metadata_key(peer_uris: list[str], key: str, expected_value: Any):
|
|
for peer_uri in peer_uris:
|
|
wait_for(_metadata_key_matches, peer_uri, key, expected_value)
|
|
|
|
def _metadata_key_matches(peer_uri: str, key: str, expected_value: Any) -> bool:
|
|
resp = requests.get(f"{peer_uri}/cluster/metadata/keys/{key}")
|
|
assert_http_ok(resp)
|
|
return resp.json()['result'] == expected_value
|