#include "fts.h"
#include "dirut.h"
#include "tempdir.h"
#include <library/cpp/testing/unittest/registar.h>
#include <library/cpp/threading/future/async.h>
#include <util/system/file.h>
#include <util/system/tempfile.h>
#include <util/generic/string.h>
class TFtsTest: public TTestBase {
UNIT_TEST_SUITE(TFtsTest);
UNIT_TEST(TestSimple);
UNIT_TEST(TestNoLeakChangingAccessToFolder);
UNIT_TEST_SUITE_END();
public:
void TestSimple();
void TestNoLeakChangingAccessToFolder();
};
void MakeFile(const char* path) {
TFile(path, CreateAlways);
}
//There potentially could be problems in listing order on different platforms
int FtsCmp(const FTSENT** ent1, const FTSENT** ent2) {
return strcmp((*ent1)->fts_accpath, (*ent2)->fts_accpath);
}
void CheckEnt(FTSENT* ent, const char* name, int type) {
UNIT_ASSERT(ent);
UNIT_ASSERT_STRINGS_EQUAL(ent->fts_path, name);
UNIT_ASSERT_EQUAL(ent->fts_info, type);
}
class TFileTree {
public:
TFileTree(char* const* argv, int options, int (*compar)(const FTSENT**, const FTSENT**)) {
Fts_ = yfts_open(argv, options, compar);
}
~TFileTree() {
yfts_close(Fts_);
}
FTS* operator()() {
return Fts_;
}
private:
FTS* Fts_;
};
void TFtsTest::TestSimple() {
const char* dotPath[2] = {"." LOCSLASH_S, nullptr};
TFileTree currentDirTree((char* const*)dotPath, 0, FtsCmp);
UNIT_ASSERT(currentDirTree());
TTempDir tempDir = MakeTempName(yfts_read(currentDirTree())->fts_path);
MakeDirIfNotExist(tempDir().data());
MakeDirIfNotExist((tempDir() + LOCSLASH_S "dir1").data());
MakeFile((tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file1").data());
MakeFile((tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file2").data());
MakeDirIfNotExist((tempDir() + LOCSLASH_S "dir2").data());
MakeFile((tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file3").data());
MakeFile((tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file4").data());
const char* path[2] = {tempDir().data(), nullptr};
TFileTree fileTree((char* const*)path, 0, FtsCmp);
UNIT_ASSERT(fileTree());
CheckEnt(yfts_read(fileTree()), tempDir().data(), FTS_D);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1").data(), FTS_D);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file1").data(), FTS_F);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1" LOCSLASH_S "file2").data(), FTS_F);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir1").data(), FTS_DP);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2").data(), FTS_D);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file3").data(), FTS_F);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2" LOCSLASH_S "file4").data(), FTS_F);
CheckEnt(yfts_read(fileTree()), (tempDir() + LOCSLASH_S "dir2").data(), FTS_DP);
CheckEnt(yfts_read(fileTree()), (tempDir()).data(), FTS_DP);
UNIT_ASSERT_EQUAL(yfts_read(fileTree()), nullptr);
}
class TTempDirWithLostAccess: public TTempDir {
public:
~TTempDirWithLostAccess() {
chmod(Name().data(), 0777);
}
};
// https://st.yandex-team.ru/YQ-318
// Test that detects memory leak in case of error in chdir in fts_build function.
void TFtsTest::TestNoLeakChangingAccessToFolder() {
TTempDirWithLostAccess tempDir;
TString tmpPath = tempDir();
if (tmpPath.EndsWith(LOCSLASH_S)) {
tmpPath.resize(tmpPath.size() - 1);
}
MakeDirIfNotExist((tmpPath + LOCSLASH_S + "subdir").data());
const char* path[2] = {tmpPath.data(), nullptr};
TFileTree fileTree((char* const*)path, FTS_SEEDOT, FtsCmp);
UNIT_ASSERT(fileTree());
CheckEnt(yfts_read(fileTree()), tmpPath.data(), FTS_D);
#ifndef _win32_
CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S ".").data(), FTS_DOT);
#endif // _win32_
CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S "..").data(), FTS_DOT);
CheckEnt(yfts_read(fileTree()), (tmpPath + LOCSLASH_S "subdir").data(), FTS_D);
auto pool = CreateThreadPool(2);
auto chmodFuture = NThreading::Async([name = tmpPath] {
UNIT_ASSERT_C(!chmod(name.data(), 0), "Errno: " << errno);
}, *pool);
auto childrenFuture = NThreading::Async([&] {
yfts_children(fileTree(), 0);
}, *pool);
childrenFuture.Wait();
chmodFuture.Wait();
}
UNIT_TEST_SUITE_REGISTRATION(TFtsTest);