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()); 
        } 
    } 
}