composio-sdk 0.3.0

Minimal Rust SDK for Composio Tool Router REST API
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

//! Integration tests for session retrieval

use composio_sdk::client::ComposioClient;
use composio_sdk::error::ComposioError;

#[tokio::test]
async fn test_get_session_with_valid_id() {
    // This test requires a real API key and session ID
    // Skip if not available
    let api_key = std::env::var("COMPOSIO_API_KEY").unwrap_or_default();
    if api_key.is_empty() {
        eprintln!("Skipping test: COMPOSIO_API_KEY not set");
        return;
    }

    let client = ComposioClient::builder()
        .api_key(api_key)
        .build()
        .unwrap();

    // First create a session
    let created_session = client
        .create_session("test_user_retrieval")
        .send()
        .await
        .unwrap();

    let session_id = created_session.session_id().to_string();

    // Now retrieve it
    let retrieved_session = client.get_session(&session_id).await.unwrap();

    // Verify the session details match
    assert_eq!(retrieved_session.session_id(), session_id);
    assert!(!retrieved_session.mcp_url().is_empty());
    assert!(!retrieved_session.tools().is_empty());
}

#[tokio::test]
async fn test_get_session_with_invalid_id() {
    let api_key = std::env::var("COMPOSIO_API_KEY").unwrap_or_default();
    if api_key.is_empty() {
        eprintln!("Skipping test: COMPOSIO_API_KEY not set");
        return;
    }

    let client = ComposioClient::builder()
        .api_key(api_key)
        .build()
        .unwrap();

    // Try to retrieve a non-existent session
    let result = client.get_session("invalid_session_id").await;

    // Should return an error
    assert!(result.is_err());
    
    // Check if it's an ApiError with 404 status
    match result {
        Err(ComposioError::ApiError { status, .. }) => {
            assert_eq!(status, 404);
        }
        _ => panic!("Expected ApiError with 404 status"),
    }
}

#[tokio::test]
async fn test_get_session_accepts_string() {
    let api_key = std::env::var("COMPOSIO_API_KEY").unwrap_or_default();
    if api_key.is_empty() {
        eprintln!("Skipping test: COMPOSIO_API_KEY not set");
        return;
    }

    let client = ComposioClient::builder()
        .api_key(api_key)
        .build()
        .unwrap();

    // Create a session
    let created_session = client
        .create_session("test_user_string")
        .send()
        .await
        .unwrap();

    let session_id = created_session.session_id().to_string();

    // Test with String
    let _retrieved = client.get_session(session_id.clone()).await.unwrap();

    // Test with &str
    let _retrieved = client.get_session(session_id.as_str()).await.unwrap();
}

#[tokio::test]
async fn test_get_session_retry_on_transient_errors() {
    // This test would require a mock server to simulate transient errors
    // For now, we'll just verify the method signature compiles
    let client = ComposioClient::builder()
        .api_key("test_key")
        .max_retries(5)
        .build()
        .unwrap();

    // The retry logic is tested in the retry module
    // This just verifies the integration
    let _ = client.get_session("test_id").await;
}