litellm-rs 0.4.16

A high-performance AI Gateway written in Rust, providing OpenAI-compatible APIs with intelligent routing, load balancing, and enterprise features
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

//! Team module tests

use super::*;
use uuid::Uuid;

#[test]
fn test_team_creation() {
    let team = Team::new("test-team".to_string(), Some("Test Team".to_string()));

    assert_eq!(team.name, "test-team");
    assert_eq!(team.display_name, Some("Test Team".to_string()));
    assert!(team.is_active());
}

#[test]
fn test_team_usage_update() {
    let mut team = Team::new("test-team".to_string(), None);

    team.update_usage(10, 1000, 0.50);

    assert_eq!(team.usage_stats.total_requests, 10);
    assert_eq!(team.usage_stats.total_tokens, 1000);
    assert_eq!(team.usage_stats.total_cost, 0.50);
}

#[test]
fn test_team_member_permissions() {
    let team_id = Uuid::new_v4();
    let user_id = Uuid::new_v4();
    let mut member = TeamMember::new(team_id, user_id, TeamRole::Member, None);

    assert!(!member.has_permission("admin"));

    member.add_permission("admin".to_string());
    assert!(member.has_permission("admin"));

    member.remove_permission("admin");
    assert!(!member.has_permission("admin"));
}

#[test]
fn test_invitation_expiry() {
    let team_id = Uuid::new_v4();
    let invited_by = Uuid::new_v4();
    let expires_at = chrono::Utc::now() - chrono::Duration::hours(1); // Expired

    let invitation = TeamInvitation::new(
        team_id,
        "test@example.com".to_string(),
        TeamRole::Member,
        "token".to_string(),
        invited_by,
        expires_at,
    );

    assert!(invitation.is_expired());
}