Home | History | Annotate | Download | only in ADT 
      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 }
     33 
     34 TEST(TwineTest, Numbers) {
     35   EXPECT_EQ("123", Twine(123U).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(-123).str());
     40 
     41   EXPECT_EQ("7b", Twine::utohexstr(123).str());
     42 }
     43 
     44 TEST(TwineTest, Characters) {
     45   EXPECT_EQ("x", Twine('x').str());
     46   EXPECT_EQ("x", Twine(static_cast<unsigned char>('x')).str());
     47   EXPECT_EQ("x", Twine(static_cast<signed char>('x')).str());
     48 }
     49 
     50 TEST(TwineTest, Concat) {
     51   // Check verse repr, since we care about the actual representation not just
     52   // the result.
     53 
     54   // Concat with null.
     55   EXPECT_EQ("(Twine null empty)",
     56             repr(Twine("hi").concat(Twine::createNull())));
     57   EXPECT_EQ("(Twine null empty)",
     58             repr(Twine::createNull().concat(Twine("hi"))));
     59 
     60   // Concat with empty.
     61   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
     62             repr(Twine("hi").concat(Twine())));
     63   EXPECT_EQ("(Twine cstring:\"hi\" empty)",
     64             repr(Twine().concat(Twine("hi"))));
     65 
     66   // Concatenation of unary ropes.
     67   EXPECT_EQ("(Twine cstring:\"a\" cstring:\"b\")",
     68             repr(Twine("a").concat(Twine("b"))));
     69 
     70   // Concatenation of other ropes.
     71   EXPECT_EQ("(Twine rope:(Twine cstring:\"a\" cstring:\"b\") cstring:\"c\")",
     72             repr(Twine("a").concat(Twine("b")).concat(Twine("c"))));
     73   EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine cstring:\"b\" cstring:\"c\"))",
     74             repr(Twine("a").concat(Twine("b").concat(Twine("c")))));
     75 }
     76 
     77 TEST(TwineTest, toNullTerminatedStringRef) {
     78   SmallString<8> storage;
     79   EXPECT_EQ(0, *Twine("hello").toNullTerminatedStringRef(storage).end());
     80   EXPECT_EQ(0,
     81            *Twine(StringRef("hello")).toNullTerminatedStringRef(storage).end());
     82 }
     83 
     84   // I suppose linking in the entire code generator to add a unit test to check
     85   // the code size of the concat operation is overkill... :)
     86 
     87 } // end anonymous namespace
     88