Function TEST_F

tests/utanalysis/TestTargetAPIFinder.cpp:10–49

class TestTargetAPIFinder : public TestBase {};

TEST_F(TestTargetAPIFinder, BasicP) {

  const std::string CODE = "void API_1();\n"
                           "void API_2();\n"
                           "void API_3();\n"
                           "void test() {\n"
                           "  API_1();\n"
                           "  API_2();\n"
                           "}\n";

  ASSERT_TRUE(loadC(CODE));

  auto *API1 = IRAH->getFunction("API_1");
  auto *API2 = IRAH->getFunction("API_2");
  ASSERT_TRUE(API1);
  ASSERT_TRUE(API2);

  auto *Test = IRAH->getFunction("test");
  ASSERT_TRUE(Test);

  std::set<llvm::Function *> APIFuncs = {API1, API2};
  std::vector<llvm::Function *> Funcs = {Test};
  auto Finder = TargetAPIFinder(APIFuncs);
  auto TargetAPIs = Finder.findAPIs(Funcs);
  ASSERT_EQ(TargetAPIs.size(), 2);
  ASSERT_NE(TargetAPIs.find(API1), TargetAPIs.end());
  ASSERT_NE(TargetAPIs.find(API2), TargetAPIs.end());

  auto APICallers = Finder.findAPICallers(Funcs);
  ASSERT_EQ(APICallers.size(), 2);
  auto *Caller = IRAH->getInstruction("test", 0, 0);
  ASSERT_TRUE(Caller);
  ASSERT_TRUE(isa<CallBase>(Caller));
  ASSERT_NE(APICallers.find(dyn_cast<CallBase>(Caller)), APICallers.end());

  Caller = IRAH->getInstruction("test", 0, 1);
  ASSERT_TRUE(Caller);
  ASSERT_TRUE(isa<CallBase>(Caller));
  ASSERT_NE(APICallers.find(dyn_cast<CallBase>(Caller)), APICallers.end());
}

TEST_F(TestTargetAPIFinder, NullAPIN) {