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

fix(users): preserve nullable rate limit updates

Browse Source
This commit is contained in:
RWDai
2026-05-07 19:16:49 +08:00
parent f3d9523502
commit a44667d2a9
5 changed files with 58 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/aether-data/src/repository/users/memory.rs
+47 -3
View File
@@ -882,6 +882,7 @@ impl UserReadRepository for InMemoryUserReadRepository {
allowed_api_formats: Option<Vec<String>>,
allowed_models_present: bool,
allowed_models: Option<Vec<String>>,
rate_limit_present: bool,
rate_limit: Option<i32>,
is_active: Option<bool>,
) -> Result<Option<StoredUserAuthRecord>, DataLayerError> {
@@ -931,8 +932,8 @@ impl UserReadRepository for InMemoryUserReadRepository {
row.allowed_providers = updated.allowed_providers.clone();
row.allowed_api_formats = updated.allowed_api_formats.clone();
row.allowed_models = updated.allowed_models.clone();
if let Some(rate_limit) = rate_limit {
row.rate_limit = Some(rate_limit);
if rate_limit_present {
row.rate_limit = rate_limit;
}
row.is_active = updated.is_active;
}
@@ -1518,7 +1519,24 @@ mod tests {
None,
)
.expect("auth user should build");
let repository = InMemoryUserReadRepository::seed_auth_users(vec![user]);
let export_user = StoredUserExportRow::new(
"user-1".to_string(),
Some("[email protected]".to_string()),
true,
"alice".to_string(),
Some("old-hash".to_string()),
"user".to_string(),
"local".to_string(),
None,
None,
None,
Some(10),
None,
true,
)
.expect("export user should build");
let repository = InMemoryUserReadRepository::seed_auth_users(vec![user])
.with_export_users([export_user]);
let updated = repository
.update_local_auth_user_profile(
@@ -1566,6 +1584,7 @@ mod tests {
None,
true,
Some(vec!["gpt-4.1".to_string()]),
true,
Some(50),
Some(false),
)
@@ -1583,6 +1602,31 @@ mod tests {
Some(vec!["gpt-4.1".to_string()])
);
assert!(!admin_updated.is_active);
assert_eq!(
repository
.find_export_user_by_id("user-1")
.await
.expect("export lookup should succeed")
.expect("export row should exist")
.rate_limit,
Some(50)
);
repository
.update_local_auth_user_admin_fields(
"user-1", None, false, None, false, None, false, None, true, None, None,
)
.await
.expect("rate limit clear should succeed")
.expect("rate limit clear should return user");
assert_eq!(
repository
.find_export_user_by_id("user-1")
.await
.expect("export lookup should succeed")
.expect("export row should exist")
.rate_limit,
None
);
assert_eq!(
repository
.update_user_model_capability_settings(