blob: 9664b42631c0b475bb723677212f8b4acda3cfa3 [file] [log] [blame]
//===-- Unittests for StringStream ----------------------------------------===//
//
// 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 "src/__support/CPP/ArrayRef.h"
#include "src/__support/CPP/stringstream.h"
#include "utils/UnitTest/Test.h"
using __llvm_libc::cpp::MutableArrayRef;
using __llvm_libc::cpp::StringStream;
TEST(LlvmLibcStringStreamTest, Simple) {
char buf[256];
MutableArrayRef<char> bufref(reinterpret_cast<void *>(buf), 256);
StringStream ss1(bufref);
ss1 << "Hello, Stream - " << int(123) << StringStream::ENDS;
ASSERT_FALSE(ss1.overflow());
ASSERT_STREQ(ss1.str().data(), "Hello, Stream - 123");
StringStream ss2(bufref);
ss2 << 'a' << 'b' << 'c' << StringStream::ENDS;
ASSERT_FALSE(ss2.overflow());
ASSERT_STREQ(ss2.str().data(), "abc");
}
TEST(LlvmLibcStringStreamTest, Overflow) {
constexpr size_t BUFSIZE = 8;
char buf[BUFSIZE];
MutableArrayRef<char> bufref(reinterpret_cast<void *>(buf), BUFSIZE);
StringStream ss1(bufref);
ss1 << "Hello, Stream - " << int(123) << StringStream::ENDS;
ASSERT_TRUE(ss1.overflow());
ASSERT_EQ(ss1.str().size(), BUFSIZE);
StringStream ss2(bufref);
ss2 << "7777777";
ASSERT_FALSE(ss2.overflow());
ASSERT_EQ(ss2.str().size(), size_t(7));
ss2 << "8";
ASSERT_FALSE(ss2.overflow());
ASSERT_EQ(ss2.str().size(), size_t(8));
ss2 << StringStream::ENDS;
ASSERT_TRUE(ss2.overflow());
ASSERT_EQ(ss2.str().size(), BUFSIZE);
}