mirror of
https://github.com/rustfs/rustfs.git
synced 2026-08-23 04:39:04 +00:00
fix(ecstore): align activation fence test imports
This commit is contained in:
@@ -443,7 +443,7 @@ pub mod rebalance {
|
||||
|
||||
#[cfg(feature = "test-util")]
|
||||
pub mod test_util {
|
||||
pub use crate::services::rebalance::entry::test_util::PausedRebalanceEntryTestFixture;
|
||||
pub use crate::services::rebalance::PausedRebalanceEntryTestFixture;
|
||||
pub use crate::services::rebalance::test_store_with_persisted_rebalance_meta;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -951,11 +951,10 @@ mod tests {
|
||||
TieredMetadataCommitBarrier,
|
||||
};
|
||||
use crate::storage_api_contracts::bucket::{BucketOperations as _, MakeBucketOptions};
|
||||
use crate::storage_api_contracts::multipart::MultipartOperations as _;
|
||||
use crate::storage_api_contracts::multipart::{CompletePart, MultipartOperations as _};
|
||||
use crate::storage_api_contracts::object::ObjectIO as _;
|
||||
use http::HeaderMap;
|
||||
use rustfs_filemeta::{FileInfo, FileMeta, ObjectPartInfo, TransitionVersionState};
|
||||
use rustfs_storage_api::CompletePart;
|
||||
use s3s::dto::{BucketLifecycleConfiguration, ExpirationStatus, LifecycleExpiration, LifecycleRule};
|
||||
use std::time::Duration as StdDuration;
|
||||
use time::OffsetDateTime;
|
||||
|
||||
@@ -47,6 +47,8 @@ mod runtime;
|
||||
mod types;
|
||||
mod worker;
|
||||
|
||||
#[cfg(feature = "test-util")]
|
||||
pub use entry::test_util::PausedRebalanceEntryTestFixture;
|
||||
pub(crate) use meta::is_rebalance_conflicting_with_decommission;
|
||||
pub use meta::{decode_rebalance_stop_propagation_record, encode_rebalance_stop_propagation_record};
|
||||
pub use types::{
|
||||
|
||||
@@ -1124,7 +1124,8 @@ mod rebalance_handler_tests {
|
||||
#[serial_test::serial]
|
||||
async fn real_admin_stop_cancels_paused_entry_before_waiting_for_activation_gate() {
|
||||
const REBALANCE_ID: &str = "admin-stop-paused-entry";
|
||||
let mut fixture = rustfs_ecstore::api::rebalance::test_util::PausedRebalanceEntryTestFixture::new(REBALANCE_ID).await;
|
||||
let mut fixture =
|
||||
crate::admin::storage_api::ecstore_rebalance::test_util::PausedRebalanceEntryTestFixture::new(REBALANCE_ID).await;
|
||||
fixture.wait_until_entry_paused().await;
|
||||
|
||||
let stop_store = fixture.store();
|
||||
@@ -1155,10 +1156,11 @@ mod rebalance_handler_tests {
|
||||
async fn real_admin_stop_accepts_same_run_terminalization_after_prepare() {
|
||||
const REBALANCE_ID: &str = "admin-stop-terminal-after-prepare";
|
||||
const REPLACEMENT_ID: &str = "admin-stop-replacement";
|
||||
let (_temp_dirs, store) = rustfs_ecstore::api::rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
let (_temp_dirs, store) =
|
||||
crate::admin::storage_api::ecstore_rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
let terminal_barrier = std::sync::Arc::new(tokio::sync::Barrier::new(2));
|
||||
let worker_barrier = std::sync::Arc::clone(&terminal_barrier);
|
||||
let worker_store = std::sync::Arc::clone(&store);
|
||||
@@ -1238,10 +1240,11 @@ mod rebalance_handler_tests {
|
||||
#[serial_test::serial]
|
||||
async fn real_admin_stop_loads_persisted_active_rebalance_from_cold_memory() {
|
||||
const REBALANCE_ID: &str = "admin-stop-cold-memory";
|
||||
let (_temp_dirs, store) = rustfs_ecstore::api::rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
let (_temp_dirs, store) =
|
||||
crate::admin::storage_api::ecstore_rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
*store.rebalance_meta.write().await = None;
|
||||
assert!(store.current_rebalance_id().await.is_none());
|
||||
|
||||
@@ -1271,10 +1274,11 @@ mod rebalance_handler_tests {
|
||||
async fn real_admin_stop_refreshes_persisted_active_over_stale_inactive_memory() {
|
||||
const PERSISTED_REBALANCE_ID: &str = "admin-stop-persisted-active";
|
||||
const STALE_REBALANCE_ID: &str = "admin-stop-stale-terminal";
|
||||
let (_temp_dirs, store) = rustfs_ecstore::api::rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(PERSISTED_REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
let (_temp_dirs, store) =
|
||||
crate::admin::storage_api::ecstore_rebalance::test_util::test_store_with_persisted_rebalance_meta(
|
||||
started_rebalance_meta(PERSISTED_REBALANCE_ID),
|
||||
)
|
||||
.await;
|
||||
*store.rebalance_meta.write().await = Some(RebalanceMeta {
|
||||
id: STALE_REBALANCE_ID.to_string(),
|
||||
stopped_at: Some(OffsetDateTime::now_utc()),
|
||||
|
||||
@@ -70,7 +70,9 @@ mod ecstore_notification {
|
||||
}
|
||||
|
||||
#[allow(unused_imports)]
|
||||
mod ecstore_rebalance {
|
||||
pub(crate) mod ecstore_rebalance {
|
||||
#[cfg(test)]
|
||||
pub(crate) use crate::storage::storage_api::ecstore_rebalance::test_util;
|
||||
pub(crate) use crate::storage::storage_api::ecstore_rebalance::{
|
||||
DiskStat, RebalSaveOpt, RebalStatus, RebalanceCleanupWarningEntry, RebalanceCleanupWarnings, RebalanceInfo,
|
||||
RebalanceMeta, RebalanceStats, RebalanceStopPropagationRecord, decode_rebalance_stop_propagation_record,
|
||||
|
||||
@@ -501,6 +501,8 @@ pub(crate) mod ecstore_notification {
|
||||
|
||||
#[allow(unused_imports)]
|
||||
pub(crate) mod ecstore_rebalance {
|
||||
#[cfg(test)]
|
||||
pub(crate) use rustfs_ecstore::api::rebalance::test_util;
|
||||
pub(crate) use rustfs_ecstore::api::rebalance::{
|
||||
DiskStat, RebalSaveOpt, RebalStatus, RebalanceCleanupWarningEntry, RebalanceCleanupWarnings, RebalanceInfo,
|
||||
RebalanceMeta, RebalanceStats, RebalanceStopPropagationRecord, decode_rebalance_stop_propagation_record,
|
||||
|
||||
Reference in New Issue
Block a user