modules/platforms/cpp/tests/fake_server/connection_test.cpp

/*
* Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "fake_server.h"
#include "ignite/client/ignite_client.h"
#include "proxy/asio_proxy.h"
#include "tests/client-test/ignite_runner_suite.h"

#include <gtest/gtest.h>
#include <thread>


using namespace ignite;
using namespace std::chrono_literals;

class connection_test : public ignite_runner_suite {
protected:
    static constexpr int PORT = 50800;

    std::vector<std::string> get_endpoints() const {
        return {"127.0.0.1:" + std::to_string(PORT)};
    }
};


TEST_F(connection_test, handshake_success) {
    fake_server fs{PORT, get_logger()};

    fs.start();

    ignite_client_configuration cfg;
    cfg.set_logger(get_logger());
    cfg.set_endpoints(get_endpoints());

    auto cl = ignite_client::start(cfg, 5s);
}

TEST_F(connection_test, request_timeout) {
    fake_server fs{
        PORT,
        get_logger(),
        [](protocol::client_operation op) -> std::unique_ptr<response_action> {
            switch (op) {
                case protocol::client_operation::CLUSTER_GET_NODES:
                    return std::make_unique<drop_action>();
                default:
                    return nullptr;
            }
        }
    };

    fs.start();

    ignite_client_configuration cfg;
    cfg.set_logger(get_logger());
    cfg.set_endpoints(get_endpoints());
    cfg.set_operation_timeout(std::chrono::milliseconds{100});

    auto cl = ignite_client::start(cfg, 5s);

    try {
        auto cluster_nodes = cl.get_cluster_nodes();
    } catch (ignite_error& err) {
        EXPECT_EQ(error::code::OPERATION_TIMEOUT, err.get_status_code());
    }
}

TEST_F(connection_test, using_asio) {
    fake_server fs{50900, get_logger()};
    fs.start();

    auto in_listener = std::make_shared<proxy::message_listener>();
    auto out_listener = std::make_shared<proxy::message_listener>();

    proxy::asio_proxy proxy{
        {
            proxy::configuration(50800, "127.0.0.1:50900", in_listener, out_listener)
        },
        get_logger()
    };


    ignite_client_configuration cfg;
    cfg.set_logger(get_logger());
    cfg.set_endpoints(get_endpoints());

    auto cl = ignite_client::start(cfg, 5s);

    auto cluster_nodes = cl.get_cluster_nodes();

    ASSERT_EQ(1, cluster_nodes.size());

    ASSERT_GT(in_listener->get_msg_queue().size(), 1);
    ASSERT_GT(out_listener->get_msg_queue().size(), 1);
}