i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-02 01:10:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Align rate limit monitoring test with group policy

Browse Source
This commit is contained in:
RWDai
2026-05-13 18:27:56 +08:00
parent 6a9017bfce
commit 87c0a915b8
1 changed files with 28 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
apps/aether-gateway/src/tests/frontdoor/public_support.rs
View File

@@ -30,7 +30,8 @@ use aether_data::repository::management_tokens::{
}; };
use aether_data::repository::usage::InMemoryUsageReadRepository; use aether_data::repository::usage::InMemoryUsageReadRepository;
use aether_data::repository::users::{ use aether_data::repository::users::{
InMemoryUserReadRepository, StoredUserAuthRecord, StoredUserExportRow, InMemoryUserReadRepository, StoredUserAuthRecord, StoredUserExportRow, UpsertUserGroupRecord,
UserReadRepository,
}; };
use aether_data::repository::wallet::{ use aether_data::repository::wallet::{
InMemoryWalletRepository, StoredWalletSnapshot, WalletWriteRepository, InMemoryWalletRepository, StoredWalletSnapshot, WalletWriteRepository,
@@ -2701,15 +2702,38 @@ async fn gateway_handles_user_monitoring_rate_limit_status_locally_without_proxy
.expect("auth api key snapshot should build") .expect("auth api key snapshot should build")
.with_user_rate_limit(Some(80)), .with_user_rate_limit(Some(80)),
)])); )]));
let user_repository: Arc<dyn UserReadRepository> =
Arc::new(InMemoryUserReadRepository::seed_auth_users(vec![
sample_auth_user(now),
]));
let group = user_repository
.create_user_group(UpsertUserGroupRecord {
name: "Monitoring Limits".to_string(),
description: None,
priority: 0,
allowed_providers: None,
allowed_providers_mode: "unrestricted".to_string(),
allowed_api_formats: None,
allowed_api_formats_mode: "unrestricted".to_string(),
allowed_models: None,
allowed_models_mode: "unrestricted".to_string(),
rate_limit: Some(80),
rate_limit_mode: "custom".to_string(),
})
.await
.expect("group should create")
.expect("group should exist");
user_repository
.add_user_to_group(&group.id, "user-auth-1")
.await
.expect("group membership should create");
let (gateway_url, upstream_hits, gateway_handle, upstream_handle) = let (gateway_url, upstream_hits, gateway_handle, upstream_handle) =
start_auth_gateway_with_builder(|| { start_auth_gateway_with_builder(|| {
let data_state = crate::data::GatewayDataState::with_auth_api_key_repository_for_tests( let data_state = crate::data::GatewayDataState::with_auth_api_key_repository_for_tests(
auth_repository, auth_repository,
) )
.with_user_reader(Arc::new(InMemoryUserReadRepository::seed_auth_users(vec![ .with_user_reader(Arc::clone(&user_repository));
sample_auth_user(now),
])));
AppState::new() AppState::new()
.expect("gateway should build") .expect("gateway should build")
.with_data_state_for_tests(data_state) .with_data_state_for_tests(data_state)