unittests/tools/llvm-exegesis/AArch64/TargetTest.cpp

   1 #include "Target.h"
   2
   3 #include <cassert>
   4 #include <memory>
   5
   6 #include "MCTargetDesc/AArch64MCTargetDesc.h"
   7 #include "llvm/Support/TargetRegistry.h"
   8 #include "llvm/Support/TargetSelect.h"
   9 #include "gmock/gmock.h"
  10 #include "gtest/gtest.h"
  11
  12 namespace exegesis {
  13
  14 void InitializeAArch64ExegesisTarget();
  15
  16 namespace {
  17
  18 using llvm::APInt;
  19 using llvm::MCInst;
  20 using testing::Gt;
  21 using testing::IsEmpty;
  22 using testing::Not;
  23 using testing::NotNull;
  24
  25 constexpr const char kTriple[] = "aarch64-unknown-linux";
  26 constexpr const char kGenericCpu[] = "generic";
  27 constexpr const char kNoFeatures[] = "";
  28
  29 class AArch64TargetTest : public ::testing::Test {
  30 protected:
  31   AArch64TargetTest()
  32       : ExegesisTarget_(ExegesisTarget::lookup(llvm::Triple(kTriple))) {
  33     EXPECT_THAT(ExegesisTarget_, NotNull());
  34     std::string error;
  35     Target_ = llvm::TargetRegistry::lookupTarget(kTriple, error);
  36     EXPECT_THAT(Target_, NotNull());
  37     STI_.reset(
  38         Target_->createMCSubtargetInfo(kTriple, kGenericCpu, kNoFeatures));
  39   }
  40
  41   static void SetUpTestCase() {
  42     LLVMInitializeAArch64TargetInfo();
  43     LLVMInitializeAArch64Target();
  44     LLVMInitializeAArch64TargetMC();
  45     InitializeAArch64ExegesisTarget();
  46   }
  47
  48   std::vector<MCInst> setRegTo(unsigned Reg, const APInt &Value) {
  49     return ExegesisTarget_->setRegTo(*STI_, Reg, Value);
  50   }
  51
  52   const llvm::Target *Target_;
  53   const ExegesisTarget *const ExegesisTarget_;
  54   std::unique_ptr<llvm::MCSubtargetInfo> STI_;
  55 };
  56
  57 TEST_F(AArch64TargetTest, SetRegToConstant) {
  58   // The AArch64 target currently doesn't know how to set register values.
  59   const auto Insts = setRegTo(llvm::AArch64::X0, llvm::APInt());
  60   EXPECT_THAT(Insts, Not(IsEmpty()));
  61 }
  62
  63 } // namespace
  64 } // namespace exegesis