https://github.com/ska-sa/spead2
Tip revision: fad736506314d44e55d677d3d9a97327fd8e46ca authored by Bruce Merry on 03 November 2020, 14:41:05 UTC
Prepare for 3.1.0 release
Prepare for 3.1.0 release
Tip revision: fad7365
unittest_memory_allocator.cpp
/* Copyright 2016, 2019 National Research Foundation (SARAO)
*
* This program is free software: you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation, either version 3 of the License, or (at your option) any
* later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
/**
* @file
*
* Unit tests for common_memory_allocator.
*/
#include <boost/test/unit_test.hpp>
#include <boost/mpl/list.hpp>
#include <memory>
#include <spead2/common_memory_allocator.h>
namespace spead2
{
namespace unittest
{
BOOST_AUTO_TEST_SUITE(common)
BOOST_AUTO_TEST_SUITE(memory_allocator)
// Wrapper to pass prefer_huge to the mmap_allocator constructor
class huge_mmap_allocator : public spead2::mmap_allocator
{
public:
huge_mmap_allocator() : spead2::mmap_allocator(0, true) {}
};
typedef boost::mpl::list<spead2::memory_allocator, spead2::mmap_allocator, huge_mmap_allocator> test_types;
BOOST_AUTO_TEST_CASE_TEMPLATE(allocator_test, T, test_types)
{
typedef typename T::pointer pointer;
std::shared_ptr<T> allocator = std::make_shared<T>();
pointer ptr = allocator->allocate(12345, nullptr);
// Scribble on the memory to make sure it is really allocated
for (std::size_t i = 0; i < 12345; i++)
ptr[i] = 1;
ptr.reset();
}
BOOST_AUTO_TEST_CASE_TEMPLATE(out_of_memory, T, test_types)
{
std::shared_ptr<T> allocator = std::make_shared<T>();
BOOST_CHECK_THROW(allocator->allocate(SIZE_MAX - 1, nullptr), std::bad_alloc);
}
BOOST_AUTO_TEST_SUITE_END() // memory_allocator
BOOST_AUTO_TEST_SUITE_END() // common
}} // namespace spead2::unittest