Home | History | Annotate | Download | only in Support 
      1 //===- SpecialCaseListTest.cpp - Unit tests for SpecialCaseList -----------===//
      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 "llvm/Support/FileSystem.h"
     11 #include "llvm/Support/MemoryBuffer.h"
     12 #include "llvm/Support/SpecialCaseList.h"
     13 #include "gtest/gtest.h"
     14 
     15 using namespace llvm;
     16 
     17 namespace {
     18 
     19 class SpecialCaseListTest : public ::testing::Test {
     20 protected:
     21   std::unique_ptr<SpecialCaseList> makeSpecialCaseList(StringRef List,
     22                                                        std::string &Error) {
     23     std::unique_ptr<MemoryBuffer> MB = MemoryBuffer::getMemBuffer(List);
     24     return SpecialCaseList::create(MB.get(), Error);
     25   }
     26 
     27   std::unique_ptr<SpecialCaseList> makeSpecialCaseList(StringRef List) {
     28     std::string Error;
     29     auto SCL = makeSpecialCaseList(List, Error);
     30     assert(SCL);
     31     assert(Error == "");
     32     return SCL;
     33   }
     34 
     35   std::string makeSpecialCaseListFile(StringRef Contents) {
     36     int FD;
     37     SmallString<64> Path;
     38     sys::fs::createTemporaryFile("SpecialCaseListTest", "temp", FD, Path);
     39     raw_fd_ostream OF(FD, true, true);
     40     OF << Contents;
     41     OF.close();
     42     return Path.str();
     43   }
     44 };
     45 
     46 TEST_F(SpecialCaseListTest, Basic) {
     47   std::unique_ptr<SpecialCaseList> SCL =
     48       makeSpecialCaseList("# This is a comment.\n"
     49                           "\n"
     50                           "src:hello\n"
     51                           "src:bye\n"
     52                           "src:hi=category\n"
     53                           "src:z*=category\n");
     54   EXPECT_TRUE(SCL->inSection("src", "hello"));
     55   EXPECT_TRUE(SCL->inSection("src", "bye"));
     56   EXPECT_TRUE(SCL->inSection("src", "hi", "category"));
     57   EXPECT_TRUE(SCL->inSection("src", "zzzz", "category"));
     58   EXPECT_FALSE(SCL->inSection("src", "hi"));
     59   EXPECT_FALSE(SCL->inSection("fun", "hello"));
     60   EXPECT_FALSE(SCL->inSection("src", "hello", "category"));
     61 }
     62 
     63 TEST_F(SpecialCaseListTest, GlobalInit) {
     64   std::unique_ptr<SpecialCaseList> SCL =
     65       makeSpecialCaseList("global:foo=init\n");
     66   EXPECT_FALSE(SCL->inSection("global", "foo"));
     67   EXPECT_FALSE(SCL->inSection("global", "bar"));
     68   EXPECT_TRUE(SCL->inSection("global", "foo", "init"));
     69   EXPECT_FALSE(SCL->inSection("global", "bar", "init"));
     70 
     71   SCL = makeSpecialCaseList("type:t2=init\n");
     72   EXPECT_FALSE(SCL->inSection("type", "t1"));
     73   EXPECT_FALSE(SCL->inSection("type", "t2"));
     74   EXPECT_FALSE(SCL->inSection("type", "t1", "init"));
     75   EXPECT_TRUE(SCL->inSection("type", "t2", "init"));
     76 
     77   SCL = makeSpecialCaseList("src:hello=init\n");
     78   EXPECT_FALSE(SCL->inSection("src", "hello"));
     79   EXPECT_FALSE(SCL->inSection("src", "bye"));
     80   EXPECT_TRUE(SCL->inSection("src", "hello", "init"));
     81   EXPECT_FALSE(SCL->inSection("src", "bye", "init"));
     82 }
     83 
     84 TEST_F(SpecialCaseListTest, Substring) {
     85   std::unique_ptr<SpecialCaseList> SCL = makeSpecialCaseList("src:hello\n"
     86                                                              "fun:foo\n"
     87                                                              "global:bar\n");
     88   EXPECT_FALSE(SCL->inSection("src", "othello"));
     89   EXPECT_FALSE(SCL->inSection("fun", "tomfoolery"));
     90   EXPECT_FALSE(SCL->inSection("global", "bartender"));
     91 
     92   SCL = makeSpecialCaseList("fun:*foo*\n");
     93   EXPECT_TRUE(SCL->inSection("fun", "tomfoolery"));
     94   EXPECT_TRUE(SCL->inSection("fun", "foobar"));
     95 }
     96 
     97 TEST_F(SpecialCaseListTest, InvalidSpecialCaseList) {
     98   std::string Error;
     99   EXPECT_EQ(nullptr, makeSpecialCaseList("badline", Error));
    100   EXPECT_EQ("malformed line 1: 'badline'", Error);
    101   EXPECT_EQ(nullptr, makeSpecialCaseList("src:bad[a-", Error));
    102   EXPECT_EQ("malformed regex in line 1: 'bad[a-': invalid character range",
    103             Error);
    104   EXPECT_EQ(nullptr, makeSpecialCaseList("src:a.c\n"
    105                                    "fun:fun(a\n",
    106                                    Error));
    107   EXPECT_EQ("malformed regex in line 2: 'fun(a': parentheses not balanced",
    108             Error);
    109   std::vector<std::string> Files(1, "unexisting");
    110   EXPECT_EQ(nullptr, SpecialCaseList::create(Files, Error));
    111   EXPECT_EQ(0U, Error.find("can't open file 'unexisting':"));
    112 }
    113 
    114 TEST_F(SpecialCaseListTest, EmptySpecialCaseList) {
    115   std::unique_ptr<SpecialCaseList> SCL = makeSpecialCaseList("");
    116   EXPECT_FALSE(SCL->inSection("foo", "bar"));
    117 }
    118 
    119 TEST_F(SpecialCaseListTest, MultipleBlacklists) {
    120   std::vector<std::string> Files;
    121   Files.push_back(makeSpecialCaseListFile("src:bar\n"
    122                                           "src:*foo*\n"
    123                                           "src:ban=init\n"));
    124   Files.push_back(makeSpecialCaseListFile("src:baz\n"
    125                                           "src:*fog*\n"));
    126   auto SCL = SpecialCaseList::createOrDie(Files);
    127   EXPECT_TRUE(SCL->inSection("src", "bar"));
    128   EXPECT_TRUE(SCL->inSection("src", "baz"));
    129   EXPECT_FALSE(SCL->inSection("src", "ban"));
    130   EXPECT_TRUE(SCL->inSection("src", "ban", "init"));
    131   EXPECT_TRUE(SCL->inSection("src", "tomfoolery"));
    132   EXPECT_TRUE(SCL->inSection("src", "tomfoglery"));
    133 }
    134 
    135 }
    136