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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#include "format.h"
#include <library/cpp/testing/unittest/registar.h>
#include <util/generic/string.h>
#include <util/string/builder.h>
#include <array>
using namespace NMonitoring;
Y_UNIT_TEST_SUITE(TFormatTest) {
Y_UNIT_TEST(ContentTypeHeader) {
UNIT_ASSERT_EQUAL(FormatFromContentType(""), EFormat::UNKNOWN);
UNIT_ASSERT_EQUAL(FormatFromContentType("application/json;some=stuff"), EFormat::JSON);
UNIT_ASSERT_EQUAL(FormatFromContentType("application/x-solomon-spack"), EFormat::SPACK);
UNIT_ASSERT_EQUAL(FormatFromContentType("application/xml"), EFormat::UNKNOWN);
UNIT_ASSERT_EQUAL(FormatFromContentType(";application/xml"), EFormat::UNKNOWN);
}
Y_UNIT_TEST(AcceptHeader) {
UNIT_ASSERT_EQUAL(FormatFromAcceptHeader(""), EFormat::UNKNOWN);
UNIT_ASSERT_EQUAL(FormatFromAcceptHeader("*/*"), EFormat::UNKNOWN);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/xml"),
EFormat::UNKNOWN);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/json"),
EFormat::JSON);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/x-solomon-spack"),
EFormat::SPACK);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/x-solomon-pb"),
EFormat::PROTOBUF);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/x-solomon-txt"),
EFormat::TEXT);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/json, text/plain"),
EFormat::JSON);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/x-solomon-spack, application/json, text/plain"),
EFormat::SPACK);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader(" , application/x-solomon-spack ,, application/json , text/plain"),
EFormat::SPACK);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/xml, application/x-solomon-spack, text/plain"),
EFormat::SPACK);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("text/plain"),
EFormat::PROMETHEUS);
UNIT_ASSERT_EQUAL(
FormatFromAcceptHeader("application/openmetrics-text;version=0.0.1;q=0.75,text/plain;version=0.0.4"),
EFormat::PROMETHEUS);
}
Y_UNIT_TEST(FormatToStrFromStr) {
const std::array<EFormat, 6> formats = {{
EFormat::UNKNOWN,
EFormat::SPACK,
EFormat::JSON,
EFormat::PROTOBUF,
EFormat::TEXT,
EFormat::PROMETHEUS,
}};
for (EFormat f : formats) {
TString str = (TStringBuilder() << f);
EFormat g = FromString<EFormat>(str);
UNIT_ASSERT_EQUAL(f, g);
}
}
Y_UNIT_TEST(AcceptEncodingHeader) {
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader(""),
ECompression::UNKNOWN);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("br"),
ECompression::UNKNOWN);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("identity"),
ECompression::IDENTITY);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("zlib"),
ECompression::ZLIB);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("lz4"),
ECompression::LZ4);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("zstd"),
ECompression::ZSTD);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("zstd, zlib"),
ECompression::ZSTD);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader(" ,, , zstd , zlib"),
ECompression::ZSTD);
UNIT_ASSERT_EQUAL(
CompressionFromAcceptEncodingHeader("br, deflate,lz4, zlib"),
ECompression::LZ4);
}
Y_UNIT_TEST(CompressionToStrFromStr) {
const std::array<ECompression, 5> algs = {{
ECompression::UNKNOWN,
ECompression::IDENTITY,
ECompression::ZLIB,
ECompression::LZ4,
ECompression::ZSTD,
}};
for (ECompression a : algs) {
TString str = (TStringBuilder() << a);
ECompression b = FromString<ECompression>(str);
UNIT_ASSERT_EQUAL(a, b);
}
}
}
|
