library/cpp/protobuf/util/repeated_field_utils_ut.cpp


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

#include "repeated_field_utils.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(RepeatedFieldUtils) {
    Y_UNIT_TEST(RemoveIf) {
        {
            NProtobufUtilUt::TWalkTest msg;
            msg.AddRepInt(0);
            msg.AddRepInt(1);
            msg.AddRepInt(2);
            msg.AddRepInt(3);
            msg.AddRepInt(4);
            msg.AddRepInt(5);
            auto cond = [](ui32 val) {
                return val % 2 == 0;
            };
            RemoveRepeatedFieldItemIf(msg.MutableRepInt(), cond);
            UNIT_ASSERT_VALUES_EQUAL(3, msg.RepIntSize());
            UNIT_ASSERT_VALUES_EQUAL(1, msg.GetRepInt(0));
            UNIT_ASSERT_VALUES_EQUAL(3, msg.GetRepInt(1));
            UNIT_ASSERT_VALUES_EQUAL(5, msg.GetRepInt(2));
        }

        {
            NProtobufUtilUt::TWalkTest msg;
            msg.AddRepSub()->SetOptInt(0);
            msg.AddRepSub()->SetOptInt(1);
            msg.AddRepSub()->SetOptInt(2);
            msg.AddRepSub()->SetOptInt(3);
            msg.AddRepSub()->SetOptInt(4);
            msg.AddRepSub()->SetOptInt(5);
            auto cond = [](const NProtobufUtilUt::TWalkTest& val) {
                return val.GetOptInt() % 2 == 0;
            };
            RemoveRepeatedFieldItemIf(msg.MutableRepSub(), cond);
            UNIT_ASSERT_VALUES_EQUAL(3, msg.RepSubSize());
            UNIT_ASSERT_VALUES_EQUAL(1, msg.GetRepSub(0).GetOptInt());
            UNIT_ASSERT_VALUES_EQUAL(3, msg.GetRepSub(1).GetOptInt());
            UNIT_ASSERT_VALUES_EQUAL(5, msg.GetRepSub(2).GetOptInt());
        }
    }
}