mirror of
https://github.com/vtnerd/monero-lws.git
synced 2025-01-08 19:59:31 +00:00
196 lines
7.2 KiB
C++
196 lines
7.2 KiB
C++
// Copyright (c) 2023, The Monero Project
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without modification, are
|
|
// permitted provided that the following conditions are met:
|
|
//
|
|
// 1. Redistributions of source code must retain the above copyright notice, this list of
|
|
// conditions and the following disclaimer.
|
|
//
|
|
// 2. Redistributions in binary form must reproduce the above copyright notice, this list
|
|
// of conditions and the following disclaimer in the documentation and/or other
|
|
// materials provided with the distribution.
|
|
//
|
|
// 3. Neither the name of the copyright holder nor the names of its contributors may be
|
|
// used to endorse or promote products derived from this software without specific
|
|
// prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY
|
|
// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
|
|
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
|
|
// THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
|
|
// STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
|
|
// THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
#include "framework.test.h"
|
|
|
|
#include <boost/core/demangle.hpp>
|
|
#include <boost/range/algorithm/equal.hpp>
|
|
#include <cstdint>
|
|
#include <type_traits>
|
|
#include "wire/field.h"
|
|
#include "wire/traits.h"
|
|
#include "wire/msgpack.h"
|
|
#include "wire/vector.h"
|
|
#include "wire/wrapper/array.h"
|
|
#include "wire/wrappers_impl.h"
|
|
|
|
#include "wire/base.test.h"
|
|
|
|
namespace
|
|
{
|
|
constexpr const char basic_string[] = u8"my_string_data";
|
|
//constexpr const char basic_[] =
|
|
// u8"{\"utf8\":\"my_string_data\",\"vec\":[0,127],\"data\":\"00ff2211\",\"choice\":true}";
|
|
constexpr const std::uint8_t basic_msgpack[] = {
|
|
0x84, 0xa4, 0x75, 0x74, 0x66, 0x38, 0xae, 0x6d, 0x79, 0x5f, 0x73, 0x74,
|
|
0x72, 0x69, 0x6e, 0x67, 0x5f, 0x64, 0x61, 0x74, 0x61, 0xa3, 0x76, 0x65,
|
|
0x63, 0x92, 0x00, 0x7f, 0xa4, 0x64, 0x61, 0x74, 0x61, 0xc4, 0x04, 0x00,
|
|
0xff, 0x22, 0x11, 0xa6, 0x63, 0x68, 0x6f, 0x69, 0x63, 0x65, 0xc3
|
|
};
|
|
constexpr const std::uint8_t advanced_msgpack[] = {
|
|
0x84, 0x00, 0xae, 0x6d, 0x79, 0x5f, 0x73, 0x74, 0x72, 0x69, 0x6e, 0x67,
|
|
0x5f, 0x64, 0x61, 0x74, 0x61, 0x01, 0x92, 0x00, 0x7f, 0x02, 0xc4, 0x04,
|
|
0x00, 0xff, 0x22, 0x11, 0xcc, 0xfe, 0xc3
|
|
};
|
|
|
|
template<typename T>
|
|
struct basic_object
|
|
{
|
|
std::string utf8;
|
|
std::vector<T> vec;
|
|
lws_test::small_blob data;
|
|
bool choice;
|
|
};
|
|
|
|
template<typename F, typename T>
|
|
void basic_object_map(F& format, T& self)
|
|
{
|
|
using vec_max = wire::max_element_count<100>;
|
|
wire::object(format,
|
|
WIRE_FIELD_ID(0, utf8),
|
|
wire::field<1>("vec", wire::array<vec_max>(std::ref(self.vec))),
|
|
WIRE_FIELD_ID(2, data),
|
|
WIRE_FIELD_ID(254, choice)
|
|
);
|
|
}
|
|
|
|
template<typename T>
|
|
void read_bytes(wire::msgpack_reader& source, basic_object<T>& dest)
|
|
{ basic_object_map(source, dest); }
|
|
|
|
template<typename T>
|
|
void write_bytes(wire::msgpack_writer& dest, const basic_object<T>& source)
|
|
{ basic_object_map(dest, source); }
|
|
|
|
template<typename T>
|
|
void test_basic_reading(lest::env& lest_env)
|
|
{
|
|
SETUP("Basic (string keys) with " + boost::core::demangle(typeid(T).name()) + " integers")
|
|
{
|
|
basic_object<T> result{};
|
|
const std::error_code error =
|
|
wire::msgpack::from_bytes(epee::byte_slice{{basic_msgpack}}, result);
|
|
EXPECT(!error);
|
|
EXPECT(result.utf8 == basic_string);
|
|
{
|
|
const std::vector<T> expected{0, 127};
|
|
EXPECT(result.vec == expected);
|
|
}
|
|
EXPECT(result.data == lws_test::blob_test1);
|
|
EXPECT(result.choice);
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void test_advanced_reading(lest::env& lest_env)
|
|
{
|
|
SETUP("Advanced (integer keys) with " + boost::core::demangle(typeid(T).name()) + " integers")
|
|
{
|
|
basic_object<T> result{};
|
|
const std::error_code error =
|
|
wire::msgpack::from_bytes(epee::byte_slice{{advanced_msgpack}}, result);
|
|
EXPECT(!error);
|
|
EXPECT(result.utf8 == basic_string);
|
|
{
|
|
const std::vector<T> expected{0, 127};
|
|
EXPECT(result.vec == expected);
|
|
}
|
|
EXPECT(result.data == lws_test::blob_test1);
|
|
EXPECT(result.choice);
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void test_basic_writing(lest::env& lest_env)
|
|
{
|
|
SETUP("Basic (string keys) with " + boost::core::demangle(typeid(T).name()) + " integers")
|
|
{
|
|
const basic_object<T> val{basic_string, std::vector<T>{0, 127}, lws_test::blob_test1, true};
|
|
epee::byte_slice result{};
|
|
const std::error_code error = wire::msgpack::to_bytes(result, val);
|
|
EXPECT(!error);
|
|
EXPECT(boost::range::equal(result, epee::byte_slice{{basic_msgpack}}));
|
|
}
|
|
}
|
|
|
|
template<typename T>
|
|
void test_advanced_writing(lest::env& lest_env)
|
|
{
|
|
SETUP("Advanced (integer keys) with " + boost::core::demangle(typeid(T).name()) + " integers")
|
|
{
|
|
const basic_object<T> val{basic_string, std::vector<T>{0, 127}, lws_test::blob_test1, true};
|
|
epee::byte_slice result{};
|
|
{
|
|
wire::msgpack_slice_writer out{true};
|
|
wire_write::bytes(out, val);
|
|
result = epee::byte_slice{out.take_sink()};
|
|
}
|
|
EXPECT(boost::range::equal(result, epee::byte_slice{{advanced_msgpack}}));
|
|
}
|
|
}
|
|
}
|
|
|
|
LWS_CASE("wire::msgpack_reader")
|
|
{
|
|
test_basic_reading<std::int16_t>(lest_env);
|
|
test_basic_reading<std::int32_t>(lest_env);
|
|
test_basic_reading<std::int64_t>(lest_env);
|
|
test_basic_reading<std::intmax_t>(lest_env);
|
|
test_basic_reading<std::uint16_t>(lest_env);
|
|
test_basic_reading<std::uint32_t>(lest_env);
|
|
test_basic_reading<std::uint64_t>(lest_env);
|
|
test_basic_reading<std::uintmax_t>(lest_env);
|
|
|
|
test_advanced_reading<std::int16_t>(lest_env);
|
|
test_advanced_reading<std::int32_t>(lest_env);
|
|
test_advanced_reading<std::int64_t>(lest_env);
|
|
test_advanced_reading<std::intmax_t>(lest_env);
|
|
test_advanced_reading<std::uint16_t>(lest_env);
|
|
test_advanced_reading<std::uint32_t>(lest_env);
|
|
test_advanced_reading<std::uint64_t>(lest_env);
|
|
test_advanced_reading<std::uintmax_t>(lest_env);
|
|
}
|
|
|
|
LWS_CASE("wire::msgpack_writer")
|
|
{
|
|
test_basic_writing<std::int16_t>(lest_env);
|
|
test_basic_writing<std::int32_t>(lest_env);
|
|
test_basic_writing<std::int64_t>(lest_env);
|
|
test_basic_writing<std::intmax_t>(lest_env);
|
|
test_basic_writing<std::uint16_t>(lest_env);
|
|
test_basic_writing<std::uint32_t>(lest_env);
|
|
test_basic_writing<std::uint64_t>(lest_env);
|
|
test_basic_writing<std::uintmax_t>(lest_env);
|
|
|
|
test_advanced_writing<std::int16_t>(lest_env);
|
|
test_advanced_writing<std::int32_t>(lest_env);
|
|
test_advanced_writing<std::int64_t>(lest_env);
|
|
test_advanced_writing<std::intmax_t>(lest_env);
|
|
test_advanced_writing<std::uint16_t>(lest_env);
|
|
test_advanced_writing<std::uint32_t>(lest_env);
|
|
test_advanced_writing<std::uint64_t>(lest_env);
|
|
test_advanced_writing<std::uintmax_t>(lest_env);
|
|
}
|