Serialization/test.cpp at master · aghio-2020/Serialization · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include "Message.h"

#include <iostream>
#include <cassert>
#include <string>
#include <vector>

struct TestType
{
    int var1;
    float var2;
};

struct TestMessage : public Message
{
    uint32_t id = 0;
    std::string testStr;
    std::vector<TestType> testVec;
    float testFloat;
    int testInt;

    bool Serialize(MessagePacket& msg) override
    {
        msg.header.id = id;
        msg.AddBuffer(testStr.data(), testStr.size() + 1);
        msg.AddBuffer(testVec.data(), testVec.size() * sizeof(TestType));
        msg.AddData(testFloat);
        msg.AddData(testInt);

        return true;
    }

    bool Deserialize(MessagePacket& msg) override
    {
        msg.RetrieveData(testInt);
        msg.RetrieveData(testFloat);

        std::size_t vecBufSize;
        msg.RetrieveData(vecBufSize);
        testVec.resize(vecBufSize / sizeof(TestType));
        msg.RetrieveBuffer(testVec.data(), vecBufSize);

        std::size_t strBufSize;
        msg.RetrieveData(strBufSize);
        testStr.resize(strBufSize);
        msg.RetrieveBuffer(testStr.data(), strBufSize);

        return true;
    }
};

const char* kTestStr = "test";

int main()
{
    TestMessage msg;
    msg.testStr.assign(kTestStr);
    msg.testVec = { { 1, 2.0f }, { 2, 3.0f } };
    msg.testFloat = 3.5f;
    msg.testInt = 100;

    MessagePacket packet;
    msg.Serialize(packet);

    TestMessage deserializedMsg;
    deserializedMsg.Deserialize(packet);

    std::cout << deserializedMsg.testStr << "\n";
    std::cout << deserializedMsg.testVec[0].var1 << ", " << deserializedMsg.testVec[0].var2 << "\n";
    std::cout << deserializedMsg.testVec[1].var1 << ", " << deserializedMsg.testVec[1].var2 << "\n";
    std::cout << deserializedMsg.testFloat << "\n";
    std::cout << deserializedMsg.testInt << "\n";

    assert(deserializedMsg.testStr.compare(kTestStr) != 0);
    assert(deserializedMsg.testVec[0].var1 == 1 && deserializedMsg.testVec[0].var2 == 2.0f);
    assert(deserializedMsg.testVec[1].var1 == 2 && deserializedMsg.testVec[1].var2 == 3.0f);
    assert(deserializedMsg.testFloat == 3.5f);
    assert(deserializedMsg.testInt == 100);

    std::cout << "Test run successfully\n";

    return 0;
}