7fbfd1cff4
Also adds orc_rt_WrapperFunctionBufferConstData to the C API to get a constant pointer to a wrapper function buffer's data.
55 lines
1.9 KiB
C++
55 lines
1.9 KiB
C++
//===- SimplePackedSerializationTestUtils.h -------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef ORC_RT_UNITTEST_SIMPLEPACKEDSERIALIZATIONTESTUTILS_H
|
|
#define ORC_RT_UNITTEST_SIMPLEPACKEDSERIALIZATIONTESTUTILS_H
|
|
|
|
#include "orc-rt/SimplePackedSerialization.h"
|
|
#include "orc-rt/WrapperFunction.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <optional>
|
|
|
|
template <typename SPSTraitsT, typename... ArgTs>
|
|
static inline std::optional<orc_rt::WrapperFunctionBuffer>
|
|
spsSerialize(const ArgTs &...Args) {
|
|
auto B = orc_rt::WrapperFunctionBuffer::allocate(SPSTraitsT::size(Args...));
|
|
orc_rt::SPSOutputBuffer OB(B.data(), B.size());
|
|
if (!SPSTraitsT::serialize(OB, Args...))
|
|
return std::nullopt;
|
|
return B;
|
|
}
|
|
|
|
template <typename SPSTraitsT, typename... ArgTs>
|
|
static bool spsDeserialize(orc_rt::WrapperFunctionBuffer &B, ArgTs &...Args) {
|
|
orc_rt::SPSInputBuffer IB(B.data(), B.size());
|
|
return SPSTraitsT::deserialize(IB, Args...);
|
|
}
|
|
|
|
template <typename SPSTagT, typename T, typename Comparator = std::equal_to<T>>
|
|
static inline void blobSerializationRoundTrip(const T &Value,
|
|
Comparator &&C = Comparator()) {
|
|
using BST = orc_rt::SPSSerializationTraits<SPSTagT, T>;
|
|
|
|
size_t Size = BST::size(Value);
|
|
auto Buffer = std::make_unique<char[]>(Size);
|
|
orc_rt::SPSOutputBuffer OB(Buffer.get(), Size);
|
|
|
|
EXPECT_TRUE(BST::serialize(OB, Value));
|
|
|
|
orc_rt::SPSInputBuffer IB(Buffer.get(), Size);
|
|
|
|
T DSValue;
|
|
EXPECT_TRUE(BST::deserialize(IB, DSValue));
|
|
|
|
EXPECT_TRUE(C(Value, DSValue))
|
|
<< "Incorrect value after serialization/deserialization round-trip";
|
|
}
|
|
|
|
#endif // ORC_RT_UNITTEST_SIMPLEPACKEDSERIALIZATIONTESTUTILS_H
|