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 "llvm/ADT/Twine.h"
  11 #include "llvm/ADT/SmallString.h"
  12 #include "llvm/Support/raw_ostream.h"
  13 #include "gtest/gtest.h"
  14 using namespace llvm;
  15
  16 namespace {
  17
  18 std::string repr(const Twine &Value) {
  19   std::string res;
  20   llvm::raw_string_ostream OS(res);
  21   Value.printRepr(OS);
  22   return OS.str();
  23 }
  24
  25 TEST(TwineTest, Construction) {
  26   EXPECT_EQ("", Twine().str());
  27   EXPECT_EQ("hi", Twine("hi").str());
  28   EXPECT_EQ("hi", Twine(std::string("hi")).str());
  29   EXPECT_EQ("hi", Twine(StringRef("hi")).str());
  30   EXPECT_EQ("hi", Twine(StringRef(std::string("hi"))).str());
  31   EXPECT_EQ("hi", Twine(StringRef("hithere", 2)).str());
  32   EXPECT_EQ("hi", Twine(SmallString<4>("hi")).str());
  33 }
  34
  35 TEST(TwineTest, Numbers) {
  36   EXPECT_EQ("123", Twine(123U).str());
  37   EXPECT_EQ("123", Twine(123).str());
  38   EXPECT_EQ("-123", Twine(-123).str());
  39   EXPECT_EQ("123", Twine(123).str());
  40   EXPECT_EQ("-123", Twine(-123).str());
  41
  42   EXPECT_EQ("7b", Twine::utohexstr(123).str());
  43 }
  44
  45 TEST(TwineTest, Characters) {
  46   EXPECT_EQ("x", Twine('x').str());
  47   EXPECT_EQ("x", Twine(static_cast<unsigned char>('x')).str());
  48   EXPECT_EQ("x", Twine(static_cast<signed char>('x')).str());
  49 }
  50
  51 TEST(TwineTest, Concat) {
  52   // Check verse repr, since we care about the actual representation not just
  53   // the result.
  54
  55   // Concat with null.
  56   EXPECT_EQ("(Twine null empty)",
  57             repr(Twine("hi").concat(Twine::createNull())));
  58   EXPECT_EQ("(Twine null empty)",
  59             repr(Twine::createNull().concat(Twine("hi"))));
  60
  61   // Concat with empty.
  62   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
  63             repr(Twine("hi").concat(Twine())));
  64   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
  65             repr(Twine().concat(Twine("hi"))));
  66   EXPECT_EQ("(Twine smallstring:\"hi\" empty)",
  67             repr(Twine().concat(Twine(SmallString<5>("hi")))));
  68   EXPECT_EQ("(Twine smallstring:\"hey\" cstring:\"there\")",
  69             repr(Twine(SmallString<7>("hey")).concat(Twine("there"))));
  70
  71   // Concatenation of unary ropes.
  72   EXPECT_EQ("(Twine cstring:\"a\" cstring:\"b\")",
  73             repr(Twine("a").concat(Twine("b"))));
  74
  75   // Concatenation of other ropes.
  76   EXPECT_EQ("(Twine rope:(Twine cstring:\"a\" cstring:\"b\") cstring:\"c\")",
  77             repr(Twine("a").concat(Twine("b")).concat(Twine("c"))));
  78   EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine cstring:\"b\" cstring:\"c\"))",
  79             repr(Twine("a").concat(Twine("b").concat(Twine("c")))));
  80   EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine smallstring:\"b\" cstring:\"c\"))",
  81             repr(Twine("a").concat(Twine(SmallString<3>("b")).concat(Twine("c")))));
  82 }
  83
  84 TEST(TwineTest, toNullTerminatedStringRef) {
  85   SmallString<8> storage;
  86   EXPECT_EQ(0, *Twine("hello").toNullTerminatedStringRef(storage).end());
  87   EXPECT_EQ(0,
  88            *Twine(StringRef("hello")).toNullTerminatedStringRef(storage).end());
  89   EXPECT_EQ(0, *Twine(SmallString<11>("hello"))
  90                     .toNullTerminatedStringRef(storage)
  91                     .end());
  92 }
  93
  94   // I suppose linking in the entire code generator to add a unit test to check
  95   // the code size of the concat operation is overkill... :)
  96
  97 } // end anonymous namespace