Compressor: add zlib unittests

Signed-off-by: Alyona Kiseleva <akiselyova@mirantis.com>

3 files changed, 160 insertions, 0 deletions

diff --git a/src/test/compressor/Makefile.am b/src/test/compressor/Makefile.am
index f5a8f4e536e..4a41354b88e 100644
--- a/src/test/compressor/Makefile.am
+++ b/src/test/compressor/Makefile.am
@@ -45,5 +45,27 @@ unittest_compression_plugin_snappy_LDADD += -ldl
 endif
 check_TESTPROGRAMS += unittest_compression_plugin_snappy
 
+unittest_compression_zlib_SOURCES = \
+	test/compressor/test_compression_zlib.cc \
+	${zlib_sources}
+unittest_compression_zlib_CXXFLAGS = $(UNITTEST_CXXFLAGS)
+unittest_compression_zlib_LDADD = $(LIBOSD) $(LIBCOMMON) $(UNITTEST_LDADD) $(CEPH_GLOBAL)
+unittest_compression_zlib_LDFLAGS = -lz
+if LINUX
+unittest_compression_zlib_LDADD += -ldl
+endif
+check_TESTPROGRAMS += unittest_compression_zlib
+
+unittest_compression_plugin_zlib_SOURCES = \
+	test/compressor/test_compression_plugin_zlib.cc \
+	${zlib_sources}
+unittest_compression_plugin_zlib_CXXFLAGS = ${AM_CXXFLAGS} ${UNITTEST_CXXFLAGS}
+unittest_compression_plugin_zlib_LDADD = $(LIBOSD) $(LIBCOMMON) $(UNITTEST_LDADD) $(CEPH_GLOBAL) $(LIBCOMPRESSOR)
+unittest_compression_plugin_zlib_LDFLAGS = -lz
+if LINUX
+unittest_compression_plugin_zlib_LDADD += -ldl
+endif
+check_TESTPROGRAMS += unittest_compression_plugin_zlib
+
 endif # WITH_OSD
 endif # ENABLE_SERVER
diff --git a/src/test/compressor/test_compression_plugin_zlib.cc b/src/test/compressor/test_compression_plugin_zlib.cc
new file mode 100644
index 00000000000..6cf108547cc
--- /dev/null
+++ b/src/test/compressor/test_compression_plugin_zlib.cc
@@ -0,0 +1,57 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- 
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph distributed storage system
+ *
+ * Copyright (C) 2015 Mirantis, Inc.
+ *
+ * Author: Alyona Kiseleva <akiselyova@mirantis.com>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ * 
+ */
+
+#include <errno.h>
+#include <signal.h>
+#include <stdlib.h>
+#include <gtest/gtest.h>
+#include "global/global_init.h"
+#include "compressor/Compressor.h"
+#include "common/ceph_argparse.h"
+#include "global/global_context.h"
+#include "common/config.h"
+
+TEST(CompressionPluginZlib, all)
+{
+  CompressorRef compressor = Compressor::create(g_ceph_context, "zlib");
+  EXPECT_TRUE(compressor.get());
+}
+
+int main(int argc, char **argv) {
+  vector<const char*> args;
+  argv_to_vec(argc, (const char **)argv, args);
+
+  global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT, CODE_ENVIRONMENT_UTILITY, 0);
+  common_init_finish(g_ceph_context);
+
+  system("mkdir -p .libs/compressor");
+  system("cp .libs/libceph_zlib.so* .libs/compressor/");
+
+  g_conf->set_val("plugin_dir", ".libs", false, false);
+
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}
+
+/*
+ * Local Variables:
+ * compile-command: "cd ../.. ; make -j4 && 
+ *   make unittest_compression_plugin_zlib && 
+ *   valgrind --tool=memcheck \
+ *      ./unittest_compression_plugin_zlib \
+ *      --gtest_filter=*.* --log-to-stderr=true --debug-osd=20"
+ * End:
+ */
diff --git a/src/test/compressor/test_compression_zlib.cc b/src/test/compressor/test_compression_zlib.cc
new file mode 100644
index 00000000000..aba2f442f54
--- /dev/null
+++ b/src/test/compressor/test_compression_zlib.cc
@@ -0,0 +1,81 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- 
+// vim: ts=8 sw=2 smarttab
+/*
+ * Ceph distributed storage system
+ *
+ * Copyright (C) 2015 Mirantis, Inc.
+ *
+ * Author: Alyona Kiseleva <akiselyova@mirantis.com>
+ *
+ *  This library is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public
+ *  License as published by the Free Software Foundation; either
+ *  version 2.1 of the License, or (at your option) any later version.
+ * 
+ */
+
+#include <errno.h>
+#include <string.h>
+#include <gtest/gtest.h>
+#include "global/global_init.h"
+#include "compressor/zlib/CompressionZlib.h"
+#include "common/ceph_argparse.h"
+#include "global/global_context.h"
+#include "common/config.h"
+
+TEST(CompressionZlib, compress_decompress)
+{
+  CompressionZlib sp;
+  EXPECT_EQ(sp.get_method_name(), "zlib");
+  char* test = "This is test text";
+  int len = strlen(test);
+  bufferlist in, out;
+  in.append(test, len);
+  int res = sp.compress(in, out);
+  EXPECT_EQ(res, 0);
+  bufferlist after;
+  res = sp.decompress(out, after);
+  EXPECT_EQ(res, 0);
+  EXPECT_STREQ(test, after.c_str());
+}
+
+TEST(CompressionZlib, compress_decompress_chunk)
+{
+  CompressionZlib sp;
+  EXPECT_EQ(sp.get_method_name(), "zlib");
+  char* test = "This is test text";
+  buffer::ptr test2 ("1234567890", 10);
+  int len = strlen(test);
+  bufferlist in, out;
+  in.append(test, len);
+  in.append(test2);
+  int res = sp.compress(in, out);
+  EXPECT_EQ(res, 0);
+  bufferlist after;
+  res = sp.decompress(out, after);
+  EXPECT_EQ(res, 0);
+  EXPECT_STREQ("This is test text1234567890", after.c_str());
+}
+
+int main(int argc, char **argv) {
+  vector<const char*> args;
+  argv_to_vec(argc, (const char **)argv, args);
+
+  global_init(NULL, args, CEPH_ENTITY_TYPE_CLIENT, CODE_ENVIRONMENT_UTILITY, 0);
+  common_init_finish(g_ceph_context);
+
+  g_conf->set_val("compression_dir", ".libs", false, false);
+
+  ::testing::InitGoogleTest(&argc, argv);
+  return RUN_ALL_TESTS();
+}
+
+/*
+ * Local Variables:
+ * compile-command: "cd ../.. ; make -j4 && 
+ *   make unittest_compression_zlib && 
+ *   valgrind --tool=memcheck \
+ *      ./unittest_compression_zlib \
+ *      --gtest_filter=*.* --log-to-stderr=true --debug-osd=20"
+ * End:
+ */