50 lines
1.2 KiB
C++
50 lines
1.2 KiB
C++
|
#include "flow/UnitTest.h"
|
||
|
|
#include "flow/ArenaAllocator.h"
|
||
|
|
#include "flow/ArenaString.h"
|
||
|
|
|
||
|
|
TEST_CASE("/flow/ArenaString") {
|
||
|
|
Arena arena;
|
||
|
|
ArenaAllocator<char> alloc(arena);
|
||
|
|
{
|
||
|
|
ArenaString s("1", alloc);
|
||
|
|
auto shortStrBuf = s.data();
|
||
|
|
s.assign(100, '1');
|
||
|
|
auto longStrBuf = s.data();
|
||
|
|
ASSERT_NE(shortStrBuf, longStrBuf);
|
||
|
|
ArenaString t = s;
|
||
|
|
auto copiedStrBuf = t.data();
|
||
|
|
ASSERT_NE(copiedStrBuf, longStrBuf);
|
||
|
|
}
|
||
|
|
{
|
||
|
|
ArenaString s(alloc);
|
||
|
|
s.assign(100, 'a');
|
||
|
|
ArenaString t(100, 'a', alloc);
|
||
|
|
ASSERT(s == t);
|
||
|
|
}
|
||
|
|
{
|
||
|
|
// Default construction of string does not specify an allocator, and Arena by extension.
|
||
|
|
// Any modification that requires allocation will throw bad_allocator() when assigning beyond
|
||
|
|
// short-string-optimized length.
|
||
|
|
ArenaString s;
|
||
|
|
bool hit = false;
|
||
|
|
try {
|
||
|
|
s.assign(100, 'a');
|
||
|
|
} catch (Error& e) {
|
||
|
|
hit = true;
|
||
|
|
ASSERT_EQ(e.code(), error_code_bad_allocator);
|
||
|
|
}
|
||
|
|
ASSERT(hit);
|
||
|
|
}
|
||
|
|
{
|
||
|
|
// string_view may be used to bridge strings with different allocators
|
||
|
|
ArenaString s(100, 'a', alloc);
|
||
|
|
std::string_view sv(s);
|
||
|
|
std::string s2(sv);
|
||
|
|
std::string_view sv2(s2);
|
||
|
|
ASSERT(sv == sv2);
|
||
|
|
}
|
||
|
|
return Void();
|
||
|
|
}
|
||
|
|
|
||
|
|
void forceLinkArenaStringTests() {}
|