95 lines
3.0 KiB
C++
95 lines
3.0 KiB
C++
|
//===- raw_pwrite_stream_test.cpp - raw_pwrite_stream 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/SmallString.h"
|
||
|
#include "llvm/Config/llvm-config.h"
|
||
|
#include "llvm/Support/FileSystem.h"
|
||
|
#include "llvm/Support/FileUtilities.h"
|
||
|
#include "llvm/Support/raw_ostream.h"
|
||
|
#include "gtest/gtest.h"
|
||
|
|
||
|
using namespace llvm;
|
||
|
|
||
|
#define ASSERT_NO_ERROR(x) \
|
||
|
if (std::error_code ASSERT_NO_ERROR_ec = x) { \
|
||
|
SmallString<128> MessageStorage; \
|
||
|
raw_svector_ostream Message(MessageStorage); \
|
||
|
Message << #x ": did not return errc::success.\n" \
|
||
|
<< "error number: " << ASSERT_NO_ERROR_ec.value() << "\n" \
|
||
|
<< "error message: " << ASSERT_NO_ERROR_ec.message() << "\n"; \
|
||
|
GTEST_FATAL_FAILURE_(MessageStorage.c_str()); \
|
||
|
} else { \
|
||
|
}
|
||
|
|
||
|
namespace {
|
||
|
|
||
|
TEST(raw_pwrite_ostreamTest, TestSVector) {
|
||
|
SmallVector<char, 0> Buffer;
|
||
|
raw_svector_ostream OS(Buffer);
|
||
|
OS << "abcd";
|
||
|
StringRef Test = "test";
|
||
|
OS.pwrite(Test.data(), Test.size(), 0);
|
||
|
EXPECT_EQ(Test, OS.str());
|
||
|
|
||
|
#ifdef GTEST_HAS_DEATH_TEST
|
||
|
#ifndef NDEBUG
|
||
|
EXPECT_DEATH(OS.pwrite("12345", 5, 0),
|
||
|
"We don't support extending the stream");
|
||
|
#endif
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
#ifdef _WIN32
|
||
|
#define setenv(name, var, ignore) _putenv_s(name, var)
|
||
|
#endif
|
||
|
|
||
|
TEST(raw_pwrite_ostreamTest, TestFD) {
|
||
|
SmallString<64> Path;
|
||
|
int FD;
|
||
|
|
||
|
// If we want to clean up from a death test, we have to remove the file from
|
||
|
// the parent process. Have the parent create the file, pass it via
|
||
|
// environment variable to the child, let the child crash, and then remove it
|
||
|
// in the parent.
|
||
|
const char *ParentPath = getenv("RAW_PWRITE_TEST_FILE");
|
||
|
if (ParentPath) {
|
||
|
Path = ParentPath;
|
||
|
ASSERT_NO_ERROR(sys::fs::openFileForRead(Path, FD));
|
||
|
} else {
|
||
|
ASSERT_NO_ERROR(sys::fs::createTemporaryFile("foo", "bar", FD, Path));
|
||
|
setenv("RAW_PWRITE_TEST_FILE", Path.c_str(), true);
|
||
|
}
|
||
|
FileRemover Cleanup(Path);
|
||
|
|
||
|
raw_fd_ostream OS(FD, true);
|
||
|
OS << "abcd";
|
||
|
StringRef Test = "test";
|
||
|
OS.pwrite(Test.data(), Test.size(), 0);
|
||
|
OS.pwrite(Test.data(), Test.size(), 0);
|
||
|
|
||
|
#ifdef GTEST_HAS_DEATH_TEST
|
||
|
#ifndef NDEBUG
|
||
|
EXPECT_DEATH(OS.pwrite("12345", 5, 0),
|
||
|
"We don't support extending the stream");
|
||
|
#endif
|
||
|
#endif
|
||
|
}
|
||
|
|
||
|
#ifdef LLVM_ON_UNIX
|
||
|
TEST(raw_pwrite_ostreamTest, TestDevNull) {
|
||
|
int FD;
|
||
|
sys::fs::openFileForWrite("/dev/null", FD, sys::fs::CD_OpenExisting);
|
||
|
raw_fd_ostream OS(FD, true);
|
||
|
OS << "abcd";
|
||
|
StringRef Test = "test";
|
||
|
OS.pwrite(Test.data(), Test.size(), 0);
|
||
|
OS.pwrite(Test.data(), Test.size(), 0);
|
||
|
}
|
||
|
#endif
|
||
|
}
|