mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-31 15:11:35 -05:00
53cd2a4e8f
* Impletement a special type of access to collection suitable for serverless * fix test * fix test
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
import random
|
|
import string
|
|
|
|
import jwt
|
|
from consensus_tests.utils import start_cluster
|
|
|
|
PORT_SEED = 10000
|
|
REST_URI = f"http://127.0.0.1:{PORT_SEED + 2}"
|
|
GRPC_URI = f"127.0.0.1:{PORT_SEED + 1}"
|
|
|
|
SECRET = "my_top_secret_key"
|
|
|
|
READ_ONLY_API_KEY = "boo-hoo, this can only read!"
|
|
|
|
API_KEY_HEADERS = {"Api-Key": SECRET}
|
|
API_KEY_METADATA = [("api-key", SECRET)]
|
|
READ_ONLY_API_KEY_METADATA = [("api-key", READ_ONLY_API_KEY)]
|
|
|
|
|
|
def start_jwt_protected_cluster(tmp_path, num_peers=1):
|
|
extra_env = {
|
|
"QDRANT__SERVICE__API_KEY": SECRET,
|
|
"QDRANT__SERVICE__READ_ONLY_API_KEY": READ_ONLY_API_KEY,
|
|
"QDRANT__SERVICE__JWT_RBAC": "true",
|
|
"QDRANT__STORAGE__WAL__WAL_CAPACITY_MB": "1", # to speed up snapshot tests
|
|
}
|
|
|
|
peer_api_uris, peer_dirs, bootstrap_uri = start_cluster(
|
|
tmp_path,
|
|
num_peers=num_peers,
|
|
port_seed=PORT_SEED,
|
|
extra_env=extra_env,
|
|
headers=API_KEY_HEADERS,
|
|
)
|
|
|
|
assert REST_URI in peer_api_uris
|
|
|
|
return peer_api_uris, peer_dirs, bootstrap_uri
|
|
|
|
|
|
def encode_jwt(claims: dict, secret: str) -> str:
|
|
return jwt.encode(claims, secret, algorithm="HS256")
|
|
|
|
def decode_jwt(token: str, secret: str) -> dict:
|
|
return jwt.decode(token, secret, algorithms=["HS256"])
|
|
|
|
def random_str():
|
|
return "".join(random.choices(string.ascii_lowercase, k=10))
|