mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-31 07:00:56 -05:00
0b0df145b3
* Skip broken tests
* Add rocksdb dropper
This is a temporary tool. It will be removed later in this PR.
* [automated] Drop rocksdb
This commit is made by running tools/rocksdb/drop.sh
* Touch-up after ast-grep
The previous automated commit removed items, but not their comments.
Also, some blocks are left with only one item.
Also, ast-grep can't handle macros like `vec![]`.
This commit completes the job.
* Remove RocksDB dropper
* Remove mentions of rocksdb feature in CI
* Fix clippy warnings
These `FIXME` comments added previously in this PR by ast-grep by
"peeling" cfg_attr like this:
-#[cfg_attr(not(feature = "rocksdb"), expect(...))]
+#[expect(...)] // FIXME(rocksdb): ...
This commit removes these allow/expect attributes and fixes clippy
lints.
* Remove leftover rocksdb-related code
Removed:
- Cargo.toml: rocksdb cargo feature flag and dependencies.
- code: rocksdb-related items that was not under feature flag.
- flags.rs: rocksdb-related qdrant feature flags.
Disabled:
- Some benchmarks because these do not compile now.
* Print warning if rocksdb leftovers found in snapshots
* Remove Clone from tokenizer
* Regenerate openapi.json
75 lines
3.2 KiB
Python
75 lines
3.2 KiB
Python
import pytest
|
|
import uuid
|
|
|
|
from e2e_tests.client_utils import ClientUtils
|
|
from e2e_tests.models import QdrantContainerConfig
|
|
|
|
# TODO(rocksdb): fix this test - storage.tar.xz contains rocksdb data
|
|
pytest.skip("disabled until low-ram test data is updated", allow_module_level=True)
|
|
|
|
|
|
class TestLowRam:
|
|
"""Test Qdrant behavior under low RAM conditions and recovery mode."""
|
|
|
|
@pytest.mark.parametrize("storage_from_archive", ["storage.tar.xz"], indirect=True)
|
|
def test_oom_with_insufficient_memory(self, qdrant_container_factory, storage_from_archive):
|
|
"""
|
|
Test that Qdrant OOMs when started with insufficient memory.
|
|
"""
|
|
print("Starting container with 128MB memory limit - expecting OOM...")
|
|
unique_suffix = str(uuid.uuid4())[:8]
|
|
config = QdrantContainerConfig(
|
|
name=f"qdrant-oom-{unique_suffix}",
|
|
mem_limit="128m",
|
|
volumes={str(storage_from_archive): {'bind': '/qdrant/storage', 'mode': 'rw'}},
|
|
environment={"QDRANT__STORAGE__HANDLE_COLLECTION_LOAD_ERRORS": "true"},
|
|
remove=False,
|
|
exit_on_error=False # Don't raise error when Qdrant fails to start
|
|
)
|
|
oom_container_info = qdrant_container_factory(config)
|
|
|
|
oom_container = oom_container_info.container
|
|
|
|
exit_code = oom_container.wait()['StatusCode']
|
|
|
|
assert exit_code == 137, f"Expected exit code 137 (OOM), but got {exit_code}"
|
|
|
|
@pytest.mark.parametrize("storage_from_archive", ["storage.tar.xz"], indirect=True)
|
|
def test_recovery_mode_with_low_memory(self, qdrant_container_factory, storage_from_archive):
|
|
"""
|
|
Test that:
|
|
1. Qdrant can start in recovery mode with low memory limit
|
|
2. Collection loads as a dummy shard in recovery mode
|
|
"""
|
|
print("Starting container in recovery mode...")
|
|
config = QdrantContainerConfig(
|
|
mem_limit="128m",
|
|
volumes={str(storage_from_archive): {'bind': '/qdrant/storage', 'mode': 'rw'}},
|
|
environment={
|
|
"QDRANT__STORAGE__HANDLE_COLLECTION_LOAD_ERRORS": "true",
|
|
"QDRANT_ALLOW_RECOVERY_MODE": "true"
|
|
}
|
|
)
|
|
recovery_container_info = qdrant_container_factory(config)
|
|
|
|
recovery_container = recovery_container_info.container
|
|
api_port = recovery_container_info.http_port
|
|
|
|
client = ClientUtils(host=recovery_container_info.host, port=api_port)
|
|
|
|
assert client.wait_for_collection_loaded("low-ram"), "Collection failed to load in ~10 seconds"
|
|
|
|
# Check for recovery mode message in logs
|
|
logs = recovery_container.logs().decode('utf-8')
|
|
recovery_msg = "Qdrant is loaded in recovery mode"
|
|
assert recovery_msg in logs, f"'{recovery_msg}' log message not found in container logs"
|
|
|
|
# Check that collection info returns 500 (dummy shard)
|
|
try:
|
|
client.get_collection_info_dict("low-ram")
|
|
assert False, "Expected collection info to fail with Out-of-Memory error"
|
|
except Exception as e:
|
|
error_text = str(e)
|
|
expected_error = "Out-of-Memory"
|
|
assert expected_error in error_text, f"'{expected_error}' not found in error: {error_text}"
|