mirror of
https://github.com/zeromq/cppzmq.git
synced 2024-12-13 18:55:11 +01:00
83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
#include <gtest/gtest.h>
|
|
#include <gmock/gmock.h>
|
|
#include <zmq.hpp>
|
|
|
|
#ifdef ZMQ_CPP11
|
|
#include <thread>
|
|
#endif
|
|
|
|
class mock_monitor_t : public zmq::monitor_t
|
|
{
|
|
public:
|
|
MOCK_METHOD2(on_event_connect_delayed, void(const zmq_event_t &, const char *));
|
|
MOCK_METHOD2(on_event_connected, void(const zmq_event_t &, const char *));
|
|
};
|
|
|
|
TEST(monitor, create_destroy)
|
|
{
|
|
zmq::monitor_t monitor;
|
|
}
|
|
|
|
TEST(monitor, init_check)
|
|
{
|
|
zmq::context_t ctx;
|
|
zmq::socket_t bind_socket(ctx, ZMQ_DEALER);
|
|
|
|
bind_socket.bind("tcp://127.0.0.1:*");
|
|
char endpoint[255];
|
|
size_t endpoint_len = sizeof(endpoint);
|
|
bind_socket.getsockopt(ZMQ_LAST_ENDPOINT, &endpoint, &endpoint_len);
|
|
|
|
zmq::socket_t connect_socket(ctx, ZMQ_DEALER);
|
|
|
|
mock_monitor_t monitor;
|
|
EXPECT_CALL(monitor, on_event_connect_delayed(testing::_, testing::_))
|
|
.Times(testing::AtLeast(1));
|
|
EXPECT_CALL(monitor, on_event_connected(testing::_, testing::_))
|
|
.Times(testing::AtLeast(1));
|
|
|
|
monitor.init(connect_socket, "inproc://foo");
|
|
|
|
ASSERT_FALSE(monitor.check_event(0));
|
|
connect_socket.connect(endpoint);
|
|
|
|
while (monitor.check_event(100)) {
|
|
}
|
|
}
|
|
|
|
#ifdef ZMQ_CPP11
|
|
TEST(monitor, init_abort)
|
|
{
|
|
zmq::context_t ctx;
|
|
zmq::socket_t bind_socket(ctx, zmq::socket_type::dealer);
|
|
|
|
bind_socket.bind("tcp://127.0.0.1:*");
|
|
char endpoint[255];
|
|
size_t endpoint_len = sizeof(endpoint);
|
|
bind_socket.getsockopt(ZMQ_LAST_ENDPOINT, &endpoint, &endpoint_len);
|
|
|
|
zmq::socket_t connect_socket(ctx, zmq::socket_type::dealer);
|
|
|
|
mock_monitor_t monitor;
|
|
monitor.init(connect_socket, "inproc://foo");
|
|
EXPECT_CALL(monitor, on_event_connect_delayed(testing::_, testing::_))
|
|
.Times(testing::AtLeast(1));
|
|
EXPECT_CALL(monitor, on_event_connected(testing::_, testing::_))
|
|
.Times(testing::AtLeast(1));
|
|
|
|
auto thread = std::thread([&monitor] {
|
|
while (monitor.check_event(-1)) {
|
|
}
|
|
});
|
|
|
|
connect_socket.connect(endpoint);
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(250));
|
|
// TODO instead of sleeping an arbitrary amount of time, we should better
|
|
// wait until the expectations have met. How can this be done with
|
|
// googlemock?
|
|
|
|
monitor.abort();
|
|
thread.join();
|
|
}
|
|
#endif
|