aboutsummaryrefslogtreecommitdiffstats
path: root/library/cpp/streams/zstd/zstd_ut.cpp
diff options
context:
space:
mode:
authorbulatman <bulatman@yandex-team.ru>2022-02-10 16:45:50 +0300
committerDaniil Cherednik <dcherednik@yandex-team.ru>2022-02-10 16:45:50 +0300
commit2f6ca198245aeffd5e2d82b65927c2465b68b4f5 (patch)
tree9142afc54d335ea52910662635b898e79e192e49 /library/cpp/streams/zstd/zstd_ut.cpp
parent6560e4993b14d193f8c879e33a3de5e5eba6e21d (diff)
downloadydb-2f6ca198245aeffd5e2d82b65927c2465b68b4f5.tar.gz
Restoring authorship annotation for <bulatman@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/streams/zstd/zstd_ut.cpp')
-rw-r--r--library/cpp/streams/zstd/zstd_ut.cpp186
1 files changed, 93 insertions, 93 deletions
diff --git a/library/cpp/streams/zstd/zstd_ut.cpp b/library/cpp/streams/zstd/zstd_ut.cpp
index 01618193fa..ef479fdd97 100644
--- a/library/cpp/streams/zstd/zstd_ut.cpp
+++ b/library/cpp/streams/zstd/zstd_ut.cpp
@@ -1,94 +1,94 @@
-#include "zstd.h"
-
+#include "zstd.h"
+
#include <library/cpp/testing/unittest/registar.h>
-
-#include <util/random/fast.h>
-#include <util/stream/null.h>
-#include <util/stream/str.h>
-
-Y_UNIT_TEST_SUITE(TZstdTestSuite) {
- TString Compress(TString data, int quality = -1) {
- TString compressed;
- TStringOutput output(compressed);
- TZstdCompress compressStream(&output, quality);
- compressStream.Write(data.data(), data.size());
- compressStream.Finish();
- output.Finish();
- return compressed;
- }
-
- TString Decompress(TString data) {
- TStringInput input(data);
- TZstdDecompress decompressStream(&input);
- return decompressStream.ReadAll();
- }
-
- void TestCase(const TString& s) {
- UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, -1)));
- UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 0)));
- UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 22)));
- UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 11)));
- UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 100500)));
- }
-
- TString GenerateRandomString(size_t size) {
- TReallyFastRng32 rng(42);
- TString result;
- result.reserve(size + sizeof(ui64));
- while (result.size() < size) {
- ui64 value = rng.GenRand64();
- result += TStringBuf(reinterpret_cast<const char*>(&value), sizeof(value));
- }
- result.resize(size);
- return result;
- }
-
- Y_UNIT_TEST(TestHelloWorld) {
- TestCase("hello world");
- }
-
- Y_UNIT_TEST(TestSeveralStreamsWithSameQuality) {
- 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(TestSeveralStreamsWithDifferentQuality) {
- auto s1 = GenerateRandomString(1 << 15);
- auto s2 = GenerateRandomString(1 << 15);
- auto c1 = Compress(s1, 1);
- auto c2 = Compress(s2, 2);
- UNIT_ASSERT_VALUES_EQUAL(s1 + s2, Decompress(c1 + c2));
- }
-
- Y_UNIT_TEST(TestIncompleteStream) {
- TString manyAs(64 * 1024, 'a');
- auto compressed = Compress(manyAs);
- TString truncated(compressed.data(), compressed.size() - 1);
- 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);
- }
-
- Y_UNIT_TEST(Test1MB) {
- TestCase(GenerateRandomString(1 * 1024 * 1024));
- }
-
- Y_UNIT_TEST(TestEmpty) {
- TestCase("");
- }
-
- Y_UNIT_TEST(TestWriteAfterFinish) {
- TNullOutput output;
- TZstdCompress compressStream(&output);
- compressStream.Finish();
- UNIT_ASSERT_EXCEPTION_CONTAINS(compressStream.Write("a", 1), std::exception, "Cannot use stream after finish.");
- UNIT_ASSERT_EXCEPTION_CONTAINS(compressStream.Flush(), std::exception, "Cannot use stream after finish.");
- }
-}
+
+#include <util/random/fast.h>
+#include <util/stream/null.h>
+#include <util/stream/str.h>
+
+Y_UNIT_TEST_SUITE(TZstdTestSuite) {
+ TString Compress(TString data, int quality = -1) {
+ TString compressed;
+ TStringOutput output(compressed);
+ TZstdCompress compressStream(&output, quality);
+ compressStream.Write(data.data(), data.size());
+ compressStream.Finish();
+ output.Finish();
+ return compressed;
+ }
+
+ TString Decompress(TString data) {
+ TStringInput input(data);
+ TZstdDecompress decompressStream(&input);
+ return decompressStream.ReadAll();
+ }
+
+ void TestCase(const TString& s) {
+ UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, -1)));
+ UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 0)));
+ UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 22)));
+ UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 11)));
+ UNIT_ASSERT_VALUES_EQUAL(s, Decompress(Compress(s, 100500)));
+ }
+
+ TString GenerateRandomString(size_t size) {
+ TReallyFastRng32 rng(42);
+ TString result;
+ result.reserve(size + sizeof(ui64));
+ while (result.size() < size) {
+ ui64 value = rng.GenRand64();
+ result += TStringBuf(reinterpret_cast<const char*>(&value), sizeof(value));
+ }
+ result.resize(size);
+ return result;
+ }
+
+ Y_UNIT_TEST(TestHelloWorld) {
+ TestCase("hello world");
+ }
+
+ Y_UNIT_TEST(TestSeveralStreamsWithSameQuality) {
+ 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(TestSeveralStreamsWithDifferentQuality) {
+ auto s1 = GenerateRandomString(1 << 15);
+ auto s2 = GenerateRandomString(1 << 15);
+ auto c1 = Compress(s1, 1);
+ auto c2 = Compress(s2, 2);
+ UNIT_ASSERT_VALUES_EQUAL(s1 + s2, Decompress(c1 + c2));
+ }
+
+ Y_UNIT_TEST(TestIncompleteStream) {
+ TString manyAs(64 * 1024, 'a');
+ auto compressed = Compress(manyAs);
+ TString truncated(compressed.data(), compressed.size() - 1);
+ 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);
+ }
+
+ Y_UNIT_TEST(Test1MB) {
+ TestCase(GenerateRandomString(1 * 1024 * 1024));
+ }
+
+ Y_UNIT_TEST(TestEmpty) {
+ TestCase("");
+ }
+
+ Y_UNIT_TEST(TestWriteAfterFinish) {
+ TNullOutput output;
+ TZstdCompress compressStream(&output);
+ compressStream.Finish();
+ UNIT_ASSERT_EXCEPTION_CONTAINS(compressStream.Write("a", 1), std::exception, "Cannot use stream after finish.");
+ UNIT_ASSERT_EXCEPTION_CONTAINS(compressStream.Flush(), std::exception, "Cannot use stream after finish.");
+ }
+}