alloc_rfunction.pass.cpp revision 1b92188a82b01e76ac6e8ad5f997293c2a078adc 
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// <functional>
11
12// class function<R(ArgTypes...)>
13
14// template<class A> function(allocator_arg_t, const A&, function&&);
15
16#include <functional>
17#include <cassert>
18
19#include "test_allocator.h"
20
21int new_called = 0;
22
23void* operator new(std::size_t s) throw(std::bad_alloc)
24{
25    ++new_called;
26    return std::malloc(s);
27}
28
29void  operator delete(void* p) throw()
30{
31    --new_called;
32    std::free(p);
33}
34
35class A
36{
37    int data_[10];
38public:
39    static int count;
40
41    A()
42    {
43        ++count;
44        for (int i = 0; i < 10; ++i)
45            data_[i] = i;
46    }
47
48    A(const A&) {++count;}
49
50    ~A() {--count;}
51
52    int operator()(int i) const
53    {
54        for (int j = 0; j < 10; ++j)
55            i += data_[j];
56        return i;
57    }
58};
59
60int A::count = 0;
61
62int main()
63{
64#ifndef _LIBCPP_HAS_NO_RVALUE_REFERENCES
65    assert(new_called == 0);
66    {
67    std::function<int(int)> f = A();
68    assert(A::count == 1);
69    assert(new_called == 1);
70    assert(f.target<A>());
71    assert(f.target<int(*)(int)>() == 0);
72    std::function<int(int)> f2(std::allocator_arg, test_allocator<A>(), std::move(f));
73    assert(A::count == 1);
74    assert(new_called == 1);
75    assert(f2.target<A>());
76    assert(f2.target<int(*)(int)>() == 0);
77    assert(f.target<A>() == 0);
78    assert(f.target<int(*)(int)>() == 0);
79    }
80#endif  // _LIBCPP_HAS_NO_RVALUE_REFERENCES
81}
82