//===-- 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);
}