SockBuffer-test.cpp

   1 #include "SockBuffer.hpp"
   2
   3 #include <fcntl.h>
   4
   5 #include <fstream>
   6 #include <iostream>
   7
   8 #include <fmt/format.h>
   9
  10 #include <glog/logging.h>
  11
  12 int main(int argc, char* argv[])
  13 {
  14   constexpr auto infile = "body.txt";
  15
  16   int fd_in = open(infile, O_RDONLY);
  17   PCHECK(fd_in != -1) << "Can't open file " << infile;
  18
  19   char outfile[] = "/tmp/SockBuffer-test-XXXXXX";
  20
  21   int fd_out = mkstemp(outfile);
  22   PCHECK(fd_out != -1);
  23
  24   auto read_hook = []() { std::cout << "read_hook\n"; };
  25
  26   boost::iostreams::stream<SockBuffer> iostream{fd_in,
  27                                                 fd_out,
  28                                                 read_hook,
  29                                                 std::chrono::seconds(10),
  30                                                 std::chrono::seconds(10),
  31                                                 std::chrono::seconds(1)};
  32
  33   std::string line;
  34   while (std::getline(iostream, line)) {
  35     iostream << line << '\n';
  36   }
  37   iostream.clear(); // unset eof bit, if set will short-circut flush
  38   iostream << std::flush;
  39
  40   auto const diff_cmd{fmt::format("diff {} {}", infile, outfile)};
  41   CHECK_EQ(system(diff_cmd.c_str()), 0);
  42
  43   PCHECK(!unlink(outfile)) << "unlink failed for " << outfile;
  44
  45   std::cout << "sizeof(SockBuffer) == " << sizeof(SockBuffer) << '\n';
  46 }