udb 0.2.0

Universal Data Broker — a Rust gRPC broker over multiple databases (Postgres, MySQL, SQLite, MongoDB, ClickHouse, Cassandra, MSSQL, Redis, Qdrant, S3, Neo4j, …) with per-tenant RLS, 2PC, sagas, and CDC.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

from __future__ import annotations

from udb.entity.v1 import types_pb2
from udb_client import Metadata, to_record_json, to_struct


def test_metadata_headers_include_required_and_routing_values() -> None:
    meta = Metadata(
        tenant_id="tenant-1",
        user_id="user-1",
        purpose="billing.test",
        correlation_id="corr-1",
        scopes=("udb:read", "udb:write"),
        service_identity="pytest",
        project_id="billing",
        target_backend="postgres",
        primary_read=True,
    )

    headers = dict(meta.to_grpc_metadata())

    assert headers["x-tenant-id"] == "tenant-1"
    assert headers["x-scopes"] == "udb:read,udb:write"
    assert headers["x-udb-project-id"] == "billing"
    assert headers["x-udb-target-backend"] == "postgres"
    assert headers["x-udb-primary-read"] == "true"


def test_metadata_builds_request_context() -> None:
    meta = Metadata(
        tenant_id="tenant-1",
        purpose="billing.test",
        correlation_id="corr-1",
        scopes=("udb:read",),
        service_identity="pytest",
    )

    context = meta.to_request_context()

    assert context.tenant_id == "tenant-1"
    assert context.scopes == ["udb:read"]
    assert context.service_identity == "pytest"


def test_helpers_build_protobuf_payloads() -> None:
    assert to_struct({"status": "paid"}).fields["status"].string_value == "paid"
    assert to_record_json({"b": 2, "a": 1}) == b'{"a":1,"b":2}'

    request = types_pb2.SelectRequest(filter=to_struct({"customer_id": "cus_1"}))
    assert request.filter.fields["customer_id"].string_value == "cus_1"