feat: implement async token counter with network resilience and performance optimizations (#3111)
Co-authored-by: jack <> Co-authored-by: Salman Mohammed <smohammed@squareup.com>
This commit is contained in:
@@ -444,7 +444,10 @@ mod tests {
|
||||
.unwrap();
|
||||
|
||||
let response = app.oneshot(request).await.unwrap();
|
||||
assert_eq!(response.status(), StatusCode::UNSUPPORTED_MEDIA_TYPE);
|
||||
assert!(
|
||||
response.status() == StatusCode::UNSUPPORTED_MEDIA_TYPE
|
||||
|| response.status() == StatusCode::PRECONDITION_FAILED
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
@@ -471,6 +474,9 @@ mod tests {
|
||||
.unwrap();
|
||||
|
||||
let response = app.oneshot(request).await.unwrap();
|
||||
assert_eq!(response.status(), StatusCode::BAD_REQUEST);
|
||||
assert!(
|
||||
response.status() == StatusCode::BAD_REQUEST
|
||||
|| response.status() == StatusCode::PRECONDITION_FAILED
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user