removed some useless spaces; not very important, I know :)
[senf.git] / Utils / intrusive_refcount.test.cc
1 // $Id$
2 //
3 // Copyright (C) 2006
4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
6 //     Stefan Bund <g0dil@berlios.de>
7 //
8 // This program is free software; you can redistribute it and/or modify
9 // it under the terms of the GNU General Public License as published by
10 // the Free Software Foundation; either version 2 of the License, or
11 // (at your option) any later version.
12 //
13 // This program is distributed in the hope that it will be useful,
14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16 // GNU General Public License for more details.
17 //
18 // You should have received a copy of the GNU General Public License
19 // along with this program; if not, write to the
20 // Free Software Foundation, Inc.,
21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
22
23 // Unit tests
24
25 //#include "intrusive_refcount.test.hh"
26 //#include "intrusive_refcount.test.ih"
27
28 // Custom includes
29 #include "intrusive_refcount.hh"
30 #include <boost/intrusive_ptr.hpp>
31
32 #include "../Utils/auto_unit_test.hh"
33 #include <boost/test/test_tools.hpp>
34
35 #define prefix_
36 ///////////////////////////////cc.p////////////////////////////////////////
37
38 namespace {
39     struct Tester
40         : public senf::intrusive_refcount
41     {
42         typedef boost::intrusive_ptr<Tester> ptr;
43
44         Tester() { ++counter; }
45         ~Tester() { --counter; }
46
47         static unsigned counter;
48     };
49
50     unsigned Tester::counter = 0;
51
52     struct TesterCustom
53         : public senf::intrusive_refcount_t<TesterCustom>
54     {
55         typedef boost::intrusive_ptr<TesterCustom> ptr;
56         typedef senf::intrusive_refcount_t<TesterCustom> super;
57
58         TesterCustom() { ++counter; }
59         ~TesterCustom() { --counter; }
60
61         void add_ref() { super::add_ref(); ++refs; }
62         bool release() { --refs; super::release(); return false; }
63
64         static unsigned counter;
65         static unsigned refs;
66     };
67
68     unsigned TesterCustom::counter = 0;
69     unsigned TesterCustom::refs = 0;
70 }
71
72 BOOST_AUTO_UNIT_TEST(intrusive_refcount)
73 {
74     BOOST_CHECK_EQUAL(Tester::counter,0u);
75
76     Tester::ptr p (new Tester);
77     BOOST_CHECK_EQUAL(Tester::counter,1u);
78     BOOST_CHECK_EQUAL(p->refcount(),1u);
79     BOOST_CHECK_EQUAL(p->is_shared(),false);
80
81     {
82         Tester::ptr pp (p);
83         BOOST_CHECK_EQUAL(Tester::counter,1u);
84         BOOST_CHECK_EQUAL(p->refcount(),2u);
85         BOOST_CHECK_EQUAL(p->is_shared(),true);
86     }
87
88     BOOST_CHECK_EQUAL(Tester::counter,1u);
89     BOOST_CHECK_EQUAL(p->refcount(),1u);
90     BOOST_CHECK_EQUAL(p->is_shared(),false);
91
92     p = 0;
93     BOOST_CHECK_EQUAL(Tester::counter,0u);
94 }
95
96 BOOST_AUTO_UNIT_TEST(intrusive_refcount_t)
97 {
98     BOOST_CHECK_EQUAL(TesterCustom::counter,0u);
99     BOOST_CHECK_EQUAL(TesterCustom::refs,0u);
100
101     TesterCustom::ptr p (new TesterCustom);
102     BOOST_CHECK_EQUAL(TesterCustom::counter,1u);
103     BOOST_CHECK_EQUAL(p->refcount(),1u);
104     BOOST_CHECK_EQUAL(p->is_shared(),false);
105     BOOST_CHECK_EQUAL(TesterCustom::refs,1u);
106     
107
108     {
109         TesterCustom::ptr pp (p);
110         BOOST_CHECK_EQUAL(TesterCustom::counter,1u);
111         BOOST_CHECK_EQUAL(p->refcount(),2u);
112         BOOST_CHECK_EQUAL(p->is_shared(),true);
113         BOOST_CHECK_EQUAL(TesterCustom::refs,2u);
114     }
115
116     BOOST_CHECK_EQUAL(TesterCustom::counter,1u);
117     BOOST_CHECK_EQUAL(p->refcount(),1u);
118     BOOST_CHECK_EQUAL(p->is_shared(),false);
119     BOOST_CHECK_EQUAL(TesterCustom::refs,1u);
120
121     {
122         TesterCustom * pp (p.get());
123         p = 0;
124         BOOST_CHECK_EQUAL(TesterCustom::counter,1u);
125         BOOST_CHECK_EQUAL(TesterCustom::refs,0u);
126         // The TesterCustom leaks ...
127         delete pp;
128         BOOST_CHECK_EQUAL(TesterCustom::counter,0u);
129     }
130 }
131
132 ///////////////////////////////cc.e////////////////////////////////////////
133 #undef prefix_
134
135 \f
136 // Local Variables:
137 // mode: c++
138 // fill-column: 100
139 // c-file-style: "senf"
140 // indent-tabs-mode: nil
141 // ispell-local-dictionary: "american"
142 // compile-command: "scons -u test"
143 // comment-column: 40
144 // End: