intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 41 insertions, 0 deletions

diff --git a/library/cpp/streams/bzip2/bzip2_ut.cpp b/library/cpp/streams/bzip2/bzip2_ut.cpp
new file mode 100644
index 0000000000..69a98f296c
--- /dev/null
+++ b/library/cpp/streams/bzip2/bzip2_ut.cpp
@@ -0,0 +1,41 @@
+#include "bzip2.h"
+
+#include <library/cpp/testing/unittest/registar.h>
+
+#include <util/stream/file.h>
+#include <util/system/tempfile.h>
+
+#define ZDATA "./zdata"
+
+Y_UNIT_TEST_SUITE(TBZipTest) {
+    static const TString data = "8s7d5vc6s5vc67sa4c65ascx6asd4xcv76adsfxv76s";
+
+    Y_UNIT_TEST(TestCompress) {
+        TUnbufferedFileOutput o(ZDATA);
+        TBZipCompress c(&o);
+
+        c.Write(data.data(), data.size());
+        c.Finish();
+        o.Finish();
+    }
+
+    Y_UNIT_TEST(TestDecompress) {
+        TTempFile tmp(ZDATA);
+
+        {
+            TUnbufferedFileInput i(ZDATA);
+            TBZipDecompress d(&i);
+
+            UNIT_ASSERT_EQUAL(d.ReadLine(), data);
+        }
+    }
+
+    Y_UNIT_TEST(TestCorrupted) {
+        TMemoryInput i("blablabla", 10);
+        TBZipDecompress d(&i);
+
+        UNIT_ASSERT_EXCEPTION(d.ReadLine(), TBZipDecompressError);
+    }
+}
+
+#undef ZDATA