rust/greptimedb
1
0
Fork 0
mirror of https://github.com/GreptimeTeam/greptimedb.git synced 2026-10-12 23:12:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

style(operator): use imported types in partition tests

Browse Source 
Signed-off-by: WenyXu <wenymedia@gmail.com>
This commit is contained in:
WenyXu
2026-10-12 13:12:10 +00:00
parent b90f5604b7
commit f7b72c0aef
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/operator/src/req_convert/common/partitioner.rs
+3 -2
View File
@@ -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);
src/operator/src/write_dispatcher/tests.rs
+4 -3
View File
@@ -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),