object_store/
test_util.rs

1// Copyright 2023 Greptime Team
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 crate::{ObjectStore, Result};
16
17/// Temp folder for object store test
18pub struct TempFolder {
19    store: ObjectStore,
20    // The path under root.
21    path: String,
22}
23
24impl TempFolder {
25    pub fn new(store: &ObjectStore, path: &str) -> Self {
26        Self {
27            store: store.clone(),
28            path: path.to_string(),
29        }
30    }
31
32    pub async fn remove_all(&self) -> Result<()> {
33        self.store.delete_with(&self.path).recursive(true).await
34    }
35}
object_store/test_util.rs

object_store/
test_util.rs
