Home | History | Annotate | Download | only in Basic
      1 //===- unittests/Basic/FileMangerTest.cpp ------------ FileManger tests ---===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 
     10 #include "clang/Basic/FileManager.h"
     11 #include "clang/Basic/FileSystemOptions.h"
     12 #include "clang/Basic/FileSystemStatCache.h"
     13 #include "gtest/gtest.h"
     14 #include "llvm/Config/llvm-config.h"
     15 
     16 using namespace llvm;
     17 using namespace clang;
     18 
     19 namespace {
     20 
     21 // Used to create a fake file system for running the tests with such
     22 // that the tests are not affected by the structure/contents of the
     23 // file system on the machine running the tests.
     24 class FakeStatCache : public FileSystemStatCache {
     25 private:
     26   // Maps a file/directory path to its desired stat result.  Anything
     27   // not in this map is considered to not exist in the file system.
     28   llvm::StringMap<FileData, llvm::BumpPtrAllocator> StatCalls;
     29 
     30   void InjectFileOrDirectory(const char *Path, ino_t INode, bool IsFile) {
     31     FileData Data;
     32     Data.Name = Path;
     33     Data.Size = 0;
     34     Data.ModTime = 0;
     35     Data.UniqueID = llvm::sys::fs::UniqueID(1, INode);
     36     Data.IsDirectory = !IsFile;
     37     Data.IsNamedPipe = false;
     38     Data.InPCH = false;
     39     StatCalls[Path] = Data;
     40   }
     41 
     42 public:
     43   // Inject a file with the given inode value to the fake file system.
     44   void InjectFile(const char *Path, ino_t INode) {
     45     InjectFileOrDirectory(Path, INode, /*IsFile=*/true);
     46   }
     47 
     48   // Inject a directory with the given inode value to the fake file system.
     49   void InjectDirectory(const char *Path, ino_t INode) {
     50     InjectFileOrDirectory(Path, INode, /*IsFile=*/false);
     51   }
     52 
     53   // Implement FileSystemStatCache::getStat().
     54   LookupResult getStat(const char *Path, FileData &Data, bool isFile,
     55                        std::unique_ptr<vfs::File> *F,
     56                        vfs::FileSystem &FS) override {
     57     if (StatCalls.count(Path) != 0) {
     58       Data = StatCalls[Path];
     59       return CacheExists;
     60     }
     61 
     62     return CacheMissing;  // This means the file/directory doesn't exist.
     63   }
     64 };
     65 
     66 // The test fixture.
     67 class FileManagerTest : public ::testing::Test {
     68  protected:
     69   FileManagerTest() : manager(options) {
     70   }
     71 
     72   FileSystemOptions options;
     73   FileManager manager;
     74 };
     75 
     76 // When a virtual file is added, its getDir() field is set correctly
     77 // (not NULL, correct name).
     78 TEST_F(FileManagerTest, getVirtualFileSetsTheDirFieldCorrectly) {
     79   const FileEntry *file = manager.getVirtualFile("foo.cpp", 42, 0);
     80   ASSERT_TRUE(file != nullptr);
     81 
     82   const DirectoryEntry *dir = file->getDir();
     83   ASSERT_TRUE(dir != nullptr);
     84   EXPECT_STREQ(".", dir->getName());
     85 
     86   file = manager.getVirtualFile("x/y/z.cpp", 42, 0);
     87   ASSERT_TRUE(file != nullptr);
     88 
     89   dir = file->getDir();
     90   ASSERT_TRUE(dir != nullptr);
     91   EXPECT_STREQ("x/y", dir->getName());
     92 }
     93 
     94 // Before any virtual file is added, no virtual directory exists.
     95 TEST_F(FileManagerTest, NoVirtualDirectoryExistsBeforeAVirtualFileIsAdded) {
     96   // An empty FakeStatCache causes all stat calls made by the
     97   // FileManager to report "file/directory doesn't exist".  This
     98   // avoids the possibility of the result of this test being affected
     99   // by what's in the real file system.
    100   manager.addStatCache(new FakeStatCache);
    101 
    102   EXPECT_EQ(nullptr, manager.getDirectory("virtual/dir/foo"));
    103   EXPECT_EQ(nullptr, manager.getDirectory("virtual/dir"));
    104   EXPECT_EQ(nullptr, manager.getDirectory("virtual"));
    105 }
    106 
    107 // When a virtual file is added, all of its ancestors should be created.
    108 TEST_F(FileManagerTest, getVirtualFileCreatesDirectoryEntriesForAncestors) {
    109   // Fake an empty real file system.
    110   manager.addStatCache(new FakeStatCache);
    111 
    112   manager.getVirtualFile("virtual/dir/bar.h", 100, 0);
    113   EXPECT_EQ(nullptr, manager.getDirectory("virtual/dir/foo"));
    114 
    115   const DirectoryEntry *dir = manager.getDirectory("virtual/dir");
    116   ASSERT_TRUE(dir != nullptr);
    117   EXPECT_STREQ("virtual/dir", dir->getName());
    118 
    119   dir = manager.getDirectory("virtual");
    120   ASSERT_TRUE(dir != nullptr);
    121   EXPECT_STREQ("virtual", dir->getName());
    122 }
    123 
    124 // getFile() returns non-NULL if a real file exists at the given path.
    125 TEST_F(FileManagerTest, getFileReturnsValidFileEntryForExistingRealFile) {
    126   // Inject fake files into the file system.
    127   FakeStatCache *statCache = new FakeStatCache;
    128   statCache->InjectDirectory("/tmp", 42);
    129   statCache->InjectFile("/tmp/test", 43);
    130 
    131 #ifdef LLVM_ON_WIN32
    132   const char *DirName = "C:.";
    133   const char *FileName = "C:test";
    134   statCache->InjectDirectory(DirName, 44);
    135   statCache->InjectFile(FileName, 45);
    136 #endif
    137 
    138   manager.addStatCache(statCache);
    139 
    140   const FileEntry *file = manager.getFile("/tmp/test");
    141   ASSERT_TRUE(file != nullptr);
    142   EXPECT_STREQ("/tmp/test", file->getName());
    143 
    144   const DirectoryEntry *dir = file->getDir();
    145   ASSERT_TRUE(dir != nullptr);
    146   EXPECT_STREQ("/tmp", dir->getName());
    147 
    148 #ifdef LLVM_ON_WIN32
    149   file = manager.getFile(FileName);
    150   ASSERT_TRUE(file != NULL);
    151 
    152   dir = file->getDir();
    153   ASSERT_TRUE(dir != NULL);
    154   EXPECT_STREQ(DirName, dir->getName());
    155 #endif
    156 }
    157 
    158 // getFile() returns non-NULL if a virtual file exists at the given path.
    159 TEST_F(FileManagerTest, getFileReturnsValidFileEntryForExistingVirtualFile) {
    160   // Fake an empty real file system.
    161   manager.addStatCache(new FakeStatCache);
    162 
    163   manager.getVirtualFile("virtual/dir/bar.h", 100, 0);
    164   const FileEntry *file = manager.getFile("virtual/dir/bar.h");
    165   ASSERT_TRUE(file != nullptr);
    166   EXPECT_STREQ("virtual/dir/bar.h", file->getName());
    167 
    168   const DirectoryEntry *dir = file->getDir();
    169   ASSERT_TRUE(dir != nullptr);
    170   EXPECT_STREQ("virtual/dir", dir->getName());
    171 }
    172 
    173 // getFile() returns different FileEntries for different paths when
    174 // there's no aliasing.
    175 TEST_F(FileManagerTest, getFileReturnsDifferentFileEntriesForDifferentFiles) {
    176   // Inject two fake files into the file system.  Different inodes
    177   // mean the files are not symlinked together.
    178   FakeStatCache *statCache = new FakeStatCache;
    179   statCache->InjectDirectory(".", 41);
    180   statCache->InjectFile("foo.cpp", 42);
    181   statCache->InjectFile("bar.cpp", 43);
    182   manager.addStatCache(statCache);
    183 
    184   const FileEntry *fileFoo = manager.getFile("foo.cpp");
    185   const FileEntry *fileBar = manager.getFile("bar.cpp");
    186   ASSERT_TRUE(fileFoo != nullptr);
    187   ASSERT_TRUE(fileBar != nullptr);
    188   EXPECT_NE(fileFoo, fileBar);
    189 }
    190 
    191 // getFile() returns NULL if neither a real file nor a virtual file
    192 // exists at the given path.
    193 TEST_F(FileManagerTest, getFileReturnsNULLForNonexistentFile) {
    194   // Inject a fake foo.cpp into the file system.
    195   FakeStatCache *statCache = new FakeStatCache;
    196   statCache->InjectDirectory(".", 41);
    197   statCache->InjectFile("foo.cpp", 42);
    198   manager.addStatCache(statCache);
    199 
    200   // Create a virtual bar.cpp file.
    201   manager.getVirtualFile("bar.cpp", 200, 0);
    202 
    203   const FileEntry *file = manager.getFile("xyz.txt");
    204   EXPECT_EQ(nullptr, file);
    205 }
    206 
    207 // The following tests apply to Unix-like system only.
    208 
    209 #ifndef LLVM_ON_WIN32
    210 
    211 // getFile() returns the same FileEntry for real files that are aliases.
    212 TEST_F(FileManagerTest, getFileReturnsSameFileEntryForAliasedRealFiles) {
    213   // Inject two real files with the same inode.
    214   FakeStatCache *statCache = new FakeStatCache;
    215   statCache->InjectDirectory("abc", 41);
    216   statCache->InjectFile("abc/foo.cpp", 42);
    217   statCache->InjectFile("abc/bar.cpp", 42);
    218   manager.addStatCache(statCache);
    219 
    220   EXPECT_EQ(manager.getFile("abc/foo.cpp"), manager.getFile("abc/bar.cpp"));
    221 }
    222 
    223 // getFile() returns the same FileEntry for virtual files that have
    224 // corresponding real files that are aliases.
    225 TEST_F(FileManagerTest, getFileReturnsSameFileEntryForAliasedVirtualFiles) {
    226   // Inject two real files with the same inode.
    227   FakeStatCache *statCache = new FakeStatCache;
    228   statCache->InjectDirectory("abc", 41);
    229   statCache->InjectFile("abc/foo.cpp", 42);
    230   statCache->InjectFile("abc/bar.cpp", 42);
    231   manager.addStatCache(statCache);
    232 
    233   manager.getVirtualFile("abc/foo.cpp", 100, 0);
    234   manager.getVirtualFile("abc/bar.cpp", 200, 0);
    235 
    236   EXPECT_EQ(manager.getFile("abc/foo.cpp"), manager.getFile("abc/bar.cpp"));
    237 }
    238 
    239 #endif  // !LLVM_ON_WIN32
    240 
    241 } // anonymous namespace
    242