Home | History | Annotate | Download | only in files
      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
     93