unittests/ADT/TwineTest.cpp

   1 //===- TwineTest.cpp - Twine 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/ADT/Twine.h"
  12 #include "llvm/Support/raw_ostream.h"
  13 using namespace llvm;
  14
  15 namespace {
  16
  17 std::string repr(const Twine &Value) {
  18   std::string res;
  19   llvm::raw_string_ostream OS(res);
  20   Value.printRepr(OS);
  21   return OS.str();
  22 }
  23
  24 TEST(TwineTest, Construction) {
  25   EXPECT_EQ("", Twine().str());
  26   EXPECT_EQ("hi", Twine("hi").str());
  27   EXPECT_EQ("hi", Twine(std::string("hi")).str());
  28   EXPECT_EQ("hi", Twine(StringRef("hi")).str());
  29   EXPECT_EQ("hi", Twine(StringRef(std::string("hi"))).str());
  30   EXPECT_EQ("hi", Twine(StringRef("hithere", 2)).str());
  31 }
  32
  33 TEST(TwineTest, Numbers) {
  34   EXPECT_EQ("123", Twine(123U).str());
  35   EXPECT_EQ("123", Twine(123).str());
  36   EXPECT_EQ("-123", Twine(-123).str());
  37   EXPECT_EQ("123", Twine(123).str());
  38   EXPECT_EQ("-123", Twine(-123).str());
  39   EXPECT_EQ("123", Twine((char) 123).str());
  40   EXPECT_EQ("-123", Twine((signed char) -123).str());
  41
  42   EXPECT_EQ("7B", Twine::utohexstr(123).str());
  43   EXPECT_EQ("FFFFFFFFFFFFFF85", Twine::itohexstr(-123).str());
  44 }
  45
  46 TEST(TwineTest, Concat) {
  47   // Check verse repr, since we care about the actual representation not just
  48   // the result.
  49
  50   // Concat with null.
  51   EXPECT_EQ("(Twine null empty)",
  52             repr(Twine("hi").concat(Twine::createNull())));
  53   EXPECT_EQ("(Twine null empty)",
  54             repr(Twine::createNull().concat(Twine("hi"))));
  55
  56   // Concat with empty.
  57   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
  58             repr(Twine("hi").concat(Twine())));
  59   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
  60             repr(Twine().concat(Twine("hi"))));
  61
  62   // Concatenation of unary ropes.
  63   EXPECT_EQ("(Twine cstring:\"a\" cstring:\"b\")",
  64             repr(Twine("a").concat(Twine("b"))));
  65
  66   // Concatenation of other ropes.
  67   EXPECT_EQ("(Twine rope:(Twine cstring:\"a\" cstring:\"b\") cstring:\"c\")",
  68             repr(Twine("a").concat(Twine("b")).concat(Twine("c"))));
  69   EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine cstring:\"b\" cstring:\"c\"))",
  70             repr(Twine("a").concat(Twine("b").concat(Twine("c")))));
  71 }
  72
  73   // I suppose linking in the entire code generator to add a unit test to check
  74   // the code size of the concat operation is overkill... :)
  75
  76 } // end anonymous namespace