123 lines
4.3 KiB
C++
123 lines
4.3 KiB
C++
//===- TwineTest.cpp - Twine unit tests -----------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/ADT/Twine.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
#include "llvm/Support/FormatAdapters.h"
|
|
#include "llvm/Support/FormatVariadic.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include "gtest/gtest.h"
|
|
using namespace llvm;
|
|
|
|
namespace {
|
|
|
|
std::string repr(const Twine &Value) {
|
|
std::string res;
|
|
llvm::raw_string_ostream OS(res);
|
|
Value.printRepr(OS);
|
|
return OS.str();
|
|
}
|
|
|
|
TEST(TwineTest, Construction) {
|
|
EXPECT_EQ("", Twine().str());
|
|
EXPECT_EQ("hi", Twine("hi").str());
|
|
EXPECT_EQ("hi", Twine(std::string("hi")).str());
|
|
EXPECT_EQ("hi", Twine(StringRef("hi")).str());
|
|
EXPECT_EQ("hi", Twine(StringRef(std::string("hi"))).str());
|
|
EXPECT_EQ("hi", Twine(StringRef("hithere", 2)).str());
|
|
EXPECT_EQ("hi", Twine(SmallString<4>("hi")).str());
|
|
EXPECT_EQ("hi", Twine(formatv("{0}", "hi")).str());
|
|
}
|
|
|
|
TEST(TwineTest, Numbers) {
|
|
EXPECT_EQ("123", Twine(123U).str());
|
|
EXPECT_EQ("123", Twine(123).str());
|
|
EXPECT_EQ("-123", Twine(-123).str());
|
|
EXPECT_EQ("123", Twine(123).str());
|
|
EXPECT_EQ("-123", Twine(-123).str());
|
|
|
|
EXPECT_EQ("7b", Twine::utohexstr(123).str());
|
|
}
|
|
|
|
TEST(TwineTest, Characters) {
|
|
EXPECT_EQ("x", Twine('x').str());
|
|
EXPECT_EQ("x", Twine(static_cast<unsigned char>('x')).str());
|
|
EXPECT_EQ("x", Twine(static_cast<signed char>('x')).str());
|
|
}
|
|
|
|
TEST(TwineTest, Concat) {
|
|
// Check verse repr, since we care about the actual representation not just
|
|
// the result.
|
|
|
|
// Concat with null.
|
|
EXPECT_EQ("(Twine null empty)",
|
|
repr(Twine("hi").concat(Twine::createNull())));
|
|
EXPECT_EQ("(Twine null empty)",
|
|
repr(Twine::createNull().concat(Twine("hi"))));
|
|
|
|
// Concat with empty.
|
|
EXPECT_EQ("(Twine cstring:\"hi\" empty)",
|
|
repr(Twine("hi").concat(Twine())));
|
|
EXPECT_EQ("(Twine cstring:\"hi\" empty)",
|
|
repr(Twine().concat(Twine("hi"))));
|
|
EXPECT_EQ("(Twine smallstring:\"hi\" empty)",
|
|
repr(Twine().concat(Twine(SmallString<5>("hi")))));
|
|
EXPECT_EQ("(Twine formatv:\"howdy\" empty)",
|
|
repr(Twine(formatv("howdy")).concat(Twine())));
|
|
EXPECT_EQ("(Twine formatv:\"howdy\" empty)",
|
|
repr(Twine().concat(Twine(formatv("howdy")))));
|
|
EXPECT_EQ("(Twine smallstring:\"hey\" cstring:\"there\")",
|
|
repr(Twine(SmallString<7>("hey")).concat(Twine("there"))));
|
|
|
|
// Concatenation of unary ropes.
|
|
EXPECT_EQ("(Twine cstring:\"a\" cstring:\"b\")",
|
|
repr(Twine("a").concat(Twine("b"))));
|
|
|
|
// Concatenation of other ropes.
|
|
EXPECT_EQ("(Twine rope:(Twine cstring:\"a\" cstring:\"b\") cstring:\"c\")",
|
|
repr(Twine("a").concat(Twine("b")).concat(Twine("c"))));
|
|
EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine cstring:\"b\" cstring:\"c\"))",
|
|
repr(Twine("a").concat(Twine("b").concat(Twine("c")))));
|
|
EXPECT_EQ("(Twine cstring:\"a\" rope:(Twine smallstring:\"b\" cstring:\"c\"))",
|
|
repr(Twine("a").concat(Twine(SmallString<3>("b")).concat(Twine("c")))));
|
|
}
|
|
|
|
TEST(TwineTest, toNullTerminatedStringRef) {
|
|
SmallString<8> storage;
|
|
EXPECT_EQ(0, *Twine("hello").toNullTerminatedStringRef(storage).end());
|
|
EXPECT_EQ(0,
|
|
*Twine(StringRef("hello")).toNullTerminatedStringRef(storage).end());
|
|
EXPECT_EQ(0, *Twine(SmallString<11>("hello"))
|
|
.toNullTerminatedStringRef(storage)
|
|
.end());
|
|
EXPECT_EQ(0, *Twine(formatv("{0}{1}", "how", "dy"))
|
|
.toNullTerminatedStringRef(storage)
|
|
.end());
|
|
}
|
|
|
|
TEST(TwineTest, LazyEvaluation) {
|
|
struct formatter : FormatAdapter<int> {
|
|
explicit formatter(int &Count) : FormatAdapter(0), Count(Count) {}
|
|
int &Count;
|
|
|
|
void format(raw_ostream &OS, StringRef Style) override { ++Count; }
|
|
};
|
|
|
|
int Count = 0;
|
|
formatter Formatter(Count);
|
|
(void)Twine(formatv("{0}", Formatter));
|
|
EXPECT_EQ(0, Count);
|
|
(void)Twine(formatv("{0}", Formatter)).str();
|
|
EXPECT_EQ(1, Count);
|
|
}
|
|
|
|
// I suppose linking in the entire code generator to add a unit test to check
|
|
// the code size of the concat operation is overkill... :)
|
|
|
|
} // end anonymous namespace
|