124 lines
2.6 KiB
C++
Executable file
124 lines
2.6 KiB
C++
Executable file
// Copyright (c) 2011-2016 The Cryptonote developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <boost/lexical_cast.hpp>
|
|
|
|
#include "Serialization/KVBinaryInputStreamSerializer.h"
|
|
#include "Serialization/KVBinaryOutputStreamSerializer.h"
|
|
#include "Serialization/SerializationOverloads.h"
|
|
#include "Serialization/SerializationTools.h"
|
|
|
|
#include <array>
|
|
|
|
using namespace CryptoNote;
|
|
|
|
namespace CryptoNote {
|
|
|
|
struct TestElement {
|
|
std::string name;
|
|
uint32_t nonce;
|
|
std::array<uint8_t, 16> blob;
|
|
std::vector<uint32_t> u32array;
|
|
|
|
bool operator == (const TestElement& other) const {
|
|
return
|
|
name == other.name &&
|
|
nonce == other.nonce &&
|
|
blob == other.blob &&
|
|
u32array == other.u32array;
|
|
}
|
|
|
|
void serialize(ISerializer& s) {
|
|
s(name, "name");
|
|
s(nonce, "nonce");
|
|
s.binary(blob.data(), blob.size(), "blob");
|
|
serializeAsBinary(u32array, "u32array", s);
|
|
}
|
|
};
|
|
|
|
struct TestStruct {
|
|
uint8_t u8;
|
|
uint32_t u32;
|
|
uint64_t u64;
|
|
std::vector<TestElement> vec1;
|
|
std::vector<TestElement> vec2;
|
|
TestElement root;
|
|
|
|
bool operator == (const TestStruct& other) const {
|
|
return
|
|
root == other.root &&
|
|
u8 == other.u8 &&
|
|
u32 == other.u32 &&
|
|
u64 == other.u64 &&
|
|
vec1 == other.vec1 &&
|
|
vec2 == other.vec2;
|
|
}
|
|
|
|
void serialize(ISerializer& s) {
|
|
s(root, "root");
|
|
s(vec1, "vec1");
|
|
s(vec2, "vec2");
|
|
s(u8, "u8");
|
|
s(u32, "u32");
|
|
s(u64, "u64");
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
#include <chrono>
|
|
|
|
typedef std::chrono::high_resolution_clock hclock;
|
|
|
|
class HiResTimer {
|
|
public:
|
|
HiResTimer() :
|
|
start(hclock::now()) {}
|
|
|
|
std::chrono::duration<double> duration() {
|
|
return hclock::now() - start;
|
|
}
|
|
|
|
private:
|
|
hclock::time_point start;
|
|
};
|
|
|
|
TEST(KVSerialize, Simple) {
|
|
TestElement testData1, testData2;
|
|
|
|
testData1.name = "hello";
|
|
testData1.nonce = 12345;
|
|
testData1.u32array.resize(128);
|
|
|
|
testData2.name = "bye";
|
|
testData2.nonce = 54321;
|
|
|
|
std::string buf = CryptoNote::storeToBinaryKeyValue(testData1);
|
|
ASSERT_TRUE(CryptoNote::loadFromBinaryKeyValue(testData2, buf));
|
|
EXPECT_EQ(testData1, testData2);
|
|
}
|
|
|
|
TEST(KVSerialize, BigCollection) {
|
|
TestStruct ts1;
|
|
|
|
ts1.u8 = 100;
|
|
ts1.u32 = 0xff0000;
|
|
ts1.u64 = 1ULL << 60;
|
|
ts1.root.name = "hello";
|
|
|
|
TestElement sample;
|
|
sample.nonce = 101;
|
|
ts1.vec1.resize(0x10000 >> 2, sample);
|
|
|
|
TestStruct ts2;
|
|
|
|
std::string buf = CryptoNote::storeToBinaryKeyValue(ts1);
|
|
ASSERT_TRUE(CryptoNote::loadFromBinaryKeyValue(ts2, buf));
|
|
EXPECT_EQ(ts1, ts2);
|
|
}
|