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 "merge.h"
#include <library/cpp/protobuf/util/ut/common_ut.pb.h>
#include <library/cpp/testing/unittest/registar.h>
using namespace NProtoBuf;
Y_UNIT_TEST_SUITE(ProtobufMerge) {
static void InitProto(NProtobufUtilUt::TMergeTest & p, bool isSrc) {
size_t start = isSrc ? 0 : 100;
p.AddMergeInt(start + 1);
p.AddMergeInt(start + 2);
p.AddNoMergeInt(start + 3);
p.AddNoMergeInt(start + 4);
NProtobufUtilUt::TMergeTestMerge* m = p.MutableMergeSub();
m->SetA(start + 5);
m->AddB(start + 6);
m->AddB(start + 7);
m->AddC(start + 14);
if (!isSrc) {
// only for dst
NProtobufUtilUt::TMergeTestMerge* mm1 = p.AddNoMergeRepSub();
mm1->SetA(start + 8);
mm1->AddB(start + 9);
mm1->AddB(start + 10);
}
NProtobufUtilUt::TMergeTestNoMerge* mm3 = p.MutableNoMergeOptSub();
mm3->SetA(start + 11);
mm3->AddB(start + 12);
mm3->AddB(start + 13);
}
Y_UNIT_TEST(CustomMerge) {
NProtobufUtilUt::TMergeTest src, dst;
InitProto(src, true);
InitProto(dst, false);
// Cerr << "\nsrc: " << src.ShortDebugString() << Endl;
// Cerr << "dst: " << dst.ShortDebugString() << Endl;
NProtoBuf::CustomMerge(src, dst);
// Cerr << "dst2:" << dst.ShortDebugString() << Endl;
// repeated uint32 MergeInt = 1;
UNIT_ASSERT_EQUAL(dst.MergeIntSize(), 4);
UNIT_ASSERT_EQUAL(dst.GetMergeInt(0), 101);
UNIT_ASSERT_EQUAL(dst.GetMergeInt(1), 102);
UNIT_ASSERT_EQUAL(dst.GetMergeInt(2), 1);
UNIT_ASSERT_EQUAL(dst.GetMergeInt(3), 2);
// repeated uint32 NoMergeInt = 2 [(DontMergeField)=true];
UNIT_ASSERT_EQUAL(dst.NoMergeIntSize(), 2);
UNIT_ASSERT_EQUAL(dst.GetNoMergeInt(0), 3);
UNIT_ASSERT_EQUAL(dst.GetNoMergeInt(1), 4);
// optional TMergeTestMerge MergeSub = 3;
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetA(), 5);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().BSize(), 4);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetB(0), 106);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetB(1), 107);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetB(2), 6);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetB(3), 7);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().CSize(), 1);
UNIT_ASSERT_EQUAL(dst.GetMergeSub().GetC(0), 14);
// repeated TMergeTestMerge NoMergeRepSub = 4 [(DontMergeField)=true];
UNIT_ASSERT_EQUAL(dst.NoMergeRepSubSize(), 1);
UNIT_ASSERT_EQUAL(dst.GetNoMergeRepSub(0).GetA(), 108);
UNIT_ASSERT_EQUAL(dst.GetNoMergeRepSub(0).BSize(), 2);
UNIT_ASSERT_EQUAL(dst.GetNoMergeRepSub(0).GetB(0), 109);
UNIT_ASSERT_EQUAL(dst.GetNoMergeRepSub(0).GetB(1), 110);
// optional TMergeTestNoMerge NoMergeOptSub = 5;
UNIT_ASSERT_EQUAL(dst.GetNoMergeOptSub().GetA(), 11);
UNIT_ASSERT_EQUAL(dst.GetNoMergeOptSub().BSize(), 2);
UNIT_ASSERT_EQUAL(dst.GetNoMergeOptSub().GetB(0), 12);
UNIT_ASSERT_EQUAL(dst.GetNoMergeOptSub().GetB(1), 13);
}
}
|
