imbue.pass.cpp revision c0d0cbad9ed434267a7af9531bdeeae52eb6d706 
1//===----------------------------------------------------------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is dual licensed under the MIT and the University of Illinois Open
6// Source Licenses. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10// <ios>
11
12// template <class charT, class traits> class basic_ios
13
14// locale imbue(const locale& loc);
15
16#include <ios>
17#include <streambuf>
18#include <cassert>
19
20#include "../../../../platform_support.h" // locale name macros
21
22struct testbuf
23    : public std::streambuf
24{
25};
26
27bool f1_called = false;
28bool f2_called = false;
29bool f3_called = false;
30
31void f1(std::ios_base::event ev, std::ios_base& stream, int index)
32{
33    if (ev == std::ios_base::imbue_event)
34    {
35        assert(!f1_called);
36        assert( f2_called);
37        assert( f3_called);
38        assert(stream.getloc().name() == LOCALE_en_US_UTF_8);
39        assert(index == 4);
40        f1_called = true;
41    }
42}
43
44void f2(std::ios_base::event ev, std::ios_base& stream, int index)
45{
46    if (ev == std::ios_base::imbue_event)
47    {
48        assert(!f1_called);
49        assert(!f2_called);
50        assert( f3_called);
51        assert(stream.getloc().name() == LOCALE_en_US_UTF_8);
52        assert(index == 5);
53        f2_called = true;
54    }
55}
56
57void f3(std::ios_base::event ev, std::ios_base& stream, int index)
58{
59    if (ev == std::ios_base::imbue_event)
60    {
61        assert(!f1_called);
62        assert(!f2_called);
63        assert(!f3_called);
64        assert(stream.getloc().name() == LOCALE_en_US_UTF_8);
65        assert(index == 6);
66        f3_called = true;
67    }
68}
69
70int main()
71{
72    {
73        std::ios ios(0);
74        ios.register_callback(f1, 4);
75        ios.register_callback(f2, 5);
76        ios.register_callback(f3, 6);
77        std::locale l = ios.imbue(std::locale(LOCALE_en_US_UTF_8));
78        assert(l.name() == std::string("C"));
79        assert(ios.getloc().name() == std::string(LOCALE_en_US_UTF_8));
80        assert(f1_called);
81        assert(f2_called);
82        assert(f3_called);
83    }
84    f1_called = false;
85    f2_called = false;
86    f3_called = false;
87    {
88        testbuf sb;
89        std::ios ios(&sb);
90        ios.register_callback(f1, 4);
91        ios.register_callback(f2, 5);
92        ios.register_callback(f3, 6);
93        std::locale l = ios.imbue(std::locale(LOCALE_en_US_UTF_8));
94        assert(l.name() == std::string("C"));
95        assert(ios.getloc().name() == std::string(LOCALE_en_US_UTF_8));
96        assert(sb.getloc().name() == std::string(LOCALE_en_US_UTF_8));
97        assert(f1_called);
98        assert(f2_called);
99        assert(f3_called);
100    }
101}
102