llvm/unittests/IR/AsmWriterTest.cpp

   1 //===- llvm/unittest/IR/AsmWriter.cpp - AsmWriter tests -------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 #include "llvm/BinaryFormat/Dwarf.h"
   9 #include "llvm/IR/DebugInfoMetadata.h"
  10 #include "llvm/IR/Function.h"
  11 #include "llvm/IR/IRBuilder.h"
  12 #include "llvm/IR/LLVMContext.h"
  13 #include "llvm/IR/MDBuilder.h"
  14 #include "llvm/IR/Module.h"
  15 #include "gtest/gtest.h"
  16
  17 using namespace llvm;
  18
  19 namespace {
  20
  21 TEST(AsmWriterTest, DebugPrintDetachedInstruction) {
  22
  23   // PR24852: Ensure that an instruction can be printed even when it
  24   // has metadata attached but no parent.
  25   LLVMContext Ctx;
  26   auto Ty = Type::getInt32Ty(Ctx);
  27   auto Poison = PoisonValue::get(Ty);
  28   std::unique_ptr<BinaryOperator> Add(BinaryOperator::CreateAdd(Poison, Poison));
  29   Add->setMetadata(
  30       "", MDNode::get(Ctx, {ConstantAsMetadata::get(ConstantInt::get(Ty, 1))}));
  31   std::string S;
  32   raw_string_ostream OS(S);
  33   Add->print(OS);
  34   std::size_t r = OS.str().find("<badref> = add i32 poison, poison, !<empty");
  35   EXPECT_TRUE(r != std::string::npos);
  36 }
  37
  38 TEST(AsmWriterTest, DebugPrintDetachedArgument) {
  39   LLVMContext Ctx;
  40   auto Ty = Type::getInt32Ty(Ctx);
  41   auto Arg = new Argument(Ty);
  42
  43   std::string S;
  44   raw_string_ostream OS(S);
  45   Arg->print(OS);
  46   EXPECT_EQ(S, "i32 <badref>");
  47   delete Arg;
  48 }
  49
  50 TEST(AsmWriterTest, DumpDIExpression) {
  51   LLVMContext Ctx;
  52   uint64_t Ops[] = {
  53     dwarf::DW_OP_constu, 4,
  54     dwarf::DW_OP_minus,
  55     dwarf::DW_OP_deref,
  56   };
  57   DIExpression *Expr = DIExpression::get(Ctx, Ops);
  58   std::string S;
  59   raw_string_ostream OS(S);
  60   Expr->print(OS);
  61   EXPECT_EQ("!DIExpression(DW_OP_constu, 4, DW_OP_minus, DW_OP_deref)",
  62             OS.str());
  63 }
  64
  65 }