dakera-client 0.11.54

Rust client SDK for Dakera AI Agent Memory Platform
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

//! Integration tests for SEC-3 AES-256-GCM encryption key rotation endpoint.
//!
//! Tests `DakeraClient::rotate_encryption_key()` — `POST /v1/admin/encryption/rotate-key`.

use dakera_client::DakeraClient;

#[tokio::test]
async fn test_rotate_encryption_key_returns_response() {
    let mut server = mockito::Server::new_async().await;
    let mock = server
        .mock("POST", "/v1/admin/encryption/rotate-key")
        .with_status(200)
        .with_header("content-type", "application/json")
        .with_body(r#"{"rotated":42,"skipped":3,"namespaces":["ns-a","ns-b"]}"#)
        .create_async()
        .await;

    let client = DakeraClient::new(server.url()).unwrap();
    let result = client
        .rotate_encryption_key("new-secret-passphrase", None)
        .await
        .unwrap();

    assert_eq!(result.rotated, 42);
    assert_eq!(result.skipped, 3);
    assert_eq!(result.namespaces, vec!["ns-a", "ns-b"]);
    mock.assert_async().await;
}

#[tokio::test]
async fn test_rotate_encryption_key_with_namespace() {
    let mut server = mockito::Server::new_async().await;
    let mock = server
        .mock("POST", "/v1/admin/encryption/rotate-key")
        .with_status(200)
        .with_header("content-type", "application/json")
        .with_body(r#"{"rotated":5,"skipped":0,"namespaces":["my-ns"]}"#)
        .create_async()
        .await;

    let client = DakeraClient::new(server.url()).unwrap();
    let result = client
        .rotate_encryption_key("new-key", Some("my-ns"))
        .await
        .unwrap();

    assert_eq!(result.rotated, 5);
    assert_eq!(result.namespaces, vec!["my-ns"]);
    mock.assert_async().await;
}

#[tokio::test]
async fn test_rotate_encryption_key_empty_response_defaults() {
    let mut server = mockito::Server::new_async().await;
    let _mock = server
        .mock("POST", "/v1/admin/encryption/rotate-key")
        .with_status(200)
        .with_header("content-type", "application/json")
        .with_body(r#"{"rotated":0,"skipped":0,"namespaces":[]}"#)
        .create_async()
        .await;

    let client = DakeraClient::new(server.url()).unwrap();
    let result = client.rotate_encryption_key("any-key", None).await.unwrap();

    assert_eq!(result.rotated, 0);
    assert_eq!(result.skipped, 0);
    assert!(result.namespaces.is_empty());
}