unittests/MC/AsmStreamerTest.cpp

   1 //===- AsmStreamerTest.cpp - Triple 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/MC/MCContext.h"
  12 #include "llvm/MC/MCSection.h"
  13 #include "llvm/MC/MCStreamer.h"
  14 #include "llvm/MC/MCValue.h"
  15 #include "llvm/Support/raw_ostream.h"
  16 using namespace llvm;
  17
  18 namespace {
  19
  20 // Helper class.
  21 class StringAsmStreamer {
  22   std::string Str;
  23   raw_string_ostream OS;
  24   MCContext Context;
  25   MCStreamer *Streamer;
  26
  27 public:
  28   StringAsmStreamer() : OS(Str), Streamer(createAsmStreamer(Context, OS)) {}
  29   ~StringAsmStreamer() {
  30     delete Streamer;
  31   }
  32
  33   MCContext &getContext() { return Context; }
  34   MCStreamer &getStreamer() { return *Streamer; }
  35
  36   const std::string &getString() {
  37     Streamer->Finish();
  38     return Str;
  39   }
  40 };
  41
  42 TEST(AsmStreamer, EmptyOutput) {
  43   StringAsmStreamer S;
  44   EXPECT_EQ("", S.getString());
  45 }
  46
  47 TEST(AsmStreamer, Sections) {
  48   StringAsmStreamer S;
  49   MCSection *Sec0 = MCSection::Create("foo", SectionKind(), S.getContext());
  50   S.getStreamer().SwitchSection(Sec0);
  51   EXPECT_EQ(".section foo\n", S.getString());
  52 }
  53
  54 TEST(AsmStreamer, Values) {
  55   StringAsmStreamer S;
  56   MCSection *Sec0 = MCSection::Create("foo", SectionKind(), S.getContext());
  57   MCSymbol *A = S.getContext().CreateSymbol("a");
  58   MCSymbol *B = S.getContext().CreateSymbol("b");
  59   S.getStreamer().SwitchSection(Sec0);
  60   S.getStreamer().EmitLabel(A);
  61   S.getStreamer().EmitLabel(B);
  62   S.getStreamer().EmitValue(MCValue::get(A, B, 10), 1);
  63   S.getStreamer().EmitValue(MCValue::get(A, B, 10), 2);
  64   S.getStreamer().EmitValue(MCValue::get(A, B, 10), 4);
  65   S.getStreamer().EmitValue(MCValue::get(A, B, 10), 8);
  66   EXPECT_EQ(".section foo\n\
  67 a:\n\
  68 b:\n\
  69 .byte a - b + 10\n\
  70 .short a - b + 10\n\
  71 .long a - b + 10\n\
  72 .quad a - b + 10\n", S.getString());
  73 }
  74
  75 TEST(AsmStreamer, Align) {
  76   StringAsmStreamer S;
  77   MCSection *Sec0 = MCSection::Create("foo", SectionKind(), S.getContext());
  78   S.getStreamer().SwitchSection(Sec0);
  79   S.getStreamer().EmitValueToAlignment(4);
  80   S.getStreamer().EmitValueToAlignment(4, /*Value=*/12, /*ValueSize=*/2);
  81   S.getStreamer().EmitValueToAlignment(8, /*Value=*/12, /*ValueSize=*/4,
  82                                        /*MaxBytesToEmit=*/24);
  83   EXPECT_EQ(".section foo\n\
  84 .p2align 2, 0\n\
  85 .p2alignw 2, 12\n\
  86 .p2alignl 3, 12, 24\n", S.getString());
  87 }
  88
  89 TEST(AsmStreamer, Org) {
  90   StringAsmStreamer S;
  91   MCSection *Sec0 = MCSection::Create("foo", SectionKind(), S.getContext());
  92   S.getStreamer().SwitchSection(Sec0);
  93   MCSymbol *A = S.getContext().CreateSymbol("a");
  94   S.getStreamer().EmitLabel(A);
  95   S.getStreamer().EmitValueToOffset(MCValue::get(A, 0, 4), 32);
  96   EXPECT_EQ(".section foo\n\
  97 a:\n\
  98 .org a + 4, 32\n", S.getString());
  99 }
 100
 101 }