| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/files/dir_reader_posix.h" | |
| 6 | |
| 7 #include <fcntl.h> | |
| 8 #include <stdio.h> | |
| 9 #include <stdlib.h> | |
| 10 #include <string.h> | |
| 11 #include <unistd.h> | |
| 12 | |
| 13 #include "base/logging.h" | |
| 14 #include "testing/gtest/include/gtest/gtest.h" | |
| 15 | |
| 16 #if defined(OS_ANDROID) | |
| 17 #include "base/os_compat_android.h" | |
| 18 #endif | |
| 19 | |
| 20 namespace base { | |
| 21 | |
| 22 TEST(DirReaderPosixUnittest, Read) { | |
| 23 static const unsigned kNumFiles = 100; | |
| 24 | |
| 25 if (DirReaderPosix::IsFallback()) | |
| 26 return; | |
| 27 | |
| 28 char kDirTemplate[] = "/tmp/org.chromium.dir-reader-posix-XXXXXX"; | |
| 29 const char* dir = mkdtemp(kDirTemplate); | |
| 30 ASSERT_TRUE(dir); | |
| 31 | |
| 32 const int prev_wd = open(".", O_RDONLY | O_DIRECTORY); | |
| 33 DCHECK_GE(prev_wd, 0); | |
| 34 | |
| 35 PCHECK(chdir(dir) == 0); | |
| 36 | |
| 37 for (unsigned i = 0; i < kNumFiles; i++) { | |
| 38 char buf[16]; | |
| 39 snprintf(buf, sizeof(buf), "%d", i); | |
| 40 const int fd = open(buf, O_CREAT | O_RDONLY | O_EXCL, 0600); | |
| 41 PCHECK(fd >= 0); | |
| 42 PCHECK(close(fd) == 0); | |
| 43 } | |
| 44 | |
| 45 std::set<unsigned> seen; | |
| 46 | |
| 47 DirReaderPosix reader(dir); | |
| 48 EXPECT_TRUE(reader.IsValid()); | |
| 49 | |
| 50 if (!reader.IsValid()) | |
| 51 return; | |
| 52 | |
| 53 bool seen_dot = false, seen_dotdot = false; | |
| 54 | |
| 55 for (; reader.Next(); ) { | |
| 56 if (strcmp(reader.name(), ".") == 0) { | |
| 57 seen_dot = true; | |
| 58 continue; | |
| 59 } | |
| 60 if (strcmp(reader.name(), "..") == 0) { | |
| 61 seen_dotdot = true; | |
| 62 continue; | |
| 63 } | |
| 64 | |
| 65 SCOPED_TRACE(testing::Message() << "reader.name(): " << reader.name()); | |
| 66 | |
| 67 char *endptr; | |
| 68 const unsigned long value = strtoul(reader.name(), &endptr, 10); | |
| 69 | |
| 70 EXPECT_FALSE(*endptr); | |
| 71 EXPECT_LT(value, kNumFiles); | |
| 72 EXPECT_EQ(0u, seen.count(value)); | |
| 73 seen.insert(value); | |
| 74 } | |
| 75 | |
| 76 for (unsigned i = 0; i < kNumFiles; i++) { | |
| 77 char buf[16]; | |
| 78 snprintf(buf, sizeof(buf), "%d", i); | |
| 79 PCHECK(unlink(buf) == 0); | |
| 80 } | |
| 81 | |
| 82 PCHECK(rmdir(dir) == 0); | |
| 83 | |
| 84 PCHECK(fchdir(prev_wd) == 0); | |
| 85 PCHECK(close(prev_wd) == 0); | |
| 86 | |
| 87 EXPECT_TRUE(seen_dot); | |
| 88 EXPECT_TRUE(seen_dotdot); | |
| 89 EXPECT_EQ(kNumFiles, seen.size()); | |
| 90 } | |
| 91 | |
| 92 } // namespace base | |
| OLD | NEW |
|---|
Chromium Code Reviews
