5706f1e6f6
Summary: The EncodingError test ensures that trying to encode a multibyte wchar with a given codepage fails. If setlocale() fails, the encoding is performed using the current locale, which may or may not fail. This patch asserts that both setlocale() operations are successful, as well as falling back to a widely available unibyte encoding for non-Windows systems. <rdar://problem/33782806> Reviewers: zturner, labath, lhames Reviewed By: zturner Subscribers: lldb-commits Differential Revision: https://reviews.llvm.org/D36496 llvm-svn: 310499
68 lines
1.9 KiB
C++
68 lines
1.9 KiB
C++
//===-- VASprintfTest.cpp ---------------------------------------*- C++ -*-===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lldb/Utility/VASPrintf.h"
|
|
#include "llvm/ADT/SmallString.h"
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <locale.h>
|
|
|
|
#if defined (_WIN32)
|
|
#define TEST_ENCODING ".932" // On Windows, test codepage 932
|
|
#else
|
|
#define TEST_ENCODING "C" // ...otherwise, any widely available uni-byte LC
|
|
#endif
|
|
|
|
using namespace lldb_private;
|
|
using namespace llvm;
|
|
|
|
static bool Sprintf(llvm::SmallVectorImpl<char> &Buffer, const char *Fmt, ...) {
|
|
va_list args;
|
|
va_start(args, Fmt);
|
|
bool Result = VASprintf(Buffer, Fmt, args);
|
|
va_end(args);
|
|
return Result;
|
|
}
|
|
|
|
TEST(VASprintfTest, NoBufferResize) {
|
|
std::string TestStr("small");
|
|
|
|
llvm::SmallString<32> BigBuffer;
|
|
ASSERT_TRUE(Sprintf(BigBuffer, "%s", TestStr.c_str()));
|
|
EXPECT_STREQ(TestStr.c_str(), BigBuffer.c_str());
|
|
EXPECT_EQ(TestStr.size(), BigBuffer.size());
|
|
}
|
|
|
|
TEST(VASprintfTest, BufferResize) {
|
|
std::string TestStr("bigger");
|
|
llvm::SmallString<4> SmallBuffer;
|
|
ASSERT_TRUE(Sprintf(SmallBuffer, "%s", TestStr.c_str()));
|
|
EXPECT_STREQ(TestStr.c_str(), SmallBuffer.c_str());
|
|
EXPECT_EQ(TestStr.size(), SmallBuffer.size());
|
|
}
|
|
|
|
TEST(VASprintfTest, EncodingError) {
|
|
// Save the current locale first.
|
|
std::string Current(::setlocale(LC_ALL, nullptr));
|
|
|
|
// Ensure tested locale is successfully set
|
|
ASSERT_TRUE(setlocale(LC_ALL, TEST_ENCODING));
|
|
|
|
wchar_t Invalid[2];
|
|
Invalid[0] = 0x100;
|
|
Invalid[1] = 0;
|
|
llvm::SmallString<32> Buffer;
|
|
EXPECT_FALSE(Sprintf(Buffer, "%ls", Invalid));
|
|
EXPECT_EQ("<Encoding error>", Buffer);
|
|
|
|
// Ensure we've restored the original locale once tested
|
|
ASSERT_TRUE(setlocale(LC_ALL, Current.c_str()));
|
|
}
|