mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-30 22:51:00 -05:00
104 lines
3.6 KiB
Python
104 lines
3.6 KiB
Python
import pytest
|
|
from datetime import datetime
|
|
|
|
from .helpers.collection_setup import basic_collection_setup, drop_collection
|
|
from .helpers.helpers import request_with_validation
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope="module")
|
|
def setup(on_disk_vectors, collection_name):
|
|
basic_collection_setup(collection_name=collection_name, on_disk_vectors=on_disk_vectors)
|
|
yield
|
|
drop_collection(collection_name=collection_name)
|
|
|
|
|
|
def test_metrics():
|
|
response = request_with_validation(
|
|
api='/metrics',
|
|
method="GET",
|
|
)
|
|
assert response.ok
|
|
|
|
# Probe some strings that must exist in the metrics output
|
|
assert '# HELP app_info information about qdrant server' in response.text
|
|
assert '# TYPE app_info gauge' in response.text
|
|
assert 'app_info{name="qdrant",version="' in response.text
|
|
assert 'collections_total ' in response.text
|
|
|
|
|
|
def test_telemetry():
|
|
response = request_with_validation(
|
|
api='/telemetry',
|
|
method="GET",
|
|
)
|
|
|
|
assert response.ok
|
|
|
|
result = response.json()['result']
|
|
|
|
assert result['collections']['number_of_collections'] >= 1
|
|
|
|
endpoint = result['requests']['rest']['responses']['PUT /collections/{name}/points']
|
|
assert endpoint['200']['count'] > 0
|
|
|
|
assert 'avg_duration_micros' in endpoint['200']
|
|
|
|
|
|
@pytest.mark.parametrize("level", [0, 1, 2, 3, 10])
|
|
def test_telemetry_detail(level: int):
|
|
response = request_with_validation(
|
|
api='/telemetry',
|
|
method="GET",
|
|
query_params={'details_level': level},
|
|
)
|
|
|
|
assert response.ok
|
|
|
|
result = response.json()['result']
|
|
|
|
assert result['collections']['number_of_collections'] >= 1
|
|
|
|
endpoint = result['requests']['rest']['responses']['PUT /collections/{name}/points']
|
|
assert endpoint['200']['count'] > 0
|
|
|
|
if level == 0:
|
|
assert set(result.keys()) == {'id', 'app', 'collections', 'cluster', 'requests'}
|
|
assert set(result['collections'].keys()) == {'number_of_collections'}
|
|
return
|
|
else:
|
|
assert set(result.keys()) == {'id', 'app', 'collections', 'cluster', 'requests', 'memory', 'hardware'}
|
|
assert set(result['collections'].keys()) == {'number_of_collections', 'collections'}
|
|
|
|
last_queried = endpoint['200']['last_responded'].replace('Z', '+00:00')
|
|
last_queried = datetime.fromisoformat(last_queried)
|
|
# Assert today
|
|
assert last_queried.date() == datetime.now().date()
|
|
|
|
collection = result['collections']['collections'][0]
|
|
|
|
if level == 1:
|
|
assert set(collection.keys()) == {'vectors', 'optimizers_status', 'params'}
|
|
elif level == 2:
|
|
assert set(collection.keys()) == {'id', 'init_time_ms', 'config'}
|
|
elif level >= 3:
|
|
assert set(collection.keys()) == {'id', 'init_time_ms', 'config', 'shards', 'transfers', 'resharding'}
|
|
|
|
shard = collection['shards'][0]
|
|
assert set(shard.keys()) == {'id', 'key', 'local', 'remote', 'replicate_states', 'partial_snapshot'}
|
|
|
|
local_shard = shard['local']
|
|
|
|
if level == 3:
|
|
assert set(local_shard.keys()) == {
|
|
'variant_name', 'status', 'total_optimized_points', 'vectors_size_bytes',
|
|
'payloads_size_bytes', 'num_points', 'num_vectors', 'optimizations', 'async_scorer'
|
|
}
|
|
elif level >= 4:
|
|
assert set(local_shard.keys()) == {
|
|
'variant_name', 'status', 'total_optimized_points', 'vectors_size_bytes',
|
|
'payloads_size_bytes', 'num_points', 'num_vectors', 'segments', 'optimizations', 'async_scorer'
|
|
}
|
|
|
|
segment = local_shard['segments'][0]
|
|
assert set(segment.keys()) == {'info', 'config', 'vector_index_searches', 'payload_field_indices'}
|