generic-library/libzmq
1
0
Fork 0
mirror of https://github.com/zeromq/libzmq.git synced 2024-12-12 18:40:27 +01:00
Code Issues Projects Releases Wiki Activity

Problem: no unit tests for udp_address.cpp

Browse Source 
Solution: add unit tests
This commit is contained in:
Lionel Flandrin 2018-05-02 23:22:06 +02:00
parent 801559c504
commit 0b36b84243
3 changed files with 170 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
Makefile.am
View File

@ -898,7 +898,8 @@ test_apps += \
unittests/unittest_poller \
unittests/unittest_ypipe \
unittests/unittest_mtrie \
unittests/unittest_ip_resolver
unittests/unittest_ip_resolver \
unittests/unittest_udp_address
unittests_unittest_poller_SOURCES = unittests/unittest_poller.cpp
unittests_unittest_poller_CPPFLAGS = -I$(top_srcdir)/src ${UNITY_CPPFLAGS} $(CODE_COVERAGE_CPPFLAGS)
@ -931,6 +932,14 @@ unittests_unittest_ip_resolver_LDADD = $(top_builddir)/src/.libs/libzmq.a \
${src_libzmq_la_LIBADD} \
${UNITY_LIBS} \
$(CODE_COVERAGE_LDFLAGS)
unittests_unittest_udp_address_SOURCES = unittests/unittest_udp_address.cpp
unittests_unittest_udp_address_CPPFLAGS = -I$(top_srcdir)/src ${UNITY_CPPFLAGS} $(CODE_COVERAGE_CPPFLAGS)
unittests_unittest_udp_address_CXXFLAGS = $(CODE_COVERAGE_CXXFLAGS)
unittests_unittest_udp_address_LDADD = $(top_builddir)/src/.libs/libzmq.a \
${src_libzmq_la_LIBADD} \
${UNITY_LIBS} \
$(CODE_COVERAGE_LDFLAGS)
endif
check_PROGRAMS = ${test_apps}

1
unittests/CMakeLists.txt
View File

@ -6,6 +6,7 @@ set(unittests
unittest_poller
unittest_mtrie
unittest_ip_resolver
unittest_udp_address
)
#IF (ENABLE_DRAFTS)

159
unittests/unittest_udp_address.cpp Normal file
View File

@ -0,0 +1,159 @@
/*
Copyright (c) 2018 Contributors as noted in the AUTHORS file
This file is part of 0MQ.
0MQ 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 3 of the License, or
(at your option) any later version.
0MQ is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <unity.h>
#include "../tests/testutil.hpp"
#include <ip.hpp>
#include <udp_address.hpp>
void setUp ()
{
}
void tearDown ()
{
}
// Test an UDP address resolution. If 'bind_addr_' is not NULL
// request a bind address. If 'dest_addr_' is NULL assume the
// resolution is supposed to fail.
static void test_resolve (const char *name_, const char *dest_addr_,
uint16_t expected_port_ = 0,
const char *bind_addr_ = NULL,
bool multicast_ = false)
{
zmq::udp_address_t addr;
bool bound = bind_addr_ != NULL;
int rc = addr.resolve (name_, bound);
if (dest_addr_ == NULL) {
TEST_ASSERT_EQUAL (-1, rc);
TEST_ASSERT_EQUAL (EINVAL, errno);
return;
} else {
TEST_ASSERT_EQUAL (0, rc);
}
TEST_ASSERT_EQUAL (multicast_, addr.is_mcast ());
struct sockaddr_in *dest = (struct sockaddr_in *)addr.dest_addr ();
struct in_addr expected_dest;
assert (test_inet_pton (AF_INET, dest_addr_, &expected_dest) == 1);
TEST_ASSERT_EQUAL (AF_INET, dest->sin_family);
TEST_ASSERT_EQUAL (expected_dest.s_addr, dest->sin_addr.s_addr);
TEST_ASSERT_EQUAL (htons (expected_port_), dest->sin_port);
struct sockaddr_in *bind = (struct sockaddr_in *)addr.bind_addr ();
struct in_addr expected_bind;
if (bind_addr_ == NULL) {
// Bind ANY
bind_addr_ = "0.0.0.0";
}
assert (test_inet_pton (AF_INET, bind_addr_, &expected_bind) == 1);
TEST_ASSERT_EQUAL (AF_INET, bind->sin_family);
TEST_ASSERT_EQUAL (expected_bind.s_addr, bind->sin_addr.s_addr);
TEST_ASSERT_EQUAL (htons (expected_port_), bind->sin_port);
}
static void test_resolve_ipv4_simple ()
{
test_resolve ("127.0.0.1:5555", "127.0.0.1", 5555);
}
static void test_resolve_ipv4_bind ()
{
test_resolve ("127.0.0.1:5555", "127.0.0.1", 5555, "127.0.0.1");
}
static void test_resolve_ipv4_bind_any ()
{
// Wildcard port not supported
test_resolve ("*:*", NULL, 0, "0.0.0.0");
}
static void test_resolve_ipv4_bind_anyport ()
{
// Wildcard port not supported
test_resolve ("127.0.0.1:*", NULL, 0, "127.0.0.1");
}
static void test_resolve_ipv4_bind_any_port ()
{
test_resolve ("*:5555", "0.0.0.0", 5555, "0.0.0.0");
}
static void test_resolve_ipv4_connect_any ()
{
// Cannot use wildcard for connection
test_resolve ("*:5555", NULL);
}
static void test_resolve_ipv4_connect_anyport ()
{
test_resolve ("127.0.0.1:*", NULL);
}
static void test_resolve_ipv4_bind_mcast ()
{
test_resolve ("239.0.0.1:1234", "239.0.0.1", 1234, "0.0.0.0", true);
}
static void test_resolve_ipv4_connect_mcast ()
{
test_resolve ("239.0.0.1:2222", "239.0.0.1", 2222, NULL, true);
}
static void test_resolve_ipv6_simple ()
{
if (!is_ipv6_available ()) {
TEST_IGNORE_MESSAGE ("ipv6 is not available");
}
// IPv6 not yet supported
test_resolve ("::1", NULL);
}
int main (void)
{
zmq::initialize_network ();
setup_test_environment ();
UNITY_BEGIN ();
RUN_TEST (test_resolve_ipv4_simple);
RUN_TEST (test_resolve_ipv4_bind);
RUN_TEST (test_resolve_ipv4_bind_any);
RUN_TEST (test_resolve_ipv4_bind_anyport);
RUN_TEST (test_resolve_ipv4_bind_any_port);
RUN_TEST (test_resolve_ipv4_connect_any);
RUN_TEST (test_resolve_ipv4_connect_anyport);
RUN_TEST (test_resolve_ipv4_bind_mcast);
RUN_TEST (test_resolve_ipv4_connect_mcast);
RUN_TEST (test_resolve_ipv6_simple);
zmq::shutdown_network ();
return UNITY_END ();
}