diff options
Diffstat (limited to 'src/mongo/transport/max_conns_override_test.cpp')
-rw-r--r-- | src/mongo/transport/max_conns_override_test.cpp | 84 |
1 files changed, 0 insertions, 84 deletions
diff --git a/src/mongo/transport/max_conns_override_test.cpp b/src/mongo/transport/max_conns_override_test.cpp deleted file mode 100644 index 195a4c030c6..00000000000 --- a/src/mongo/transport/max_conns_override_test.cpp +++ /dev/null @@ -1,84 +0,0 @@ -/* Copyright 2018 MongoDB Inc. - * - * This program is free software: you can redistribute it and/or modify - * it under the terms of the GNU Affero General Public License, version 3, - * as published by the Free Software Foundation. - * - * This program 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 Affero General Public License for more details. - * - * You should have received a copy of the GNU Affero General Public License - * along with this program. If not, see <http://www.gnu.org/licenses/>. - * - * As a special exception, the copyright holders give permission to link the - * code of portions of this program with the OpenSSL library under certain - * conditions as described in each individual source file and distribute - * linked combinations including the program with the OpenSSL library. You - * must comply with the GNU Affero General Public License in all respects - * for all of the code used other than as permitted herein. If you modify - * file(s) with this exception, you may extend this exception to your - * version of the file(s), but you are not obligated to do so. If you do not - * wish to do so, delete this exception statement from your version. If you - * delete this exception statement from all source files in the program, - * then also delete it in the license file. - */ - -#include "mongo/transport/mock_session.h" -#include "mongo/transport/service_entry_point_impl.h" -#include "mongo/unittest/unittest.h" - -namespace mongo { -namespace { - -using ExemptionVector = std::vector<stdx::variant<CIDR, std::string>>; - -template <typename T> -stdx::variant<CIDR, std::string> makeExemption(T exemption) { - auto swCIDR = CIDR::parse(exemption); - if (swCIDR.isOK()) { - return swCIDR.getValue(); - } else { - return std::string{exemption}; - } -} - -transport::SessionHandle makeIPSession(StringData ip) { - return transport::MockSession::create( - HostAndPort(SockAddr(ip, 27017, AF_INET)), HostAndPort(), nullptr); -} - -transport::SessionHandle makeUNIXSession(StringData path) { - return transport::MockSession::create(HostAndPort(SockAddr(""_sd, -1, AF_UNIX)), - HostAndPort(SockAddr(path, -1, AF_UNIX)), - nullptr); -} - -TEST(MaxConnsOverride, NormalCIDR) { - ExemptionVector cidrOnly{makeExemption("127.0.0.1"), makeExemption("10.0.0.0/24")}; - - ASSERT_TRUE(shouldOverrideMaxConns(makeIPSession("127.0.0.1"), cidrOnly)); - ASSERT_TRUE(shouldOverrideMaxConns(makeIPSession("10.0.0.35"), cidrOnly)); - ASSERT_FALSE(shouldOverrideMaxConns(makeIPSession("192.168.0.53"), cidrOnly)); -} - -TEST(MaxConnsOverride, UNIXPaths) { - ExemptionVector mixed{makeExemption("127.0.0.1"), - makeExemption("10.0.0.0/24"), - makeExemption("/tmp/mongod.sock")}; - - ASSERT_TRUE(shouldOverrideMaxConns(makeIPSession("127.0.0.1"), mixed)); - ASSERT_TRUE(shouldOverrideMaxConns(makeIPSession("10.0.0.35"), mixed)); - ASSERT_FALSE(shouldOverrideMaxConns(makeIPSession("192.168.0.53"), mixed)); -#ifdef _WIN32 - ASSERT_FALSE(shouldOverrideMaxConns(makeUNIXSession("/tmp/mongod.sock"), mixed)); -#else - ASSERT_TRUE(shouldOverrideMaxConns(makeUNIXSession("/tmp/mongod.sock"), mixed)); -#endif - ASSERT_FALSE(shouldOverrideMaxConns(makeUNIXSession("/tmp/other-mongod.sock"), mixed)); -} - - -} // namespace -} // namespace |