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"
14using namespace llvm;
15
16namespace {
17
18std::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
25TEST(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
35TEST(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
45TEST(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
51TEST(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
84TEST(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
98