unittests/Support/MDBuilderTest.cpp

   1 //===- llvm/unittests/Support/MDBuilderTest.cpp - MDBuilder unit 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 "gtest/gtest.h"
  11 #include "llvm/Support/MDBuilder.h"
  12 using namespace llvm;
  13
  14 namespace {
  15
  16 class MDBuilderTest : public testing::Test {
  17 protected:
  18   LLVMContext Context;
  19 };
  20
  21 TEST_F(MDBuilderTest, CreateString) {
  22   MDBuilder MDHelper(Context);
  23   MDString *Str0 = MDHelper.CreateString("");
  24   MDString *Str1 = MDHelper.CreateString("string");
  25   EXPECT_EQ(Str0->getString(), StringRef(""));
  26   EXPECT_EQ(Str1->getString(), StringRef("string"));
  27 }
  28
  29 TEST_F(MDBuilderTest, CreateRangeMetadata) {
  30   MDBuilder MDHelper(Context);
  31   APInt A(8, 1), B(8, 2);
  32   MDNode *R0 = MDHelper.CreateRange(A, A);
  33   MDNode *R1 = MDHelper.CreateRange(A, B);
  34   EXPECT_EQ(R0, (MDNode *)0);
  35   EXPECT_NE(R1, (MDNode *)0);
  36   EXPECT_EQ(R1->getNumOperands(), 2U);
  37   EXPECT_TRUE(isa<ConstantInt>(R1->getOperand(0)));
  38   EXPECT_TRUE(isa<ConstantInt>(R1->getOperand(1)));
  39   ConstantInt *C0 = cast<ConstantInt>(R1->getOperand(0));
  40   ConstantInt *C1 = cast<ConstantInt>(R1->getOperand(1));
  41   EXPECT_EQ(C0->getValue(), A);
  42   EXPECT_EQ(C1->getValue(), B);
  43 }
  44 TEST_F(MDBuilderTest, CreateAnonymousTBAARoot) {
  45   MDBuilder MDHelper(Context);
  46   MDNode *R0 = MDHelper.CreateAnonymousTBAARoot();
  47   MDNode *R1 = MDHelper.CreateAnonymousTBAARoot();
  48   EXPECT_NE(R0, R1);
  49   EXPECT_GE(R0->getNumOperands(), 1U);
  50   EXPECT_GE(R1->getNumOperands(), 1U);
  51   EXPECT_EQ(R0->getOperand(0), R0);
  52   EXPECT_EQ(R1->getOperand(0), R1);
  53   EXPECT_TRUE(R0->getNumOperands() == 1 || R0->getOperand(1) == 0);
  54   EXPECT_TRUE(R1->getNumOperands() == 1 || R1->getOperand(1) == 0);
  55 }
  56 TEST_F(MDBuilderTest, CreateTBAARoot) {
  57   MDBuilder MDHelper(Context);
  58   MDNode *R0 = MDHelper.CreateTBAARoot("Root");
  59   MDNode *R1 = MDHelper.CreateTBAARoot("Root");
  60   EXPECT_EQ(R0, R1);
  61   EXPECT_GE(R0->getNumOperands(), 1U);
  62   EXPECT_TRUE(isa<MDString>(R0->getOperand(0)));
  63   EXPECT_EQ(cast<MDString>(R0->getOperand(0))->getString(), "Root");
  64   EXPECT_TRUE(R0->getNumOperands() == 1 || R0->getOperand(1) == 0);
  65 }
  66 TEST_F(MDBuilderTest, CreateTBAANode) {
  67   MDBuilder MDHelper(Context);
  68   MDNode *R = MDHelper.CreateTBAARoot("Root");
  69   MDNode *N0 = MDHelper.CreateTBAANode("Node", R);
  70   MDNode *N1 = MDHelper.CreateTBAANode("edoN", R);
  71   MDNode *N2 = MDHelper.CreateTBAANode("Node", R, true);
  72   MDNode *N3 = MDHelper.CreateTBAANode("Node", R);
  73   EXPECT_EQ(N0, N3);
  74   EXPECT_NE(N0, N1);
  75   EXPECT_NE(N0, N2);
  76   EXPECT_GE(N0->getNumOperands(), 2U);
  77   EXPECT_GE(N1->getNumOperands(), 2U);
  78   EXPECT_GE(N2->getNumOperands(), 3U);
  79   EXPECT_TRUE(isa<MDString>(N0->getOperand(0)));
  80   EXPECT_TRUE(isa<MDString>(N1->getOperand(0)));
  81   EXPECT_TRUE(isa<MDString>(N2->getOperand(0)));
  82   EXPECT_EQ(cast<MDString>(N0->getOperand(0))->getString(), "Node");
  83   EXPECT_EQ(cast<MDString>(N1->getOperand(0))->getString(), "edoN");
  84   EXPECT_EQ(cast<MDString>(N2->getOperand(0))->getString(), "Node");
  85   EXPECT_EQ(N0->getOperand(1), R);
  86   EXPECT_EQ(N1->getOperand(1), R);
  87   EXPECT_EQ(N2->getOperand(1), R);
  88   EXPECT_TRUE(isa<ConstantInt>(N2->getOperand(2)));
  89   EXPECT_EQ(cast<ConstantInt>(N2->getOperand(2))->getZExtValue(), 1U);
  90 }
  91 }