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
84
85
86
87
88
89
90
91
92
93
94
|
#include <library/cpp/testing/gtest/gtest.h>
#include <library/cpp/yt/misc/cast.h>
#include <library/cpp/yt/misc/enum.h>
namespace NYT {
namespace {
////////////////////////////////////////////////////////////////////////////////
DEFINE_ENUM_WITH_UNDERLYING_TYPE(ECardinal, char,
((West) (0))
((North) (1))
((East) (2))
((South) (3))
);
DEFINE_BIT_ENUM_WITH_UNDERLYING_TYPE(EFeatures, ui8,
((None) (0x0000))
((First) (0x0001))
((Second)(0x0002))
);
DEFINE_BIT_ENUM(ELangsWithUnknown,
((None) (0x00))
((Cpp) (0x01))
((Go) (0x02))
((Rust) (0x04))
((Python) (0x08))
((JavaScript) (0x10))
((CppGo) (0x03))
((All) (0x1f))
((Unknown) (0x20))
);
DEFINE_ENUM_UNKNOWN_VALUE(ELangsWithUnknown, Unknown);
DEFINE_ENUM(EColorWithUnknown,
((Red) (0))
((Green) (1))
((Unknown) (2))
);
DEFINE_ENUM_UNKNOWN_VALUE(EColorWithUnknown, Unknown);
TEST(TCastTest, TryCheckedEnumCast)
{
EXPECT_EQ((TryCheckedEnumCast<ECardinal, char>(2)), ECardinal::East);
EXPECT_EQ((TryCheckedEnumCast<ECardinal, int>(3)), ECardinal::South);
EXPECT_FALSE((TryCheckedEnumCast<ECardinal, char>(100)));
EXPECT_FALSE((TryCheckedEnumCast<ECardinal, int>(300)));
EXPECT_EQ((TryCheckedEnumCast<EFeatures, ui8>(0)), EFeatures::None);
EXPECT_EQ((TryCheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::First))), EFeatures::First);
EXPECT_EQ((TryCheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::Second))), EFeatures::Second);
EXPECT_EQ((TryCheckedEnumCast<EFeatures, int>(ToUnderlying(EFeatures::First))), EFeatures::First);
EXPECT_EQ((TryCheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::First | EFeatures::Second))), EFeatures::First | EFeatures::Second);
EXPECT_FALSE((TryCheckedEnumCast<EFeatures, ui8>(0x10)));
EXPECT_FALSE(TryCheckedEnumCast<EColorWithUnknown>(3));
EXPECT_EQ(TryCheckedEnumCast<EColorWithUnknown>(3, /*enableUnknown*/ true), EColorWithUnknown::Unknown);
EXPECT_FALSE(TryCheckedEnumCast<ELangsWithUnknown>(0x40));
EXPECT_EQ(TryCheckedEnumCast<ELangsWithUnknown>(0x40, /*enableUnknown*/ true), ELangsWithUnknown::Unknown);
EXPECT_EQ(TryCheckedEnumCast<ELangsWithUnknown>(0x41, /*enableUnknown*/ true), ELangsWithUnknown::Unknown | ELangsWithUnknown::Cpp);
}
TEST(TCastTest, CheckedEnumCast)
{
EXPECT_EQ((CheckedEnumCast<ECardinal, char>(2)), ECardinal::East);
EXPECT_EQ((CheckedEnumCast<ECardinal, int>(3)), ECardinal::South);
EXPECT_THROW((CheckedEnumCast<ECardinal, char>(100)), TSimpleException);
EXPECT_THROW((CheckedEnumCast<ECardinal, int>(300)), TSimpleException);
EXPECT_EQ((CheckedEnumCast<EFeatures, ui8>(0)), EFeatures::None);
EXPECT_EQ((CheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::First))), EFeatures::First);
EXPECT_EQ((CheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::Second))), EFeatures::Second);
EXPECT_EQ((CheckedEnumCast<EFeatures, int>(ToUnderlying(EFeatures::First))), EFeatures::First);
EXPECT_EQ((CheckedEnumCast<EFeatures, ui8>(ToUnderlying(EFeatures::First | EFeatures::Second))), EFeatures::First | EFeatures::Second);
EXPECT_THROW((CheckedEnumCast<EFeatures, ui8>(0x10)), TSimpleException);
EXPECT_EQ(CheckedEnumCast<EColorWithUnknown>(3), EColorWithUnknown::Unknown);
EXPECT_EQ(CheckedEnumCast<ELangsWithUnknown>(0x40), ELangsWithUnknown::Unknown);
EXPECT_EQ(CheckedEnumCast<ELangsWithUnknown>(0x41), ELangsWithUnknown::Unknown | ELangsWithUnknown::Cpp);
}
////////////////////////////////////////////////////////////////////////////////
} // namespace
} // namespace NYT
|
