libstdc++-v3/testsuite/21_strings/basic_string_view/inserters/wchar_t/2.cc

   1 // { dg-options "-std=gnu++17" }
   2 // { dg-require-fileio "" }
   3
   4 // Copyright (C) 2013-2020 Free Software Foundation, Inc.
   5 //
   6 // This file is part of the GNU ISO C++ Library.  This library is free
   7 // software; you can redistribute it and/or modify it under the
   8 // terms of the GNU General Public License as published by the
   9 // Free Software Foundation; either version 3, or (at your option)
  10 // any later version.
  11
  12 // This library is distributed in the hope that it will be useful,
  13 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 // GNU General Public License for more details.
  16
  17 // You should have received a copy of the GNU General Public License along
  18 // with this library; see the file COPYING3.  If not see
  19 // <http://www.gnu.org/licenses/>.
  20
  21 // inserters
  22
  23 // NB: This file is predicated on sstreams, istreams, and ostreams
  24 // working, not to mention other major details like char_traits, and
  25 // all of the string_view class.
  26
  27 #include <string_view>
  28 #include <string>
  29 #include <fstream>
  30 #include <iostream>
  31 #include <testsuite_hooks.h>
  32
  33 // testing basic_filebuf::xsputn via stress testing with large string_views
  34 // based on a bug report libstdc++ 9
  35 // mode == out
  36 void
  37 test05(std::size_t size)
  38 {
  39   bool test = true;
  40
  41   const char filename[] = "inserters_extractors-2.txt";
  42   const wchar_t fillc = L'f';
  43   std::wofstream ofs(filename);
  44   std::wstring str(size, fillc);
  45   std::wstring_view strv(str);
  46
  47   // sanity checks
  48   VERIFY( str.size() == size );
  49   VERIFY( ofs.good() );
  50
  51   // stress test
  52   ofs << str << std::endl;
  53   if (!ofs.good())
  54     test = false;
  55
  56   ofs << str << std::endl;
  57   if (!ofs.good())
  58     test = false;
  59
  60   VERIFY( str.size() == size );
  61   VERIFY( ofs.good() );
  62
  63   ofs.close();
  64
  65   // sanity check on the written file
  66   std::wifstream ifs(filename);
  67   std::size_t count = 0;
  68   wchar_t c;
  69   while (count <= (2 * size) + 4)
  70     {
  71       ifs >> c;
  72       if (ifs.good() && c == fillc)
  73         {
  74           ++count;
  75           c = '0';
  76         }
  77       else
  78         break;
  79     }
  80
  81   VERIFY( count == 2 * size );
  82 }
  83
  84 int
  85 main()
  86 {
  87   test05(1);
  88   test05(1000);
  89   test05(10000);
  90
  91   return 0;
  92 }