test: add unit test for seastar buffer

Signed-off-by: Casey Bodley <cbodley@redhat.com>

diff --git a/src/test/CMakeLists.txt b/src/test/CMakeLists.txt
index cafe67879cc64319abf994881f5d386941a44287..7ddb6c697c8ebd2c648e3fa34ed2d6a2c4bc8ed9 100644 (file)
--- a/src/test/CMakeLists.txt
+++ b/src/test/CMakeLists.txt
@@ -54,6 +54,9 @@ endif(${WITH_RADOSGW})
 if(WITH_RBD)
   add_subdirectory(rbd_mirror)
 endif(WITH_RBD)
+if(WITH_SEASTAR)
+  add_subdirectory(crimson)
+endif()
 add_subdirectory(system)
 if(WITH_FIO OR WITH_SYSTEM_FIO)
   add_subdirectory(fio)

diff --git a/src/test/crimson/CMakeLists.txt b/src/test/crimson/CMakeLists.txt
new file mode 100644 (file)
index 0000000..b10ed9c
--- /dev/null
+++ b/src/test/crimson/CMakeLists.txt
@@ -0,0 +1,6 @@
+set(test_buffer_srcs
+  test_buffer.cc
+  $<TARGET_OBJECTS:seastar_buffer_obj>)
+add_executable(unittest_seastar_buffer ${test_buffer_srcs})
+add_ceph_unittest(unittest_seastar_buffer)
+target_link_libraries(unittest_seastar_buffer ceph-common Seastar::seastar)

diff --git a/src/test/crimson/test_buffer.cc b/src/test/crimson/test_buffer.cc
new file mode 100644 (file)
index 0000000..8de84cc
--- /dev/null
+++ b/src/test/crimson/test_buffer.cc
@@ -0,0 +1,49 @@
+#include "include/buffer.h"
+#include <core/app-template.hh>
+#include <core/future-util.hh>
+#include <core/reactor.hh>
+
+// allocate a foreign buffer on each cpu, collect them all into a bufferlist,
+// and destruct it on this cpu
+seastar::future<> test_foreign_bufferlist()
+{
+  auto make_foreign_buffer = [] (unsigned cpu) {
+    return seastar::smp::submit_to(cpu, [=] {
+      bufferlist bl;
+      seastar::temporary_buffer<char> buf("abcd", 4);
+      bl.append(buffer::create(std::move(buf)));
+      return bl;
+    });
+  };
+  auto reduce = [] (bufferlist&& lhs, bufferlist&& rhs) {
+    bufferlist bl;
+    bl.claim_append(lhs);
+    bl.claim_append(rhs);
+    return bl;
+  };
+  return seastar::map_reduce(seastar::smp::all_cpus(), make_foreign_buffer,
+                             bufferlist(), reduce).then(
+    [] (bufferlist&& bl) {
+      if (bl.length() != 4 * seastar::smp::count) {
+        auto e = std::make_exception_ptr(std::runtime_error("wrong buffer size"));
+        return seastar::make_exception_future<>(e);
+      }
+      bl.clear();
+      return seastar::make_ready_future<>();
+    });
+}
+
+int main(int argc, char** argv)
+{
+  seastar::app_template app;
+  return app.run(argc, argv, [] {
+    return seastar::now().then(
+      &test_foreign_bufferlist
+    ).then([] {
+      std::cout << "All tests succeeded" << std::endl;
+    }).handle_exception([] (auto eptr) {
+      std::cout << "Test failure" << std::endl;
+      return seastar::make_exception_future<>(eptr);
+    });
+  });
+}