risingwave_hummock_test/
local_state_store_test_utils.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use std::future::Future;
16
17use risingwave_common::util::epoch::EpochPair;
18use risingwave_storage::error::StorageResult;
19use risingwave_storage::store::{InitOptions, LocalStateStore};
20
21pub trait LocalStateStoreTestExt: LocalStateStore {
22    fn init_for_test(&mut self, epoch: u64) -> impl Future<Output = StorageResult<()>> + Send + '_ {
23        self.init(InitOptions::new(EpochPair::new_test_epoch(epoch)))
24    }
25
26    fn init_for_test_with_prev_epoch(
27        &mut self,
28        epoch: u64,
29        prev_epoch: u64,
30    ) -> impl Future<Output = StorageResult<()>> + Send + '_ {
31        self.init(InitOptions::new(EpochPair::new(epoch, prev_epoch)))
32    }
33}
34impl<T: LocalStateStore> LocalStateStoreTestExt for T {}