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
|
#include <yql/essentials/parser/pg_wrapper/interface/compare.h>
#include <yql/essentials/minikql/mkql_alloc.h>
#include <yql/essentials/parser/pg_catalog/catalog.h>
#include <library/cpp/testing/unittest/registar.h>
namespace NKikimr::NMiniKQL {
namespace {
ui32 PgTypeId(TStringBuf typeName) {
return NYql::NPg::LookupType(TString(typeName)).TypeId;
}
constexpr TStringBuf SmallNumeric = "99999999999999999999999999999999999999.99";
constexpr TStringBuf LargeNumeric = "100000000000000000000000000000000000000.01";
} // namespace
Y_UNIT_TEST_SUITE(PgCompareWithCastsTests) {
Y_UNIT_TEST(TestInt) {
auto less = PgCompareWithCasts("1", PgTypeId("int4"), "2", PgTypeId("int4"), EPgCompareType::Less);
UNIT_ASSERT(less.has_value() && *less);
auto greater = PgCompareWithCasts("2", PgTypeId("int4"), "1", PgTypeId("int4"), EPgCompareType::Greater);
UNIT_ASSERT(greater.has_value() && *greater);
auto notLess = PgCompareWithCasts("5", PgTypeId("int4"), "5", PgTypeId("int4"), EPgCompareType::Less);
UNIT_ASSERT(notLess.has_value() && !*notLess);
auto notGreater = PgCompareWithCasts("1", PgTypeId("int4"), "2", PgTypeId("int4"), EPgCompareType::Greater);
UNIT_ASSERT(notGreater.has_value() && !*notGreater);
}
Y_UNIT_TEST(TestInterval) {
auto less = PgCompareWithCasts("1 hour", PgTypeId("interval"), "2 hours", PgTypeId("interval"), EPgCompareType::Less);
UNIT_ASSERT(less.has_value() && *less);
auto greater = PgCompareWithCasts("2 hours", PgTypeId("interval"), "1 hour", PgTypeId("interval"), EPgCompareType::Greater);
UNIT_ASSERT(greater.has_value() && *greater);
auto notLess = PgCompareWithCasts("1 day", PgTypeId("interval"), "1 day", PgTypeId("interval"), EPgCompareType::Less);
UNIT_ASSERT(notLess.has_value() && !*notLess);
auto notGreater = PgCompareWithCasts("1 hour", PgTypeId("interval"), "2 hours", PgTypeId("interval"), EPgCompareType::Greater);
UNIT_ASSERT(notGreater.has_value() && !*notGreater);
}
Y_UNIT_TEST(TestNumeric) {
auto less = PgCompareWithCasts(SmallNumeric, PgTypeId("numeric"), LargeNumeric, PgTypeId("numeric"), EPgCompareType::Less);
UNIT_ASSERT(less.has_value() && *less);
auto greater = PgCompareWithCasts(LargeNumeric, PgTypeId("numeric"), SmallNumeric, PgTypeId("numeric"), EPgCompareType::Greater);
UNIT_ASSERT(greater.has_value() && *greater);
auto notLess = PgCompareWithCasts(SmallNumeric, PgTypeId("numeric"), SmallNumeric, PgTypeId("numeric"), EPgCompareType::Less);
UNIT_ASSERT(notLess.has_value() && !*notLess);
auto notGreater = PgCompareWithCasts(SmallNumeric, PgTypeId("numeric"), LargeNumeric, PgTypeId("numeric"), EPgCompareType::Greater);
UNIT_ASSERT(notGreater.has_value() && !*notGreater);
}
Y_UNIT_TEST(TestNumericWithInt) {
auto less = PgCompareWithCasts("1", PgTypeId("int4"), "2.5", PgTypeId("numeric"), EPgCompareType::Less);
UNIT_ASSERT(less.has_value() && *less);
auto greater = PgCompareWithCasts("3.5", PgTypeId("numeric"), "2", PgTypeId("int4"), EPgCompareType::Greater);
UNIT_ASSERT(greater.has_value() && *greater);
auto notLess = PgCompareWithCasts("5", PgTypeId("int4"), "5", PgTypeId("numeric"), EPgCompareType::Less);
UNIT_ASSERT(notLess.has_value() && !*notLess);
auto notGreater = PgCompareWithCasts("1.5", PgTypeId("numeric"), "2", PgTypeId("int4"), EPgCompareType::Greater);
UNIT_ASSERT(notGreater.has_value() && !*notGreater);
}
Y_UNIT_TEST(TestIncompatibleTypes) {
auto result = PgCompareWithCasts("1", PgTypeId("int4"), "hello", PgTypeId("text"), EPgCompareType::Less);
UNIT_ASSERT(!result.has_value());
UNIT_ASSERT_EQUAL_C(result.error(), TString("Unable to find an overload for operator < with given argument type(s): (int4,text)"), result.error());
}
} // Y_UNIT_TEST_SUITE(TPgCompareWithCastsTests)
} // namespace NKikimr::NMiniKQL
|
