Merge "Make the ftw(3) tests a bit more thorough."

This commit is contained in:
Elliott Hughes 2014-11-20 00:09:32 +00:00 committed by Gerrit Code Review
commit 770198d3f7

View File

@ -14,27 +14,36 @@
* limitations under the License. * limitations under the License.
*/ */
#include <gtest/gtest.h>
#include "TemporaryFile.h"
#include <ftw.h> #include <ftw.h>
#include <stdlib.h> #include <stdlib.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <gtest/gtest.h>
void sanity_check_ftw(const char* fpath, const struct stat* sb, int tflag) { void sanity_check_ftw(const char* fpath, const struct stat* sb, int tflag) {
ASSERT_TRUE(fpath != NULL); ASSERT_TRUE(fpath != NULL);
ASSERT_TRUE(sb != NULL); ASSERT_TRUE(sb != NULL);
bool is_dir = S_ISDIR(sb->st_mode); if (S_ISDIR(sb->st_mode)) {
ASSERT_TRUE((is_dir && tflag == FTW_D) || (!is_dir && tflag == FTW_F)); ASSERT_TRUE(tflag == FTW_D || tflag == FTW_DNR || tflag == FTW_DP) << fpath;
} else if (S_ISLNK(sb->st_mode)) {
ASSERT_EQ(FTW_SL, tflag) << fpath;
} else {
ASSERT_EQ(FTW_F, tflag) << fpath;
}
} }
void sanity_check_nftw( void sanity_check_nftw(const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
sanity_check_ftw(fpath, sb, tflag); sanity_check_ftw(fpath, sb, tflag);
// either the parent dir or the file
bool is_dir = S_ISDIR(sb->st_mode); size_t slash_count = 0;
ASSERT_TRUE( const char* p = fpath;
(is_dir && ftwbuf->level == 0) || (!is_dir && ftwbuf->level == 1)); while ((p = strchr(p + 1, '/')) != NULL) {
++slash_count;
}
ASSERT_EQ('/', fpath[ftwbuf->base - 1]) << fpath;
ASSERT_EQ(slash_count, static_cast<size_t>(ftwbuf->level)) << fpath;
} }
int check_ftw(const char* fpath, const struct stat* sb, int tflag) { int check_ftw(const char* fpath, const struct stat* sb, int tflag) {
@ -47,39 +56,28 @@ int check_ftw64(const char* fpath, const struct stat64* sb, int tflag) {
return 0; return 0;
} }
int check_nftw( int check_nftw(const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
const char* fpath, const struct stat* sb, int tflag, struct FTW* ftwbuf) {
sanity_check_nftw(fpath, sb, tflag, ftwbuf); sanity_check_nftw(fpath, sb, tflag, ftwbuf);
return 0; return 0;
} }
int check_nftw64( int check_nftw64(const char* fpath, const struct stat64* sb, int tflag, struct FTW* ftwbuf) {
const char* fpath, const struct stat64* sb, int tflag, struct FTW* ftwbuf) { sanity_check_nftw(fpath, reinterpret_cast<const struct stat*>(sb), tflag, ftwbuf);
sanity_check_nftw(fpath, reinterpret_cast<const struct stat*>(sb),
tflag, ftwbuf);
return 0; return 0;
} }
TEST(ftw, ftw) { TEST(ftw, ftw) {
TemporaryDir td; ASSERT_EQ(0, ftw("/sys", check_ftw, 128));
TemporaryFile tf(td.dirname);
ftw(td.dirname, check_ftw, 1);
} }
TEST(ftw, ftw64) { TEST(ftw, ftw64) {
TemporaryDir td; ASSERT_EQ(0, ftw64("/sys", check_ftw64, 128));
TemporaryFile tf(td.dirname, mkstemp64);
ftw64(td.dirname, check_ftw64, 1);
} }
TEST(ftw, nftw) { TEST(ftw, nftw) {
TemporaryDir td; ASSERT_EQ(0, nftw("/sys", check_nftw, 128, 0));
TemporaryFile tf(td.dirname);
nftw(td.dirname, check_nftw, 1, 0);
} }
TEST(ftw, nftw64) { TEST(ftw, nftw64) {
TemporaryDir td; ASSERT_EQ(0, nftw64("/sys", check_nftw64, 128, 0));
TemporaryFile tf(td.dirname, mkstemp64);
nftw64(td.dirname, check_nftw64, 1, 0);
} }