diff options
author | asaitgalin <asaitgalin@yandex-team.ru> | 2022-02-10 16:47:28 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:47:28 +0300 |
commit | cb85b6b6ed9608dfa24ee0362fd286dd27fd40ae (patch) | |
tree | 9814fbd1c3effac9b8377c5d604b367b14e2db55 /library/cpp/streams/brotli/brotli_ut.cpp | |
parent | c0780d05ad256f75dc8e0fa36aee5dbce402e8f6 (diff) | |
download | ydb-cb85b6b6ed9608dfa24ee0362fd286dd27fd40ae.tar.gz |
Restoring authorship annotation for <asaitgalin@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/streams/brotli/brotli_ut.cpp')
-rw-r--r-- | library/cpp/streams/brotli/brotli_ut.cpp | 74 |
1 files changed, 37 insertions, 37 deletions
diff --git a/library/cpp/streams/brotli/brotli_ut.cpp b/library/cpp/streams/brotli/brotli_ut.cpp index fff37e3684..aeb2e284dc 100644 --- a/library/cpp/streams/brotli/brotli_ut.cpp +++ b/library/cpp/streams/brotli/brotli_ut.cpp @@ -5,28 +5,28 @@ #include <util/random/fast.h> Y_UNIT_TEST_SUITE(TBrotliTestSuite) { - TString Compress(TString data) { - TString compressed; - TStringOutput output(compressed); - TBrotliCompress compressStream(&output, 11); + TString Compress(TString data) { + TString compressed; + TStringOutput output(compressed); + TBrotliCompress compressStream(&output, 11); compressStream.Write(data.data(), data.size()); - compressStream.Finish(); - output.Finish(); - return compressed; - } - - TString Decompress(TString data) { - TStringInput input(data); - TBrotliDecompress decompressStream(&input); - return decompressStream.ReadAll(); - } - - void TestCase(const TString& s) { - UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s))); - } - - TString GenerateRandomString(size_t size) { - TReallyFastRng32 rng(42); + compressStream.Finish(); + output.Finish(); + return compressed; + } + + TString Decompress(TString data) { + TStringInput input(data); + TBrotliDecompress decompressStream(&input); + return decompressStream.ReadAll(); + } + + void TestCase(const TString& s) { + UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s))); + } + + TString GenerateRandomString(size_t size) { + TReallyFastRng32 rng(42); TString result; result.reserve(size + sizeof(ui64)); while (result.size() < size) { @@ -38,7 +38,7 @@ Y_UNIT_TEST_SUITE(TBrotliTestSuite) { } Y_UNIT_TEST(TestHelloWorld) { - TestCase("hello world"); + TestCase("hello world"); } Y_UNIT_TEST(TestFlush) { @@ -59,31 +59,31 @@ Y_UNIT_TEST_SUITE(TBrotliTestSuite) { } Y_UNIT_TEST(TestSeveralStreams) { - auto s1 = GenerateRandomString(1 << 15); - auto s2 = GenerateRandomString(1 << 15); - auto c1 = Compress(s1); - auto c2 = Compress(s2); - UNIT_ASSERT_VALUES_EQUAL(s1 + s2, Decompress(c1 + c2)); + auto s1 = GenerateRandomString(1 << 15); + auto s2 = GenerateRandomString(1 << 15); + auto c1 = Compress(s1); + auto c2 = Compress(s2); + UNIT_ASSERT_VALUES_EQUAL(s1 + s2, Decompress(c1 + c2)); } Y_UNIT_TEST(TestIncompleteStream) { - TString manyAs(64 * 1024, 'a'); - auto compressed = Compress(manyAs); + TString manyAs(64 * 1024, 'a'); + auto compressed = Compress(manyAs); TString truncated(compressed.data(), compressed.size() - 1); - UNIT_CHECK_GENERATED_EXCEPTION(Decompress(truncated), std::exception); + UNIT_CHECK_GENERATED_EXCEPTION(Decompress(truncated), std::exception); } Y_UNIT_TEST(Test64KB) { - auto manyAs = TString(64 * 1024, 'a'); - TString str("Hello from the Matrix!@#% How are you?}{\n\t\a"); - TestCase(manyAs + str + manyAs); + auto manyAs = TString(64 * 1024, 'a'); + TString str("Hello from the Matrix!@#% How are you?}{\n\t\a"); + TestCase(manyAs + str + manyAs); } Y_UNIT_TEST(Test1MB) { - TestCase(GenerateRandomString(1 * 1024 * 1024)); - } - + TestCase(GenerateRandomString(1 * 1024 * 1024)); + } + Y_UNIT_TEST(TestEmpty) { - TestCase(""); + TestCase(""); } } |