libstdc++-v3/testsuite/27_io/basic_filebuf/close/char/12790-4.cc

   1 // Copyright (C) 2003 Free Software Foundation, Inc.
   2 //
   3 // This file is part of the GNU ISO C++ Library.  This library is free
   4 // software; you can redistribute it and/or modify it under the
   5 // terms of the GNU General Public License as published by the
   6 // Free Software Foundation; either version 2, or (at your option)
   7 // any later version.
   8
   9 // This library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13
  14 // You should have received a copy of the GNU General Public License along
  15 // with this library; see the file COPYING.  If not, write to the Free
  16 // Software Foundation, 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
  17 // USA.
  18
  19 // 27.8.1.3 filebuf member functions
  20
  21 #include <locale>
  22 #include <fstream>
  23 #include <testsuite_hooks.h>
  24
  25 class Cvt : public std::codecvt<char, char, std::mbstate_t>
  26 {
  27 public:
  28   mutable bool unshift_called;
  29
  30   Cvt()
  31   : unshift_called(false)
  32   { }
  33
  34 protected:
  35   bool
  36   do_always_noconv() const throw()
  37   { return false; }
  38
  39   int
  40   do_encoding() const throw()
  41   { return -1; }
  42
  43   std::codecvt_base::result
  44   do_unshift(std::mbstate_t&, char* to, char*, char*& to_next) const
  45   {
  46     unshift_called = true;
  47     to_next = to;
  48     return std::codecvt_base::ok;
  49   }
  50 };
  51
  52 // libstdc++/12790
  53 // basic_filebuf::close() should call codecvt::unshift()
  54 void test01()
  55 {
  56   using namespace std;
  57
  58   bool test __attribute__((unused)) = true;
  59   const char* name = "tmp_close_12790";
  60
  61   Cvt* cvt = new Cvt;
  62   locale loc(locale::classic(), cvt);
  63
  64   filebuf fb;
  65   fb.pubimbue(loc);
  66
  67   fb.open(name, ios_base::out);
  68   fb.sputc('a');
  69   fb.pubsync(); // Does not call unshift()
  70
  71   VERIFY( !cvt->unshift_called );
  72   fb.close();
  73   VERIFY( cvt->unshift_called );
  74 }
  75
  76 int main()
  77 {
  78   test01();
  79   return 0;
  80 }