redis/tests/conn_connect.cpp

/* Copyright (c) 2018-2022 Marcelo Zimbres Silva (mzimbres@gmail.com)
 *
 * Distributed under the Boost Software License, Version 1.0. (See
 * accompanying file LICENSE.txt)
 */

#include <iostream>
#include <boost/asio.hpp>
#include <boost/system/errc.hpp>

#define BOOST_TEST_MODULE low level
#include <boost/test/included/unit_test.hpp>

#include <aedis.hpp>
#include <aedis/src.hpp>

namespace net = boost::asio;

using connection = aedis::connection<>;
using endpoint = aedis::endpoint;
using error_code = boost::system::error_code;

bool is_host_not_found(error_code ec)
{
   if (ec == net::error::netdb_errors::host_not_found) return true;
   if (ec == net::error::netdb_errors::host_not_found_try_again) return true;
   return false;
}

error_code test_async_run(endpoint ep, connection::timeouts cfg = {})
{
   net::io_context ioc;
   connection db{ioc};
   error_code ret;
   db.async_run(ep, cfg, [&](auto ec) { ret = ec; });
   ioc.run();
   return ret;
}

BOOST_AUTO_TEST_CASE(resolve_bad_host)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   endpoint ep;
   ep.host = "Atibaia";
   ep.port = "6379";

   connection::timeouts cfg;
   cfg.resolve_timeout = std::chrono::seconds{100};
   auto const ec = test_async_run(ep, cfg);

   BOOST_TEST(is_host_not_found(ec));
}

BOOST_AUTO_TEST_CASE(resolve_with_timeout)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   endpoint ep;
   ep.host = "Atibaia";
   ep.port = "6379";

   connection::timeouts cfg;
   // Low-enough to cause a timeout always.
   cfg.resolve_timeout = std::chrono::milliseconds{1};
   auto const ec = test_async_run(ep, cfg);
   BOOST_CHECK_EQUAL(ec, aedis::error::resolve_timeout);
}

BOOST_AUTO_TEST_CASE(connect_bad_port)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   endpoint ep;
   ep.host = "127.0.0.1";
   ep.port = "1";

   connection::timeouts cfg;
   cfg.connect_timeout = std::chrono::seconds{100};
   auto const ec = test_async_run(ep, cfg);
   BOOST_CHECK_EQUAL(ec, net::error::basic_errors::connection_refused);
}

BOOST_AUTO_TEST_CASE(connect_with_timeout)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   endpoint ep;
   ep.host = "example.com";
   ep.port = "1";

   connection::timeouts cfg;
   cfg.connect_timeout = std::chrono::milliseconds{1};
   auto const ec = test_async_run(ep, cfg);
   BOOST_CHECK_EQUAL(ec, aedis::error::connect_timeout);
}

BOOST_AUTO_TEST_CASE(bad_hello_response)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   // Succeeds with the tcp connection but fails the hello.
   endpoint ep;
   ep.host = "google.com";
   ep.port = "80";

   auto const ec = test_async_run(ep);
   BOOST_CHECK_EQUAL(ec, aedis::error::invalid_data_type);
}

BOOST_AUTO_TEST_CASE(plain_conn_on_tls_endpoint)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   endpoint ep;
   ep.host = "google.com";
   ep.port = "443";

   auto const ec = test_async_run(ep);
   BOOST_TEST(!!ec);
}

auto auth_fail_error(boost::system::error_code ec)
{
   return ec == aedis::error::resp3_handshake_error ||
          ec == aedis::error::exec_timeout;
}

BOOST_AUTO_TEST_CASE(auth_fail)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   // Should cause an error in the authentication as our redis server
   // has no authentication configured.
   endpoint ep;
   ep.host = "127.0.0.1";
   ep.port = "6379";
   ep.username = "caboclo-do-mato";
   ep.password = "jabuticaba";

   auto const ec = test_async_run(ep);
   BOOST_TEST(auth_fail_error(ec));
}

auto wrong_role_error(boost::system::error_code ec)
{
   return ec == aedis::error::unexpected_server_role ||
          ec == aedis::error::exec_timeout;
}

BOOST_AUTO_TEST_CASE(wrong_role)
{
   std::cout << boost::unit_test::framework::current_test_case().p_name << std::endl;

   // Should cause an error in the authentication as our redis server
   // has no authentication configured.
   endpoint ep;
   ep.host = "127.0.0.1";
   ep.port = "6379";
   ep.role = "errado";

   auto const ec = test_async_run(ep);
   BOOST_TEST(wrong_role_error(ec));
}