blob: e505c3d56003e39009e9536857d77353a42a22fa [file] [log] [blame]
Junxiao Shi02b51252016-09-04 04:30:31 +00001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
Davide Pesavento74daf742018-11-23 18:14:13 -05002/*
Davide Pesavento3de7c522024-01-25 18:50:12 -05003 * Copyright (c) 2013-2024 Regents of the University of California.
Junxiao Shi02b51252016-09-04 04:30:31 +00004 *
5 * This file is part of ndn-cxx library (NDN C++ library with eXperimental eXtensions).
6 *
7 * ndn-cxx library is free software: you can redistribute it and/or modify it under the
8 * terms of the GNU Lesser General Public License as published by the Free Software
9 * Foundation, either version 3 of the License, or (at your option) any later version.
10 *
11 * ndn-cxx library is distributed in the hope that it will be useful, but WITHOUT ANY
12 * WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A
13 * PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.
14 *
15 * You should have received copies of the GNU General Public License and GNU Lesser
16 * General Public License along with ndn-cxx, e.g., in COPYING.md file. If not, see
17 * <http://www.gnu.org/licenses/>.
18 *
19 * See AUTHORS.md for complete list of ndn-cxx authors and contributors.
20 */
21
Davide Pesavento7e780642018-11-24 15:51:34 -050022#include "ndn-cxx/encoding/buffer-stream.hpp"
Junxiao Shi02b51252016-09-04 04:30:31 +000023
Davide Pesavento7e780642018-11-24 15:51:34 -050024#include "tests/boost-test.hpp"
Junxiao Shi02b51252016-09-04 04:30:31 +000025
Davide Pesavento47ce2ee2023-05-09 01:33:33 -040026namespace ndn::tests {
Junxiao Shi02b51252016-09-04 04:30:31 +000027
28BOOST_AUTO_TEST_SUITE(Encoding)
29BOOST_AUTO_TEST_SUITE(TestBufferStream)
30
31BOOST_AUTO_TEST_CASE(Empty)
32{
33 OBufferStream os;
34
Davide Pesavento3de7c522024-01-25 18:50:12 -050035 std::shared_ptr<Buffer> buf = os.buf();
Junxiao Shi02b51252016-09-04 04:30:31 +000036 BOOST_CHECK_EQUAL(buf->size(), 0);
37}
38
39BOOST_AUTO_TEST_CASE(Put)
40{
41 OBufferStream os;
42 os.put(0x33);
43 os.put(0x44);
44
Davide Pesavento3de7c522024-01-25 18:50:12 -050045 std::shared_ptr<Buffer> buf = os.buf();
Junxiao Shi02b51252016-09-04 04:30:31 +000046 BOOST_REQUIRE_EQUAL(buf->size(), 2);
47 BOOST_CHECK_EQUAL(buf->at(0), 0x33);
48 BOOST_CHECK_EQUAL(buf->at(1), 0x44);
49}
50
51BOOST_AUTO_TEST_CASE(Write)
52{
53 OBufferStream os;
54 os.write("\x11\x22", 2);
55
Davide Pesavento3de7c522024-01-25 18:50:12 -050056 std::shared_ptr<Buffer> buf = os.buf();
Junxiao Shi02b51252016-09-04 04:30:31 +000057 BOOST_REQUIRE_EQUAL(buf->size(), 2);
58 BOOST_CHECK_EQUAL(buf->at(0), 0x11);
59 BOOST_CHECK_EQUAL(buf->at(1), 0x22);
60}
61
62BOOST_AUTO_TEST_CASE(Destructor) // Bug 3727
63{
Davide Pesavento3de7c522024-01-25 18:50:12 -050064 auto os = std::make_unique<OBufferStream>();
Davide Pesaventoe16ac862022-09-25 20:48:22 -040065 auto buf = os->buf();
Junxiao Shi02b51252016-09-04 04:30:31 +000066 *os << 'x';
Davide Pesaventoe16ac862022-09-25 20:48:22 -040067 // do NOT flush or call buf() here
Davide Pesaventoeee3e822016-11-26 19:19:34 +010068
Davide Pesaventoe16ac862022-09-25 20:48:22 -040069 os.reset(); // should not cause use-after-free
70 BOOST_CHECK_EQUAL(buf->size(), 1);
71}
72
73BOOST_AUTO_TEST_CASE(Close) // Bug 5240
74{
75 OBufferStream os;
76 os << "foo";
77 os.close();
78
79 auto buf = os.buf(); // should not cause assertion failure
80 BOOST_CHECK_EQUAL(buf->size(), 3);
Junxiao Shi02b51252016-09-04 04:30:31 +000081}
82
83BOOST_AUTO_TEST_SUITE_END() // TestBufferStream
84BOOST_AUTO_TEST_SUITE_END() // Encoding
85
Davide Pesavento47ce2ee2023-05-09 01:33:33 -040086} // namespace ndn::tests