1bee0fca05
Summary: Currently, `DeleteDir` only deletes the directory if there are no other directories under the target dir. This PR makes it delete directories recursively. Pull Request resolved: https://github.com/facebook/rocksdb/pull/6934 Test Plan: Added a new unit test in testutil_test.cc. `make testutil_test` Reviewed By: zhichao-cao Differential Revision: D21884211 Pulled By: cheng-chang fbshipit-source-id: 0b9a48a200f494ee007aef5d1763b4aa331f8b5a
43 lines
1.2 KiB
C++
43 lines
1.2 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#include "test_util/testutil.h"
|
|
|
|
#include "port/port.h"
|
|
#include "port/stack_trace.h"
|
|
#include "test_util/testharness.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
void CreateFile(Env* env, const std::string& path) {
|
|
std::unique_ptr<WritableFile> f;
|
|
ASSERT_OK(env->NewWritableFile(path, &f, EnvOptions()));
|
|
f->Close();
|
|
}
|
|
|
|
TEST(TestUtil, DestroyDirRecursively) {
|
|
auto env = Env::Default();
|
|
// test_util/file
|
|
// /dir
|
|
// /dir/file
|
|
std::string test_dir = test::PerThreadDBPath("test_util");
|
|
ASSERT_OK(env->CreateDir(test_dir));
|
|
CreateFile(env, test_dir + "/file");
|
|
ASSERT_OK(env->CreateDir(test_dir + "/dir"));
|
|
CreateFile(env, test_dir + "/dir/file");
|
|
|
|
ASSERT_OK(test::DestroyDir(env, test_dir));
|
|
auto s = env->FileExists(test_dir);
|
|
ASSERT_TRUE(s.IsNotFound());
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|
|
|
|
int main(int argc, char** argv) {
|
|
ROCKSDB_NAMESPACE::port::InstallStackTraceHandler();
|
|
::testing::InitGoogleTest(&argc, argv);
|
|
return RUN_ALL_TESTS();
|
|
}
|