style: use imported types in region request tests

Signed-off-by: WenyXu <wenymedia@gmail.com>
This commit is contained in:
WenyXu
2026-10-12 13:11:23 +00:00
parent fc0933e50f
commit 1627e6c06b
2 changed files with 8 additions and 7 deletions
+6 -6
View File
@@ -655,7 +655,7 @@ mod tests {
let error = requester.handle(Arc::clone(&shared)).await.unwrap_err();
assert_eq!(error.retry_hint(), RetryHint::Retryable);
assert!(
matches!(error, error::Error::RegionServer { addr, code: tonic::Code::Unavailable, .. } if addr == server.addr)
matches!(error, Error::RegionServer { addr, code: tonic::Code::Unavailable, .. } if addr == server.addr)
);
requester.handle(Arc::clone(&shared)).await.unwrap();
let requests = server.service.requests.lock().unwrap();
@@ -699,7 +699,7 @@ mod tests {
hint.parse().unwrap(),
);
}
let error: error::Error = status.into();
let error: Error = status.into();
assert_eq!(error.retry_hint(), expected_hint, "{name}");
assert_eq!(error.status_code(), StatusCode::RegionNotReady, "{name}");
assert_eq!(
@@ -743,7 +743,7 @@ mod tests {
))
.await
.unwrap();
let baseline: error::Error = baseline
let baseline: Error = baseline
.handle(build_insert_request())
.await
.unwrap_err()
@@ -792,11 +792,11 @@ mod tests {
let shared = Arc::new(build_insert_request());
assert!(matches!(
requester.handle(Arc::clone(&shared)).await.unwrap_err(),
error::Error::IllegalDatabaseResponse { .. }
Error::IllegalDatabaseResponse { .. }
));
assert!(matches!(
requester.handle(shared).await.unwrap_err(),
error::Error::Server {
Error::Server {
code: StatusCode::RegionNotFound,
..
}
@@ -822,7 +822,7 @@ mod tests {
.unwrap()
.max_encoding_message_size(config.max_send_message_size.as_bytes() as usize)
.max_decoding_message_size(config.max_recv_message_size.as_bytes() as usize);
let baseline: error::Error = baseline
let baseline: Error = baseline
.handle(build_insert_request())
.await
.unwrap_err()
+2 -1
View File
@@ -1734,6 +1734,7 @@ mod tests {
use snafu::IntoError;
use super::*;
use crate::region::RegionLeaderState;
#[test]
fn test_partition_version_mismatch_retry_hint() {
@@ -1750,7 +1751,7 @@ mod tests {
#[test]
fn test_region_state_retry_hint() {
let region_id = RegionId::new(1, 1);
let writable = RegionRoleState::Leader(crate::region::RegionLeaderState::Writable);
let writable = RegionRoleState::Leader(RegionLeaderState::Writable);
let follower_error = RegionStateSnafu {
region_id,
state: RegionRoleState::Follower,