From e2d926fe33573a27fa5f8888cedc61cc23f35fa3 Mon Sep 17 00:00:00 2001 From: Ricardo Dias Date: Mon, 22 Feb 2016 12:06:44 +0000 Subject: [PATCH] mon: MonMap: Added unit tests to test MonMap build_initial The dns_resolve unit test tests the DNSResolver wrapper, and the MonMap unit test tests the construction of monmap using DNS SRV records. Signed-off-by: Ricardo Dias --- src/test/Makefile.am | 10 ++ src/test/common/dns_messages.h | 101 +++++++++++++ src/test/common/dns_resolve.cc | 255 +++++++++++++++++++++++++++++++++ src/test/mon/MonMap.cc | 220 ++++++++++++++++++++++++++++ 4 files changed, 586 insertions(+) create mode 100644 src/test/common/dns_messages.h create mode 100644 src/test/common/dns_resolve.cc create mode 100644 src/test/mon/MonMap.cc diff --git a/src/test/Makefile.am b/src/test/Makefile.am index 885a0cdf8dc4a..a50e9f8ac6edb 100644 --- a/src/test/Makefile.am +++ b/src/test/Makefile.am @@ -452,6 +452,16 @@ unittest_async_compressor_CXXFLAGS = $(UNITTEST_CXXFLAGS) unittest_async_compressor_LDADD = $(UNITTEST_LDADD) $(CEPH_GLOBAL) $(LIBCOMPRESSOR) $(LIBCOMMON) check_PROGRAMS += unittest_async_compressor +unittest_dns_resolve_SOURCES = test/common/dns_resolve.cc +unittest_dns_resolve_LDADD = $(UNITTEST_LDADD) $(CEPH_GLOBAL) $(RESOLV_LIBS) +unittest_dns_resolve_CXXFLAGS = $(UNITTEST_CXXFLAGS) +check_TESTPROGRAMS += unittest_dns_resolve + +unittest_monmap_SOURCES = test/mon/MonMap.cc +unittest_monmap_LDADD = $(UNITTEST_LDADD) $(CEPH_GLOBAL) $(RESOLV_LIBS) +unittest_monmap_CXXFLAGS = $(UNITTEST_CXXFLAGS) +check_TESTPROGRAMS += unittest_monmap + if LINUX # # XXX FREEBSD diff --git a/src/test/common/dns_messages.h b/src/test/common/dns_messages.h new file mode 100644 index 0000000000000..d0b5a13d98ffc --- /dev/null +++ b/src/test/common/dns_messages.h @@ -0,0 +1,101 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2016 SUSE LINUX GmbH + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ +#ifndef CEPH_TEST_DNS_MESSAGES_H +#define CEPH_TEST_DNS_MESSAGES_H + +#include "common/dns_resolve.h" +#include "test/unit.h" +#include "gmock/gmock.h" + +u_char ns_search_msg_ok_payload[] = { + 0x00, 0x55, 0x85, 0x80, 0x00, 0x01, 0x00, 0x03, 0x00, 0x02, 0x00, 0x05, 0x09, + 0x5F, 0x63, 0x65, 0x70, 0x68, 0x2D, 0x6D, 0x6F, 0x6E, 0x04, 0x5F, 0x74, 0x63, + 0x70, 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, 0x6D, 0x00, 0x00, 0x21, + 0x00, 0x01, 0xC0, 0x0C, 0x00, 0x21, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, + 0x16, 0x00, 0x0A, 0x00, 0x28, 0x1A, 0x85, 0x03, 0x6D, 0x6F, 0x6E, 0x01, 0x61, + 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, 0x6D, 0x00, 0xC0, 0x0C, 0x00, + 0x21, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x16, 0x00, 0x0A, 0x00, 0x19, + 0x1A, 0x85, 0x03, 0x6D, 0x6F, 0x6E, 0x01, 0x63, 0x04, 0x63, 0x65, 0x70, 0x68, + 0x03, 0x63, 0x6F, 0x6D, 0x00, 0xC0, 0x0C, 0x00, 0x21, 0x00, 0x01, 0x00, 0x09, + 0x3A, 0x80, 0x00, 0x16, 0x00, 0x0A, 0x00, 0x23, 0x1A, 0x85, 0x03, 0x6D, 0x6F, + 0x6E, 0x01, 0x62, 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, 0x6D, 0x00, + 0xC0, 0x85, 0x00, 0x02, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, 0x03, + 0x6E, 0x73, 0x32, 0xC0, 0x85, 0xC0, 0x85, 0x00, 0x02, 0x00, 0x01, 0x00, 0x09, + 0x3A, 0x80, 0x00, 0x06, 0x03, 0x6E, 0x73, 0x31, 0xC0, 0x85, 0xC0, 0x5D, 0x00, + 0x01, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x0D, + 0xC0, 0x7F, 0x00, 0x01, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, + 0xA8, 0x01, 0x0C, 0xC0, 0x3B, 0x00, 0x01, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, + 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x0B, 0xC0, 0xAD, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x59, 0xC0, 0x9B, 0x00, 0x01, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0xFE +}; + + +u_char ns_query_msg_mon_c_payload[] = { + 0x46, 0x4D, 0x85, 0x80, 0x00, 0x01, 0x00, 0x01, 0x00, 0x02, 0x00, 0x02, 0x03, + 0x6D, 0x6F, 0x6E, 0x01, 0x63, 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, + 0x6D, 0x00, 0x00, 0x01, 0x00, 0x01, 0xC0, 0x0C, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x0D, 0xC0, 0x12, 0x00, 0x02, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, 0x03, 0x6E, 0x73, 0x31, 0xC0, + 0x12, 0xC0, 0x12, 0x00, 0x02, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, + 0x03, 0x6E, 0x73, 0x32, 0xC0, 0x12, 0xC0, 0x3C, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x59, 0xC0, 0x4E, 0x00, 0x01, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0xFE +}; + +u_char ns_query_msg_mon_b_payload[] = { + 0x64, 0xCC, 0x85, 0x80, 0x00, 0x01, 0x00, 0x01, 0x00, 0x02, 0x00, 0x02, 0x03, + 0x6D, 0x6F, 0x6E, 0x01, 0x62, 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, + 0x6D, 0x00, 0x00, 0x01, 0x00, 0x01, 0xC0, 0x0C, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x0C, 0xC0, 0x12, 0x00, 0x02, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, 0x03, 0x6E, 0x73, 0x32, 0xC0, + 0x12, 0xC0, 0x12, 0x00, 0x02, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, + 0x03, 0x6E, 0x73, 0x31, 0xC0, 0x12, 0xC0, 0x4E, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x59, 0xC0, 0x3C, 0x00, 0x01, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0xFE +}; + +u_char ns_query_msg_mon_a_payload[] = { + 0x86, 0xAD, 0x85, 0x80, 0x00, 0x01, 0x00, 0x01, 0x00, 0x02, 0x00, 0x02, 0x03, + 0x6D, 0x6F, 0x6E, 0x01, 0x61, 0x04, 0x63, 0x65, 0x70, 0x68, 0x03, 0x63, 0x6F, + 0x6D, 0x00, 0x00, 0x01, 0x00, 0x01, 0xC0, 0x0C, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x0B, 0xC0, 0x12, 0x00, 0x02, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, 0x03, 0x6E, 0x73, 0x32, 0xC0, + 0x12, 0xC0, 0x12, 0x00, 0x02, 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x06, + 0x03, 0x6E, 0x73, 0x31, 0xC0, 0x12, 0xC0, 0x4E, 0x00, 0x01, 0x00, 0x01, 0x00, + 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0x59, 0xC0, 0x3C, 0x00, 0x01, + 0x00, 0x01, 0x00, 0x09, 0x3A, 0x80, 0x00, 0x04, 0xC0, 0xA8, 0x01, 0xFE +}; + +class MockResolvHWrapper : public ResolvHWrapper { + +public: +#ifdef HAVE_RES_NQUERY + MOCK_METHOD6(res_nquery, int(res_state s, const char *hostname, int cls, + int type, u_char *buf, int bufsz)); + + MOCK_METHOD6(res_nsearch, int(res_state s, const char *hostname, int cls, + int type, u_char *buf, int bufsz)); +#else + MOCK_METHOD5(res_query, int(const char *hostname, int cls, + int type, u_char *buf, int bufsz)); + + MOCK_METHOD5(res_search, int(const char *hostname, int cls, + int type, u_char *buf, int bufsz)); +#endif + +}; + + +#endif diff --git a/src/test/common/dns_resolve.cc b/src/test/common/dns_resolve.cc new file mode 100644 index 0000000000000..d3e0677b8dc9e --- /dev/null +++ b/src/test/common/dns_resolve.cc @@ -0,0 +1,255 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2016 SUSE LINUX GmbH + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ +#include "common/dns_resolve.h" +#include "test/common/dns_messages.h" + +#include "test/unit.h" +#include "common/debug.h" +#include "gmock/gmock.h" + + +#include + +#define TEST_DEBUG 20 + +#define dout_subsys ceph_subsys_ + + +using ::testing::Return; +using ::testing::_; +using ::testing::SetArrayArgument; +using ::testing::DoAll; +using ::testing::StrEq; + +class DNSResolverTest : public ::testing::Test { + protected: + virtual void SetUp() { + g_ceph_context->_conf->subsys.set_log_level(dout_subsys, TEST_DEBUG); + } + + virtual void TearDown() { + DNSResolver::get_instance(nullptr); + } +}; + +TEST_F(DNSResolverTest, resolve_ip_addr) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + + int lena = sizeof(ns_query_msg_mon_a_payload); +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); +#else + EXPECT_CALL(*resolvH, res_query(StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); +#endif + + entity_addr_t addr; + DNSResolver::get_instance(resolvH)->resolve_ip_addr(g_ceph_context, + "mon.a.ceph.com", &addr); + + std::ostringstream os; + os << addr; + ASSERT_EQ(os.str(), "192.168.1.11:0/0"); +} + +TEST_F(DNSResolverTest, resolve_ip_addr_fail) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("not_exists.com"), C_IN, T_A,_,_)) + .WillOnce(Return(0)); +#else + EXPECT_CALL(*resolvH, res_query(StrEq("not_exists.com"), C_IN, T_A,_,_)) + .WillOnce(Return(0)); +#endif + + entity_addr_t addr; + int ret = DNSResolver::get_instance(resolvH)->resolve_ip_addr(g_ceph_context, + "not_exists.com", &addr); + + ASSERT_EQ(ret, -1); + std::ostringstream os; + os << addr; + ASSERT_EQ(os.str(), ":/0"); +} + + +TEST_F(DNSResolverTest, resolve_srv_hosts_empty_domain) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + + + int len = sizeof(ns_search_msg_ok_payload); + int lena = sizeof(ns_query_msg_mon_a_payload); + int lenb = sizeof(ns_query_msg_mon_b_payload); + int lenc = sizeof(ns_query_msg_mon_c_payload); + + using ::testing::InSequence; + { + InSequence s; + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_cephmon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_nquery(_, StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_cephmon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#endif + } + + map addrs; + DNSResolver::get_instance(resolvH)->resolve_srv_hosts(g_ceph_context, "cephmon", + DNSResolver::SRV_Protocol::TCP, &addrs); + + ASSERT_EQ(addrs.size(), (unsigned int)3); + map::iterator it = addrs.find("mon.a"); + ASSERT_NE(it, addrs.end()); + std::ostringstream os; + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.11:6789/0"); + os.str(""); + it = addrs.find("mon.b"); + ASSERT_NE(it, addrs.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.12:6789/0"); + os.str(""); + it = addrs.find("mon.c"); + ASSERT_NE(it, addrs.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.13:6789/0"); +} + +TEST_F(DNSResolverTest, resolve_srv_hosts_full_domain) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + + + int len = sizeof(ns_search_msg_ok_payload); + int lena = sizeof(ns_query_msg_mon_a_payload); + int lenb = sizeof(ns_query_msg_mon_b_payload); + int lenc = sizeof(ns_query_msg_mon_c_payload); + + using ::testing::InSequence; + { + InSequence s; + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_cephmon._tcp.ceph.com"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_nquery(_, StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_cephmon._tcp.ceph.com"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#endif + } + + map addrs; + DNSResolver::get_instance(resolvH)->resolve_srv_hosts(g_ceph_context, "cephmon", + DNSResolver::SRV_Protocol::TCP, "ceph.com", &addrs); + + ASSERT_EQ(addrs.size(), (unsigned int)3); + map::iterator it = addrs.find("mon.a"); + ASSERT_NE(it, addrs.end()); + std::ostringstream os; + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.11:6789/0"); + os.str(""); + it = addrs.find("mon.b"); + ASSERT_NE(it, addrs.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.12:6789/0"); + os.str(""); + it = addrs.find("mon.c"); + ASSERT_NE(it, addrs.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.13:6789/0"); +} + +TEST_F(DNSResolverTest, resolve_srv_hosts_fail) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + + + using ::testing::InSequence; + { + InSequence s; + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_noservice._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(Return(0)); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_noservice._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(Return(0)); +#endif + } + + map addrs; + int ret = DNSResolver::get_instance(resolvH)->resolve_srv_hosts( + g_ceph_context, "noservice", DNSResolver::SRV_Protocol::TCP, "", &addrs); + + ASSERT_EQ(ret, 0); + ASSERT_EQ(addrs.size(), (unsigned int)0); +} + diff --git a/src/test/mon/MonMap.cc b/src/test/mon/MonMap.cc new file mode 100644 index 0000000000000..83d9e6e46f522 --- /dev/null +++ b/src/test/mon/MonMap.cc @@ -0,0 +1,220 @@ +// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- +// vim: ts=8 sw=2 smarttab +/* + * Ceph - scalable distributed file system + * + * Copyright (C) 2016 SUSE LINUX GmbH + * + * This is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License version 2.1, as published by the Free Software + * Foundation. See file COPYING. + * + */ +#include "mon/MonMap.h" +#include "common/ceph_context.h" +#include "common/dns_resolve.h" +#include "test/common/dns_messages.h" +#include "test/unit.h" + +#include "common/debug.h" +#include "gmock/gmock.h" + +#include + +#define TEST_DEBUG 20 + +#define dout_subsys ceph_subsys_mon + + +using ::testing::Return; +using ::testing::_; +using ::testing::SetArrayArgument; +using ::testing::DoAll; +using ::testing::StrEq; + + +class MonMapTest : public ::testing::Test { + protected: + virtual void SetUp() { + g_ceph_context->_conf->subsys.set_log_level(dout_subsys, TEST_DEBUG); + } + + virtual void TearDown() { + DNSResolver::get_instance(nullptr); + } +}; + +TEST_F(MonMapTest, build_initial_config_from_dns) { + + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + DNSResolver::get_instance(resolvH); + + int len = sizeof(ns_search_msg_ok_payload); + int lena = sizeof(ns_query_msg_mon_a_payload); + int lenb = sizeof(ns_query_msg_mon_b_payload); + int lenc = sizeof(ns_query_msg_mon_c_payload); + + using ::testing::InSequence; + { + InSequence s; + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_cephmon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_nquery(_, StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_cephmon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#endif + } + + + + CephContext *cct = (new CephContext(CEPH_ENTITY_TYPE_MON))->get(); + cct->_conf->set_val("mon_dns_srv_name", "cephmon", false); + MonMap monmap; + int r = monmap.build_initial(cct, std::cerr); + + ASSERT_EQ(r, 0); + ASSERT_EQ(monmap.mon_addr.size(), (unsigned int)3); + map::iterator it = monmap.mon_addr.find("mon.a"); + ASSERT_NE(it, monmap.mon_addr.end()); + std::ostringstream os; + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.11:6789/0"); + os.str(""); + it = monmap.mon_addr.find("mon.b"); + ASSERT_NE(it, monmap.mon_addr.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.12:6789/0"); + os.str(""); + it = monmap.mon_addr.find("mon.c"); + ASSERT_NE(it, monmap.mon_addr.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.13:6789/0"); +} + +TEST_F(MonMapTest, build_initial_config_from_dns_fail) { + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + DNSResolver::get_instance(resolvH); + + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_ceph-mon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(Return(0)); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_ceph-mon._tcp"), C_IN, T_SRV, _, _)) + .WillOnce(Return(0)); +#endif + + CephContext *cct = (new CephContext(CEPH_ENTITY_TYPE_MON))->get(); + // using default value of mon_dns_srv_name option + MonMap monmap; + int r = monmap.build_initial(cct, std::cerr); + + ASSERT_EQ(r, -ENOENT); + ASSERT_EQ(monmap.mon_addr.size(), (unsigned int)0); + +} + +TEST_F(MonMapTest, build_initial_config_from_dns_with_domain) { + + MockResolvHWrapper *resolvH = new MockResolvHWrapper(); + DNSResolver::get_instance(resolvH); + + int len = sizeof(ns_search_msg_ok_payload); + int lena = sizeof(ns_query_msg_mon_a_payload); + int lenb = sizeof(ns_query_msg_mon_b_payload); + int lenc = sizeof(ns_query_msg_mon_c_payload); + + using ::testing::InSequence; + { + InSequence s; + +#ifdef HAVE_RES_NQUERY + EXPECT_CALL(*resolvH, res_nsearch(_, StrEq("_cephmon._tcp.ceph.com"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_nquery(_, StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_nquery(_,StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<4>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#else + EXPECT_CALL(*resolvH, res_search(StrEq("_cephmon._tcp.ceph.com"), C_IN, T_SRV, _, _)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_search_msg_ok_payload, + ns_search_msg_ok_payload+len), Return(len))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.a.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_a_payload, + ns_query_msg_mon_a_payload+lena), Return(lena))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.c.ceph.com"), C_IN, T_A,_,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_c_payload, + ns_query_msg_mon_c_payload+lenc), Return(lenc))); + + EXPECT_CALL(*resolvH, res_query(StrEq("mon.b.ceph.com"), C_IN, T_A, _,_)) + .WillOnce(DoAll(SetArrayArgument<3>(ns_query_msg_mon_b_payload, + ns_query_msg_mon_b_payload+lenb), Return(lenb))); +#endif + } + + + + CephContext *cct = (new CephContext(CEPH_ENTITY_TYPE_MON))->get(); + cct->_conf->set_val("mon_dns_srv_name", "cephmon_ceph.com", false); + MonMap monmap; + int r = monmap.build_initial(cct, std::cerr); + + ASSERT_EQ(r, 0); + ASSERT_EQ(monmap.mon_addr.size(), (unsigned int)3); + map::iterator it = monmap.mon_addr.find("mon.a"); + ASSERT_NE(it, monmap.mon_addr.end()); + std::ostringstream os; + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.11:6789/0"); + os.str(""); + it = monmap.mon_addr.find("mon.b"); + ASSERT_NE(it, monmap.mon_addr.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.12:6789/0"); + os.str(""); + it = monmap.mon_addr.find("mon.c"); + ASSERT_NE(it, monmap.mon_addr.end()); + os << it->second; + ASSERT_EQ(os.str(), "192.168.1.13:6789/0"); +} + -- 2.39.5