unittests/Support/raw_pwrite_stream_test.cpp

   1 //===- raw_pwrite_stream_test.cpp - raw_pwrite_stream tests ---------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/ADT/SmallString.h"
  11 #include "llvm/Config/llvm-config.h"
  12 #include "llvm/Support/FileSystem.h"
  13 #include "llvm/Support/FileUtilities.h"
  14 #include "llvm/Support/raw_ostream.h"
  15 #include "gtest/gtest.h"
  16
  17 using namespace llvm;
  18
  19 #define ASSERT_NO_ERROR(x)                                                     \
  20   if (std::error_code ASSERT_NO_ERROR_ec = x) {                                \
  21     SmallString<128> MessageStorage;                                           \
  22     raw_svector_ostream Message(MessageStorage);                               \
  23     Message << #x ": did not return errc::success.\n"                          \
  24             << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n"          \
  25             << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n";      \
  26     GTEST_FATAL_FAILURE_(MessageStorage.c_str());                              \
  27   } else {                                                                     \
  28   }
  29
  30 namespace {
  31
  32 TEST(raw_pwrite_ostreamTest, TestSVector) {
  33   SmallVector<char, 0> Buffer;
  34   raw_svector_ostream OS(Buffer);
  35   OS << "abcd";
  36   StringRef Test = "test";
  37   OS.pwrite(Test.data(), Test.size(), 0);
  38   EXPECT_EQ(Test, OS.str());
  39
  40 #ifdef GTEST_HAS_DEATH_TEST
  41 #ifndef NDEBUG
  42   EXPECT_DEATH(OS.pwrite("12345", 5, 0),
  43                "We don't support extending the stream");
  44 #endif
  45 #endif
  46 }
  47
  48 #ifdef _WIN32
  49 #define setenv(name, var, ignore) _putenv_s(name, var)
  50 #endif
  51
  52 TEST(raw_pwrite_ostreamTest, TestFD) {
  53   SmallString<64> Path;
  54   int FD;
  55
  56   // If we want to clean up from a death test, we have to remove the file from
  57   // the parent process. Have the parent create the file, pass it via
  58   // environment variable to the child, let the child crash, and then remove it
  59   // in the parent.
  60   const char *ParentPath = getenv("RAW_PWRITE_TEST_FILE");
  61   if (ParentPath) {
  62     Path = ParentPath;
  63     ASSERT_NO_ERROR(sys::fs::openFileForRead(Path, FD));
  64   } else {
  65     ASSERT_NO_ERROR(sys::fs::createTemporaryFile("foo", "bar", FD, Path));
  66     setenv("RAW_PWRITE_TEST_FILE", Path.c_str(), true);
  67   }
  68   FileRemover Cleanup(Path);
  69
  70   raw_fd_ostream OS(FD, true);
  71   OS << "abcd";
  72   StringRef Test = "test";
  73   OS.pwrite(Test.data(), Test.size(), 0);
  74   OS.pwrite(Test.data(), Test.size(), 0);
  75
  76 #ifdef GTEST_HAS_DEATH_TEST
  77 #ifndef NDEBUG
  78   EXPECT_DEATH(OS.pwrite("12345", 5, 0),
  79                "We don't support extending the stream");
  80 #endif
  81 #endif
  82 }
  83
  84 #ifdef LLVM_ON_UNIX
  85 TEST(raw_pwrite_ostreamTest, TestDevNull) {
  86   int FD;
  87   sys::fs::openFileForWrite("/dev/null", FD, sys::fs::CD_OpenExisting);
  88   raw_fd_ostream OS(FD, true);
  89   OS << "abcd";
  90   StringRef Test = "test";
  91   OS.pwrite(Test.data(), Test.size(), 0);
  92   OS.pwrite(Test.data(), Test.size(), 0);
  93 }
  94 #endif
  95 }