1*5a83710eSEric Fiselier //===----------------------------------------------------------------------===//
2*5a83710eSEric Fiselier //
3*5a83710eSEric Fiselier //                     The LLVM Compiler Infrastructure
4*5a83710eSEric Fiselier //
5*5a83710eSEric Fiselier // This file is dual licensed under the MIT and the University of Illinois Open
6*5a83710eSEric Fiselier // Source Licenses. See LICENSE.TXT for details.
7*5a83710eSEric Fiselier //
8*5a83710eSEric Fiselier //===----------------------------------------------------------------------===//
9*5a83710eSEric Fiselier 
10*5a83710eSEric Fiselier // <string>
11*5a83710eSEric Fiselier 
12*5a83710eSEric Fiselier // size_type capacity() const;
13*5a83710eSEric Fiselier 
14*5a83710eSEric Fiselier #include <string>
15*5a83710eSEric Fiselier #include <cassert>
16*5a83710eSEric Fiselier 
17*5a83710eSEric Fiselier #include "test_allocator.h"
18*5a83710eSEric Fiselier #include "min_allocator.h"
19*5a83710eSEric Fiselier 
20*5a83710eSEric Fiselier template <class S>
21*5a83710eSEric Fiselier void
22*5a83710eSEric Fiselier test(S s)
23*5a83710eSEric Fiselier {
24*5a83710eSEric Fiselier     S::allocator_type::throw_after = 0;
25*5a83710eSEric Fiselier     try
26*5a83710eSEric Fiselier     {
27*5a83710eSEric Fiselier         while (s.size() < s.capacity())
28*5a83710eSEric Fiselier             s.push_back(typename S::value_type());
29*5a83710eSEric Fiselier         assert(s.size() == s.capacity());
30*5a83710eSEric Fiselier     }
31*5a83710eSEric Fiselier     catch (...)
32*5a83710eSEric Fiselier     {
33*5a83710eSEric Fiselier         assert(false);
34*5a83710eSEric Fiselier     }
35*5a83710eSEric Fiselier     S::allocator_type::throw_after = INT_MAX;
36*5a83710eSEric Fiselier }
37*5a83710eSEric Fiselier 
38*5a83710eSEric Fiselier int main()
39*5a83710eSEric Fiselier {
40*5a83710eSEric Fiselier     {
41*5a83710eSEric Fiselier     typedef std::basic_string<char, std::char_traits<char>, test_allocator<char> > S;
42*5a83710eSEric Fiselier     S s;
43*5a83710eSEric Fiselier     test(s);
44*5a83710eSEric Fiselier     s.assign(10, 'a');
45*5a83710eSEric Fiselier     s.erase(5);
46*5a83710eSEric Fiselier     test(s);
47*5a83710eSEric Fiselier     s.assign(100, 'a');
48*5a83710eSEric Fiselier     s.erase(50);
49*5a83710eSEric Fiselier     test(s);
50*5a83710eSEric Fiselier     }
51*5a83710eSEric Fiselier #if __cplusplus >= 201103L
52*5a83710eSEric Fiselier     {
53*5a83710eSEric Fiselier     typedef std::basic_string<char, std::char_traits<char>, min_allocator<char>> S;
54*5a83710eSEric Fiselier     S s;
55*5a83710eSEric Fiselier     assert(s.capacity() > 0);
56*5a83710eSEric Fiselier     }
57*5a83710eSEric Fiselier #endif
58*5a83710eSEric Fiselier }
59