mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-23 11:11:00 -05:00
8fb203eeb5
* refactor(SegmentOptimizer::name): make it `&'static`
* refactor(TrackerStatus): add is_running()
* refactor: add TrackerSegmentInfo
* feat: rework optimizations endpoint
* feat(collections/{}/optimizations): trim spaces
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
import pytest
|
|
|
|
from .helpers.collection_setup import basic_collection_setup, drop_collection
|
|
from .helpers.helpers import request_with_validation
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup(collection_name):
|
|
basic_collection_setup(collection_name=collection_name)
|
|
yield
|
|
drop_collection(collection_name=collection_name)
|
|
|
|
|
|
def test_optimizations(collection_name):
|
|
response = request_with_validation(
|
|
api="/collections/{collection_name}/optimizations",
|
|
method="GET",
|
|
path_params={"collection_name": collection_name},
|
|
)
|
|
assert response.ok
|
|
result = response.json()["result"]
|
|
assert result["running"] == []
|
|
assert "queued" not in result
|
|
assert "completed" not in result
|
|
assert "idle_segments" not in result
|
|
|
|
|
|
def test_optimizations_with_all(collection_name):
|
|
response = request_with_validation(
|
|
api="/collections/{collection_name}/optimizations",
|
|
method="GET",
|
|
path_params={"collection_name": collection_name},
|
|
query_params={"with": "queued,completed,idle_segments"},
|
|
)
|
|
assert response.ok
|
|
result = response.json()["result"]
|
|
assert result["running"] == []
|
|
assert result["queued"] == []
|
|
assert result["completed"] == []
|
|
assert len(result["idle_segments"]) > 0
|
|
|
|
|
|
def test_optimizations_with_all_spaces(collection_name):
|
|
response = request_with_validation(
|
|
api="/collections/{collection_name}/optimizations",
|
|
method="GET",
|
|
path_params={"collection_name": collection_name},
|
|
query_params={"with": "queued , completed , idle_segments"},
|
|
)
|
|
assert response.ok
|
|
result = response.json()["result"]
|
|
assert result["running"] == []
|
|
assert result["queued"] == []
|
|
assert result["completed"] == []
|
|
assert len(result["idle_segments"]) > 0
|