udb 0.2.1

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

from __future__ import annotations

import pytest

from udb.entity.v1 import types_pb2
from udb_client import Metadata, UdbClient
from udb_client.exceptions import UdbConfigurationError


def test_client_injects_context_without_mutating_original_request() -> None:
    client = UdbClient(
        "unused",
        Metadata(
            tenant_id="tenant-1",
            purpose="billing.test",
            correlation_id="corr-1",
            scopes=("udb:read",),
        ),
    )
    request = types_pb2.SelectRequest(message_type="acme.billing.v1.Customer")

    cloned = client._with_context(request, None)

    assert request.context.tenant_id == ""
    assert cloned.context.tenant_id == "tenant-1"
    assert cloned.context.scopes == ["udb:read"]
    client.close()


def test_client_requires_metadata_before_rpc_context_injection() -> None:
    client = UdbClient("unused")

    with pytest.raises(UdbConfigurationError):
        client._with_context(types_pb2.SelectRequest(), None)

    client.close()