From d064aeaede309a53320e89dbf6fb70efa260536f Mon Sep 17 00:00:00 2001 From: Casey Bodley Date: Sun, 8 Oct 2017 16:28:30 -0400 Subject: [PATCH] test: add unit test for seastar buffer Signed-off-by: Casey Bodley --- src/test/CMakeLists.txt | 3 ++ src/test/crimson/CMakeLists.txt | 6 ++++ src/test/crimson/test_buffer.cc | 49 +++++++++++++++++++++++++++++++++ 3 files changed, 58 insertions(+) create mode 100644 src/test/crimson/CMakeLists.txt create mode 100644 src/test/crimson/test_buffer.cc diff --git a/src/test/CMakeLists.txt b/src/test/CMakeLists.txt index cafe67879cc..7ddb6c697c8 100644 --- 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 index 00000000000..b10ed9c245a --- /dev/null +++ b/src/test/crimson/CMakeLists.txt @@ -0,0 +1,6 @@ +set(test_buffer_srcs + test_buffer.cc + $) +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 index 00000000000..8de84cce409 --- /dev/null +++ b/src/test/crimson/test_buffer.cc @@ -0,0 +1,49 @@ +#include "include/buffer.h" +#include +#include +#include + +// 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 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); + }); + }); +} -- 2.39.5