Merge commit '31cbcd39be6aef3ed43121da5b797d8ec9b0fd31' as 'libs/cppzmq'
This commit is contained in:
121
libs/cppzmq/tests/send_multipart.cpp
Normal file
121
libs/cppzmq/tests/send_multipart.cpp
Normal file
@@ -0,0 +1,121 @@
|
||||
#include <catch2/catch.hpp>
|
||||
#include <zmq_addon.hpp>
|
||||
#ifdef ZMQ_CPP11
|
||||
|
||||
#include <forward_list>
|
||||
|
||||
TEST_CASE("send_multipart test", "[send_multipart]")
|
||||
{
|
||||
zmq::context_t context(1);
|
||||
zmq::socket_t output(context, ZMQ_PAIR);
|
||||
zmq::socket_t input(context, ZMQ_PAIR);
|
||||
output.bind("inproc://multipart.test");
|
||||
input.connect("inproc://multipart.test");
|
||||
|
||||
SECTION("send 0 messages") {
|
||||
std::vector<zmq::message_t> imsgs;
|
||||
auto iret = zmq::send_multipart(input, imsgs);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 0);
|
||||
}
|
||||
SECTION("send 1 message") {
|
||||
std::array<zmq::message_t, 1> imsgs = {zmq::message_t(3)};
|
||||
auto iret = zmq::send_multipart(input, imsgs);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 1);
|
||||
|
||||
std::vector<zmq::message_t> omsgs;
|
||||
auto oret = zmq::recv_multipart(output, std::back_inserter(omsgs));
|
||||
REQUIRE(oret);
|
||||
CHECK(*oret == 1);
|
||||
REQUIRE(omsgs.size() == 1);
|
||||
CHECK(omsgs[0].size() == 3);
|
||||
}
|
||||
SECTION("send 2 messages") {
|
||||
std::array<zmq::message_t, 2> imsgs = {zmq::message_t(3), zmq::message_t(4)};
|
||||
auto iret = zmq::send_multipart(input, imsgs);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
|
||||
std::vector<zmq::message_t> omsgs;
|
||||
auto oret = zmq::recv_multipart(output, std::back_inserter(omsgs));
|
||||
REQUIRE(oret);
|
||||
CHECK(*oret == 2);
|
||||
REQUIRE(omsgs.size() == 2);
|
||||
CHECK(omsgs[0].size() == 3);
|
||||
CHECK(omsgs[1].size() == 4);
|
||||
}
|
||||
SECTION("send 2 messages, const_buffer") {
|
||||
std::array<zmq::const_buffer, 2> imsgs = {zmq::str_buffer("foo"),
|
||||
zmq::str_buffer("bar!")};
|
||||
auto iret = zmq::send_multipart(input, imsgs);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
|
||||
std::vector<zmq::message_t> omsgs;
|
||||
auto oret = zmq::recv_multipart(output, std::back_inserter(omsgs));
|
||||
REQUIRE(oret);
|
||||
CHECK(*oret == 2);
|
||||
REQUIRE(omsgs.size() == 2);
|
||||
CHECK(omsgs[0].size() == 3);
|
||||
CHECK(omsgs[1].size() == 4);
|
||||
}
|
||||
SECTION("send 2 messages, mutable_buffer") {
|
||||
char buf[4] = {};
|
||||
std::array<zmq::mutable_buffer, 2> imsgs = {
|
||||
zmq::buffer(buf, 3), zmq::buffer(buf)};
|
||||
auto iret = zmq::send_multipart(input, imsgs);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
|
||||
std::vector<zmq::message_t> omsgs;
|
||||
auto oret = zmq::recv_multipart(output, std::back_inserter(omsgs));
|
||||
REQUIRE(oret);
|
||||
CHECK(*oret == 2);
|
||||
REQUIRE(omsgs.size() == 2);
|
||||
CHECK(omsgs[0].size() == 3);
|
||||
CHECK(omsgs[1].size() == 4);
|
||||
}
|
||||
SECTION("send 2 messages, dontwait") {
|
||||
zmq::socket_t push(context, ZMQ_PUSH);
|
||||
push.bind("inproc://multipart.test.push");
|
||||
|
||||
std::array<zmq::message_t, 2> imsgs = {zmq::message_t(3), zmq::message_t(4)};
|
||||
auto iret = zmq::send_multipart(push, imsgs, zmq::send_flags::dontwait);
|
||||
REQUIRE_FALSE(iret);
|
||||
}
|
||||
SECTION("send, misc. containers") {
|
||||
std::vector<zmq::message_t> msgs_vec;
|
||||
msgs_vec.emplace_back(3);
|
||||
msgs_vec.emplace_back(4);
|
||||
auto iret = zmq::send_multipart(input, msgs_vec);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
|
||||
std::forward_list<zmq::message_t> msgs_list;
|
||||
msgs_list.emplace_front(4);
|
||||
msgs_list.emplace_front(3);
|
||||
iret = zmq::send_multipart(input, msgs_list);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
|
||||
// init. list
|
||||
const auto msgs_il = {zmq::str_buffer("foo"), zmq::str_buffer("bar!")};
|
||||
iret = zmq::send_multipart(input, msgs_il);
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
// rvalue
|
||||
iret = zmq::send_multipart(input,
|
||||
std::initializer_list<zmq::const_buffer>{
|
||||
zmq::str_buffer("foo"),
|
||||
zmq::str_buffer("bar!")});
|
||||
REQUIRE(iret);
|
||||
CHECK(*iret == 2);
|
||||
}
|
||||
SECTION("send with invalid socket") {
|
||||
std::vector<zmq::message_t> msgs(1);
|
||||
CHECK_THROWS_AS(zmq::send_multipart(zmq::socket_ref(), msgs),
|
||||
zmq::error_t);
|
||||
}
|
||||
}
|
||||
#endif
|
||||
Reference in New Issue
Block a user