diff --git a/src/operator/src/req_convert/common/partitioner.rs b/src/operator/src/req_convert/common/partitioner.rs index d02f9acf112..044d3b97db5 100644 --- a/src/operator/src/req_convert/common/partitioner.rs +++ b/src/operator/src/req_convert/common/partitioner.rs @@ -195,6 +195,7 @@ mod tests { use moka::future::Cache; use super::*; + use crate::error::Error; use crate::test_util; #[tokio::test] @@ -240,8 +241,8 @@ mod tests { .await .unwrap_err(); - assert!(matches!(insert_error, error::Error::SplitInsert { .. })); - assert!(matches!(delete_error, error::Error::SplitDelete { .. })); + assert!(matches!(insert_error, Error::SplitInsert { .. })); + assert!(matches!(delete_error, Error::SplitDelete { .. })); for error in [insert_error, delete_error] { assert_eq!(error.status_code(), expected_status); assert_eq!(error.retry_hint(), expected_hint); diff --git a/src/operator/src/write_dispatcher/tests.rs b/src/operator/src/write_dispatcher/tests.rs index 8e5c59f8b68..1191f73d55a 100644 --- a/src/operator/src/write_dispatcher/tests.rs +++ b/src/operator/src/write_dispatcher/tests.rs @@ -30,6 +30,7 @@ use common_error::ext::ErrorExt; use common_grpc::channel_manager::ChannelConfig; use common_meta::cache as meta_cache; use common_meta::ddl::test_util::datanode_handler::DatanodeWatcher; +use common_meta::key::TableMetadataManager; use common_meta::key::table_route::TableRouteValue; use common_meta::test_util::{MockDatanodeHandler, MockDatanodeManager}; use common_query::request::QueryRequest; @@ -60,7 +61,7 @@ fn build_insert_request(regions: &[u32]) -> RegionRequest { async fn test_resolve_peers_batches_logical_and_physical_regions() { let backend = test_util::prepare_mocked_backend().await; let partition_manager = test_util::create_partition_rule_manager(backend.clone()).await; - common_meta::key::TableMetadataManager::new(backend) + TableMetadataManager::new(backend) .create_table_metadata( test_util::new_test_table_info(42, "dispatcher_logical", [1, 2, 3].into_iter()), TableRouteValue::logical(1), @@ -101,7 +102,7 @@ async fn test_resolve_peers_batches_logical_and_physical_regions() { async fn test_dispatch_preserves_write_payloads_and_envelopes() { let backend = test_util::prepare_mocked_backend().await; let partition_manager = test_util::create_partition_rule_manager(backend.clone()).await; - common_meta::key::TableMetadataManager::new(backend) + TableMetadataManager::new(backend) .create_table_metadata( test_util::new_test_table_info(42, "dispatcher_logical", [1, 2, 3].into_iter()), TableRouteValue::logical(1), @@ -276,7 +277,7 @@ async fn test_resolve_peers_cache_lookup_ablation() { common_telemetry::init_default_ut_logging(); let backend = test_util::prepare_mocked_backend().await; test_util::create_partition_rule_manager(backend.clone()).await; - common_meta::key::TableMetadataManager::new(backend.clone()) + TableMetadataManager::new(backend.clone()) .create_table_metadata( test_util::new_test_table_info(42, "resolution_ablation", [1, 2, 3].into_iter()), TableRouteValue::logical(1),