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 // UNSUPPORTED: c++98, c++03, c++11
11
12 // <experimental/iterator>
13 //
14 // template <class _Delim, class _CharT = char, class _Traits = char_traits<_CharT>>
15 // class ostream_joiner;
16 //
17 // template<typename T>
18 // ostream_joiner & operator=(const T&)
19 //
20
21 #include <experimental/iterator>
22 #include <iostream>
23 #include <sstream>
24 #include <cassert>
25
26 #include "test_macros.h"
27 #include "test_iterators.h"
28
29 struct mutating_delimiter {
mutating_delimitermutating_delimiter30 mutating_delimiter(char c = ' ') : c_(c) {}
getmutating_delimiter31 char get () const { return c_++; }
32 mutable char c_;
33 };
34
35 template<class _CharT, class _Traits>
36 std::basic_ostream<_CharT, _Traits>&
operator <<(std::basic_ostream<_CharT,_Traits> & os,const mutating_delimiter & d)37 operator<<(std::basic_ostream<_CharT, _Traits>& os, const mutating_delimiter &d)
38 { return os << d.get(); }
39
40
41 struct mutating_delimiter2 { // same as above, w/o the const and the mutable
mutating_delimiter2mutating_delimiter242 mutating_delimiter2(char c = ' ') : c_(c) {}
getmutating_delimiter243 char get () { return c_++; }
44 char c_;
45 };
46
47 template<class _CharT, class _Traits>
48 std::basic_ostream<_CharT, _Traits>&
operator <<(std::basic_ostream<_CharT,_Traits> & os,mutating_delimiter2 & d)49 operator<<(std::basic_ostream<_CharT, _Traits>& os, mutating_delimiter2 &d)
50 { return os << d.get(); }
51
52
53 namespace exp = std::experimental;
54
55 template <class Delim, class Iter, class CharT = char, class Traits = std::char_traits<CharT>>
test(Delim && d,Iter first,Iter last,const CharT * expected)56 void test (Delim &&d, Iter first, Iter last, const CharT *expected ) {
57 typedef exp::ostream_joiner<typename std::decay<Delim>::type, CharT, Traits> Joiner;
58
59 static_assert((std::is_copy_constructible<Joiner>::value == std::is_copy_constructible<typename std::decay<Delim>::type>::value), "" );
60 static_assert((std::is_move_constructible<Joiner>::value == std::is_move_constructible<typename std::decay<Delim>::type>::value), "" );
61 static_assert((std::is_copy_assignable<Joiner> ::value == std::is_copy_assignable<typename std::decay<Delim>::type> ::value), "" );
62 static_assert((std::is_move_assignable<Joiner> ::value == std::is_move_assignable<typename std::decay<Delim>::type> ::value), "" );
63
64 std::basic_stringstream<CharT, Traits> sstream;
65 Joiner joiner(sstream, d);
66 while (first != last)
67 *joiner++ = *first++;
68 assert(sstream.str() == expected);
69 }
70
main()71 int main () {
72 {
73 const char chars[] = "0123456789";
74 const int ints [] = { 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
75
76 test('X', chars, chars+10, "0X1X2X3X4X5X6X7X8X9");
77 test('x', ints, ints+10, "10x11x12x13x14x15x16x17x18x19");
78 test('X', input_iterator<const char*>(chars), input_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9");
79 test('x', input_iterator<const int*>(ints), input_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19");
80 test('X', forward_iterator<const char*>(chars), forward_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9");
81 test('x', forward_iterator<const int*>(ints), forward_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19");
82 test('X', random_access_iterator<const char*>(chars), random_access_iterator<const char*>(chars+10), "0X1X2X3X4X5X6X7X8X9");
83 test('x', random_access_iterator<const int*>(ints), random_access_iterator<const int*>(ints+10), "10x11x12x13x14x15x16x17x18x19");
84
85 test("Z", chars, chars+10, "0Z1Z2Z3Z4Z5Z6Z7Z8Z9");
86 test("z", ints, ints+10, "10z11z12z13z14z15z16z17z18z19");
87
88 test<char, const char *, wchar_t> ('X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9");
89 test<char, const int *, wchar_t> ('x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19");
90 // test<char, const char *, char16_t>('X', chars, chars+10, u"0X1X2X3X4X5X6X7X8X9");
91 // test<char, const int *, char16_t>('x', ints, ints+10, u"10x11x12x13x14x15x16x17x18x19");
92 // test<char, const char *, char32_t>('X', chars, chars+10, U"0X1X2X3X4X5X6X7X8X9");
93 // test<char, const int *, char32_t>('x', ints, ints+10, U"10x11x12x13x14x15x16x17x18x19");
94
95 test(mutating_delimiter(), chars, chars+10, "0 1!2\"3#4$5%6&7'8(9");
96 test(mutating_delimiter2(), chars, chars+10, "0 1!2\"3#4$5%6&7'8(9");
97 }
98
99 {
100 const wchar_t chars[] = L"0123456789";
101 const int ints [] = { 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 };
102 test(L'X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9");
103 test(L'x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19");
104 test(L'X', input_iterator<const wchar_t*>(chars), input_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9");
105 test(L'x', input_iterator<const int*>(ints), input_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19");
106 test(L'X', forward_iterator<const wchar_t*>(chars), forward_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9");
107 test(L'x', forward_iterator<const int*>(ints), forward_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19");
108 test(L'X', random_access_iterator<const wchar_t*>(chars), random_access_iterator<const wchar_t*>(chars+10), L"0X1X2X3X4X5X6X7X8X9");
109 test(L'x', random_access_iterator<const int*>(ints), random_access_iterator<const int*>(ints+10), L"10x11x12x13x14x15x16x17x18x19");
110
111 test(L"Z", chars, chars+10, L"0Z1Z2Z3Z4Z5Z6Z7Z8Z9");
112 test(L"z", ints, ints+10, L"10z11z12z13z14z15z16z17z18z19");
113
114 test<char, const wchar_t *, wchar_t> ('X', chars, chars+10, L"0X1X2X3X4X5X6X7X8X9");
115 test<char, const int *, wchar_t> ('x', ints, ints+10, L"10x11x12x13x14x15x16x17x18x19");
116
117 test(mutating_delimiter(), chars, chars+10, L"0 1!2\"3#4$5%6&7'8(9");
118 }
119
120 }
121