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

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