Fix error message (#4204)
Co-authored-by: Ahmed Ibrahim <aibrahim@openai.com>
This commit is contained in:
@@ -384,12 +384,12 @@ fn rate_limit_warnings_emit_thresholds() {
|
||||
let mut state = RateLimitWarningState::default();
|
||||
let mut warnings: Vec<String> = Vec::new();
|
||||
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(55.0)));
|
||||
warnings.extend(state.take_warnings(Some(55.0), Some(10.0)));
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(80.0)));
|
||||
warnings.extend(state.take_warnings(Some(80.0), Some(10.0)));
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(95.0)));
|
||||
warnings.extend(state.take_warnings(Some(95.0), Some(10.0)));
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(10079), Some(55.0), Some(299)));
|
||||
warnings.extend(state.take_warnings(Some(55.0), Some(10081), Some(10.0), Some(299)));
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(10081), Some(80.0), Some(299)));
|
||||
warnings.extend(state.take_warnings(Some(80.0), Some(10081), Some(10.0), Some(299)));
|
||||
warnings.extend(state.take_warnings(Some(10.0), Some(10081), Some(95.0), Some(299)));
|
||||
warnings.extend(state.take_warnings(Some(95.0), Some(10079), Some(10.0), Some(299)));
|
||||
|
||||
assert_eq!(
|
||||
warnings,
|
||||
@@ -411,6 +411,21 @@ fn rate_limit_warnings_emit_thresholds() {
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_rate_limit_warnings_monthly() {
|
||||
let mut state = RateLimitWarningState::default();
|
||||
let mut warnings: Vec<String> = Vec::new();
|
||||
|
||||
warnings.extend(state.take_warnings(Some(75.0), Some(43199), None, None));
|
||||
assert_eq!(
|
||||
warnings,
|
||||
vec![String::from(
|
||||
"Heads up, you've used over 75% of your monthly limit. Run /status for a breakdown.",
|
||||
),],
|
||||
"expected one warning per limit for the highest crossed threshold"
|
||||
);
|
||||
}
|
||||
|
||||
// (removed experimental resize snapshot test)
|
||||
|
||||
#[test]
|
||||
|
||||
Reference in New Issue
Block a user