mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-06 10:00:58 -05:00
fc77e3e511
* Use systematically assert_http_ok for better error reporting * extraction assertion to use it outside of pytest
56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
import pathlib
|
|
|
|
from .utils import *
|
|
from .assertions import assert_http_ok
|
|
|
|
N_PEERS = 5
|
|
N_COLLECTIONS = 20
|
|
|
|
|
|
def test_many_collections(tmp_path: pathlib.Path):
|
|
assert_project_root()
|
|
peer_dirs = make_peer_folders(tmp_path, N_PEERS)
|
|
|
|
# Gathers REST API uris
|
|
peer_api_uris = []
|
|
|
|
# Start bootstrap
|
|
(bootstrap_api_uri, bootstrap_uri) = start_first_peer(
|
|
peer_dirs[0], "peer_0_0.log")
|
|
peer_api_uris.append(bootstrap_api_uri)
|
|
|
|
# Wait for leader
|
|
leader = wait_peer_added(bootstrap_api_uri)
|
|
|
|
# Start other peers
|
|
for i in range(1, len(peer_dirs)):
|
|
peer_api_uris.append(start_peer(
|
|
peer_dirs[i], f"peer_0_{i}.log", bootstrap_uri))
|
|
|
|
# Wait for cluster
|
|
wait_for_uniform_cluster_status(peer_api_uris, leader)
|
|
|
|
# Check that there are no collections on all peers
|
|
for uri in peer_api_uris:
|
|
r = requests.get(f"{uri}/collections")
|
|
assert_http_ok(r)
|
|
assert len(r.json()["result"]["collections"]) == 0
|
|
|
|
# Create N_COLLECTIONS on different peers
|
|
for i in range(1, N_COLLECTIONS + 1):
|
|
peer = peer_api_uris[i % N_PEERS]
|
|
print(f"creating test_collection_{i} on {peer}")
|
|
r = requests.put(
|
|
f"{peer}/collections/test_collection_{i}", json={
|
|
"vectors": {
|
|
"size": 4,
|
|
"distance": "Dot"
|
|
},
|
|
"shard_number": 1 # single shard
|
|
})
|
|
assert_http_ok(r)
|
|
|
|
# Check that all collections exist on all peers
|
|
for i in range(1, N_COLLECTIONS + 1):
|
|
wait_collection_exists_and_active_on_all_peers(collection_name=f"test_collection_{i}", peer_api_uris=peer_api_uris)
|