Packets: Migrate VariantParser to use AuxParser/container infrstructure
[senf.git] / Packets / VariantParser.test.cc
1 // $Id$
2 //
3 // Copyright (C) 2007
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 /** \file
24     \brief VariantParser.test unit tests */
25
26 //#include "VariantParser.test.hh"
27 //#include "VariantParser.test.ih"
28
29 // Custom includes
30 #include "Packets.hh"
31
32 #include "../Utils/auto_unit_test.hh"
33 #include <boost/test/test_tools.hpp>
34
35 #define prefix_
36 ///////////////////////////////cc.p////////////////////////////////////////
37
38 BOOST_AUTO_UNIT_TEST(VariantParser)
39 {
40     typedef senf::ArrayParser<10, senf::UInt8Parser> Array10;
41     typedef senf::VariantParser< senf::detail::FixedAuxParserPolicy<senf::UInt8Parser, 1>,
42         boost::mpl::vector<senf::VoidPacketParser, Array10, senf:: UInt32Parser> > Variant;
43     
44     unsigned char data[] = { 0x01, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18,
45                              0x19, 0x1A, 0x1B };
46     senf::DataPacket p (senf::DataPacket::create(data));
47     {
48         Variant v ( boost::next(p.data().begin()), & p.data() );
49         BOOST_REQUIRE_EQUAL( v.variant(), 1u );
50         BOOST_CHECK_EQUAL( senf::bytes(v), 10u );
51         BOOST_CHECK_EQUAL_COLLECTIONS( v.get<1>().begin(), v.get<1>().end(),
52                                        data+1, data+11 );
53         v.init();
54         // The container size should not change in this case but now variant should be 0 ...
55         BOOST_REQUIRE_EQUAL( p.data().size(), 11u );
56         BOOST_REQUIRE_EQUAL( v.variant(), 0u );
57         BOOST_CHECK_EQUAL( senf::bytes(v), 0u );
58         
59         v.init<2>();
60         // v invalidated
61     }
62     {
63         Variant v ( boost::next(p.data().begin()), & p.data() );
64
65         BOOST_CHECK_EQUAL( p.data()[0], 2u );
66         BOOST_REQUIRE_EQUAL( v.variant(), 2u );
67         BOOST_CHECK_EQUAL( senf::bytes(v), 4u );
68
69         unsigned char data2[] = { 0x02, 0x00, 0x00, 0x00, 0x00,
70                                   0x12, 0x13, 0x14, 0x15, 0x16, 0x17, 0x18,
71                                   0x19, 0x1A, 0x1B };
72         BOOST_CHECK_EQUAL_COLLECTIONS( p.data().begin(), p.data().end(),
73                                        data2, data2+sizeof(data2) );
74         BOOST_CHECK_EQUAL( v.get<2>().value(), 0u );
75         BOOST_CHECK_EQUAL( p.data().size(), 15u );
76         p.data()[4] = 0x01u;
77         BOOST_CHECK_EQUAL( v.get<2>().value(), 1u );
78     };
79 }
80
81 namespace {
82     
83     struct SubParser : public senf::PacketParserBase
84     { 
85 #       include SENF_FIXED_PARSER()
86
87         SENF_PARSER_FIELD( foo, senf::UInt32Parser );
88         SENF_PARSER_FIELD( bar, senf::Int16Parser );
89
90         SENF_PARSER_FINALIZE(SubParser);
91     };
92
93     struct TestParser : public senf::PacketParserBase
94     {
95 #       include SENF_PARSER()
96
97         struct TestTransform {
98             typedef unsigned value_type;
99             static unsigned get(unsigned v) { return v/2; }
100             static unsigned set(unsigned v) { return 2*v; }
101         };
102         
103         SENF_PARSER_SKIP_BITS( 4 );
104         SENF_PARSER_BITFIELD_RO( type, 4, unsigned );
105         SENF_PARSER_PRIVATE_VARIANT( content_, transform(TestTransform, type),
106                                      (senf::VoidPacketParser)(SubParser) );
107
108         bool hasContent() const { return content_().variant() == 1; }
109         void hasContent(bool v) const { if (v) content_().init<1>(); else content_().init<0>(); }
110         SubParser content() const { return content_().get<1>(); }
111
112         SENF_PARSER_FINALIZE(TestParser);
113     };
114     
115 }
116
117 BOOST_AUTO_UNIT_TEST(VariantParserMacro)
118 {
119     senf::DataPacket p (senf::DataPacket::create(senf::init_bytes<TestParser>::value));
120     
121     {
122         TestParser v (p.data().begin(), & p.data());
123         BOOST_CHECK( ! v.hasContent() );
124         BOOST_CHECK_EQUAL( senf::bytes(v), 1u );
125         BOOST_CHECK_EQUAL( v.type(), 0u );
126         v.hasContent(true);
127         // Parser invalidated
128     }
129     {
130         TestParser v (p.data().begin(), & p.data());
131         BOOST_CHECK( v.hasContent() );
132         BOOST_CHECK_EQUAL( senf::bytes(v), 7u );
133         BOOST_CHECK_EQUAL( v.content().foo(), 0u );
134         BOOST_CHECK_EQUAL( v.type(), 2u );
135     }
136 }
137
138 ///////////////////////////////cc.e////////////////////////////////////////
139 #undef prefix_
140
141 \f
142 // Local Variables:
143 // mode: c++
144 // fill-column: 100
145 // comment-column: 40
146 // c-file-style: "senf"
147 // indent-tabs-mode: nil
148 // ispell-local-dictionary: "american"
149 // compile-command: "scons -u test"
150 // End: